wangpengfei
2023-08-25 25c573d55986e02cf5f70cc3868e2b94a4be98e2
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package upload
 
import (
    "errors"
    "mime/multipart"
    "time"
 
    "github.com/aliyun/aliyun-oss-go-sdk/oss"
    "go.uber.org/zap"
    "srm/global"
)
 
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())
    }
    defer f.Close() // 创建文件 defer 关闭
    // 上传阿里云路径 文件名格式 自己可以改 建议保证唯一性
    // yunFileTmpPath := filepath.Join("uploads", time.Now().Format("2006-01-02")) + "/" + file.Filename
    yunFileTmpPath := global.GVA_CONFIG.AliyunOSS.BasePath + "/" + "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())
    }
 
    // 删除单个文件。objectName表示删除OSS文件时需要指定包含文件后缀在内的完整路径,例如abc/efg/123.jpg。
    // 如需删除文件夹,请将objectName设置为对应的文件夹名称。如果文件夹非空,则需要将文件夹下的所有object删除后才能删除该文件夹。
    err = bucket.DeleteObject(key)
    if err != nil {
        global.GVA_LOG.Error("function bucketManager.Delete() failed", zap.Any("err", err.Error()))
        return errors.New("function bucketManager.Delete() failed, err:" + err.Error())
    }
 
    return nil
}
 
func NewBucket() (*oss.Bucket, error) {
    // 创建OSSClient实例。
    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
}