|
@@ -0,0 +1,74 @@
|
|
|
+package upload
|
|
|
+
|
|
|
+import (
|
|
|
+ "errors"
|
|
|
+ "gin-vue-admin/global"
|
|
|
+ "github.com/aliyun/aliyun-oss-go-sdk/oss"
|
|
|
+ "go.uber.org/zap"
|
|
|
+ "mime/multipart"
|
|
|
+ "path/filepath"
|
|
|
+ "time"
|
|
|
+)
|
|
|
+
|
|
|
+type AliyunOSS struct{}
|
|
|
+
|
|
|
+func (*AliyunOSS) UploadFile(file *multipart.FileHeader) (string, string, error) {
|
|
|
+ bucket, err := NewBucket()
|
|
|
+ if err != nil {
|
|
|
+ global.GVA_LOG.Error("function AliyunOSS.NewBucket() Failed", zap.Any("err", err.Error()))
|
|
|
+ return "", "", errors.New("function AliyunOSS.NewBucket() Failed, err:" + err.Error())
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ f, openError := file.Open()
|
|
|
+ if openError != nil {
|
|
|
+ global.GVA_LOG.Error("function file.Open() Failed", zap.Any("err", openError.Error()))
|
|
|
+ return "", "", errors.New("function file.Open() Failed, err:" + openError.Error())
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ yunFileTmpPath := filepath.Join("uploads", time.Now().Format("2006-01-02")) + "/" + file.Filename
|
|
|
+
|
|
|
+
|
|
|
+ err = bucket.PutObject(yunFileTmpPath, f)
|
|
|
+ if err != nil {
|
|
|
+ global.GVA_LOG.Error("function formUploader.Put() Failed", zap.Any("err", err.Error()))
|
|
|
+ return "", "", errors.New("function formUploader.Put() Failed, err:" + err.Error())
|
|
|
+ }
|
|
|
+
|
|
|
+ return global.GVA_CONFIG.AliyunOSS.BucketUrl + "/" + yunFileTmpPath, yunFileTmpPath, nil
|
|
|
+}
|
|
|
+
|
|
|
+func (*AliyunOSS) DeleteFile(key string) error {
|
|
|
+ bucket, err := NewBucket()
|
|
|
+ if err != nil {
|
|
|
+ global.GVA_LOG.Error("function AliyunOSS.NewBucket() Failed", zap.Any("err", err.Error()))
|
|
|
+ return errors.New("function AliyunOSS.NewBucket() Failed, err:" + err.Error())
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ err = bucket.DeleteObject(key)
|
|
|
+ if err != nil {
|
|
|
+ global.GVA_LOG.Error("function bucketManager.Delete() Filed", zap.Any("err", err.Error()))
|
|
|
+ return errors.New("function bucketManager.Delete() Filed, err:" + err.Error())
|
|
|
+ }
|
|
|
+
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+func NewBucket() (*oss.Bucket, error) {
|
|
|
+
|
|
|
+ client, err := oss.New(global.GVA_CONFIG.AliyunOSS.Endpoint, global.GVA_CONFIG.AliyunOSS.AccessKeyId, global.GVA_CONFIG.AliyunOSS.AccessKeySecret)
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ bucket, err := client.Bucket(global.GVA_CONFIG.AliyunOSS.BucketName)
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+
|
|
|
+ return bucket, nil
|
|
|
+}
|