feature: 集成华为云obs 对象存储

server/utils/upload/obs.go

@@ -0,0 +1,65 @@
+package upload
+
+import (
+	"github.com/flipped-aurora/gin-vue-admin/server/global"
+	"github.com/huaweicloud/huaweicloud-sdk-go-obs/obs"
+	"github.com/pkg/errors"
+	"mime/multipart"
+)
+
+var HuaWeiObs = new(_obs)
+
+type _obs struct{}
+
+func NewHuaWeiObsClient() (client *obs.ObsClient, err error) {
+	return obs.New(global.GVA_CONFIG.HuaWeiObs.AccessKey, global.GVA_CONFIG.HuaWeiObs.SecretKey, global.GVA_CONFIG.HuaWeiObs.Endpoint)
+}
+
+func (o *_obs) UploadFile(file *multipart.FileHeader) (filename string, filepath string, err error) {
+	var open multipart.File
+	open, err = file.Open()
+	if err != nil {
+		return filename, filepath, err
+	}
+	filename = file.Filename
+	input := &obs.PutObjectInput{
+		PutObjectBasicInput: obs.PutObjectBasicInput{
+			ObjectOperationInput: obs.ObjectOperationInput{
+				Bucket: global.GVA_CONFIG.HuaWeiObs.Bucket,
+				Key:    filename,
+			},
+			ContentType: file.Header.Get("content-type"),
+		},
+		Body: open,
+	}
+
+	var client *obs.ObsClient
+	client, err = NewHuaWeiObsClient()
+	if err != nil {
+		return filepath, filename, errors.Wrap(err, "获取华为对象存储对象失败!")
+	}
+
+	_, err = client.PutObject(input)
+	if err != nil {
+		return filepath, filename, errors.Wrap(err, "文件上传失败!")
+	}
+	filepath = global.GVA_CONFIG.HuaWeiObs.Path + "/" + filename
+	return filepath, filename, err
+}
+
+func (o *_obs) DeleteFile(key string) error {
+	client, err := NewHuaWeiObsClient()
+	if err != nil {
+		return errors.Wrap(err, "获取华为对象存储对象失败!")
+	}
+	input := &obs.DeleteObjectInput{
+		Bucket: global.GVA_CONFIG.HuaWeiObs.Bucket,
+		Key:    key,
+	}
+	var output *obs.DeleteObjectOutput
+	output, err = client.DeleteObject(input)
+	if err != nil {
+		return errors.Wrapf(err, "删除对象(%s)失败!, output: %v", key, output)
+	}
+	return nil
+}

server/utils/upload/upload.go

@@ -6,23 +6,17 @@ import (
 	"github.com/flipped-aurora/gin-vue-admin/server/global"
 )
 
-//@author: [ccfish86](https://github.com/ccfish86)
-//@author: [SliverHorn](https://github.com/SliverHorn)
-//@interface_name: OSS
-//@description: OSS接口
-
+// OSS 对象存储接口
+// Author [SliverHorn](https://github.com/SliverHorn)
+// Author [ccfish86](https://github.com/ccfish86)
 type OSS interface {
 	UploadFile(file *multipart.FileHeader) (string, string, error)
 	DeleteFile(key string) error
 }
 
-//@author: [ccfish86](https://github.com/ccfish86)
-//@author: [SliverHorn](https://github.com/SliverHorn)
-//@function: NewOss
-//@description: OSS接口
-//@description: OSS的实例化方法
-//@return: OSS
-
+// NewOss OSS的实例化方法
+// Author [SliverHorn](https://github.com/SliverHorn)
+// Author [ccfish86](https://github.com/ccfish86)
 func NewOss() OSS {
 	switch global.GVA_CONFIG.System.OssType {
 	case "local":
@@ -33,6 +27,8 @@ func NewOss() OSS {
 		return &TencentCOS{}
 	case "aliyun-oss":
 		return &AliyunOSS{}
+	case "huawei-obs":
+		return HuaWeiObs
 	default:
 		return &Local{}
 	}