feat: 添加oss api

2025-10-11 15:13:58 +08:00
parent 237ac665e6
commit 7e282b36d7
21 changed files with 545 additions and 5 deletions
@@ -0,0 +1,103 @@
+package upload
+
+import (
+	"bytes"
+	"context"
+	"errors"
+	"io"
+	"mime/multipart"
+	"path/filepath"
+	"strings"
+	"sundynix-go/global"
+	"sundynix-go/utils"
+	"time"
+
+	"github.com/minio/minio-go/v7"
+	"github.com/minio/minio-go/v7/pkg/credentials"
+	"go.uber.org/zap"
+)
+
+var MinioClient *Minio // 优化性能，但是不支持动态配置
+type Minio struct {
+	Client *minio.Client
+	bucket string
+}
+
+func GetMinio(endpoint, accessKey, secretKey, bucketName string, useSSL bool) (*Minio, error) {
+	if MinioClient != nil {
+		return MinioClient, nil
+	}
+	// Initialize minio client object.
+	minioClient, err := minio.New(endpoint, &minio.Options{
+		Creds:  credentials.NewStaticV4(accessKey, secretKey, ""),
+		Secure: useSSL, // Set to true if using https
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	// 创建bucket
+	err = minioClient.MakeBucket(context.Background(), bucketName, minio.MakeBucketOptions{})
+	if err != nil {
+		// 判断是否已经存在
+		exists, errBucketExists := minioClient.BucketExists(context.Background(), bucketName)
+		if errBucketExists == nil && exists {
+			global.Logger.Info("Bucket already exists")
+		} else {
+			return nil, err
+		}
+	}
+	MinioClient = &Minio{
+		Client: minioClient,
+		bucket: bucketName,
+	}
+	return MinioClient, nil
+}
+
+func (m *Minio) UploadFile(file *multipart.FileHeader) (filePathres, key string, uploadErr error) {
+	f, openErr := file.Open()
+	// mutipart.File to os.File
+	if openErr != nil {
+		global.Logger.Error("function file.Open() Failed", zap.Any("err", openErr.Error()))
+		return "", "", errors.New("function file.Open() Failed, err:" + openErr.Error())
+	}
+	buffer := bytes.Buffer{}
+	_, err := io.Copy(&buffer, f)
+	if err != nil {
+		global.Logger.Error("读取文件失败", zap.Any("err", err.Error()))
+		return "", "", errors.New("读取文件失败, err:" + err.Error())
+	}
+	f.Close() // 创建文件 defer 关闭
+
+	//对文件名进行加密存储
+	ext := filepath.Ext(file.Filename)
+	filename := utils.MD5V([]byte(strings.TrimSuffix(file.Filename, ext))) + ext
+	if global.Config.Minio.BasePath == "" {
+		filePathres = "uploads/" + time.Now().Format("2006-01-02") + "/" + filename // uploads/2025-09-17/xxxx.png
+	} else {
+		filePathres = global.Config.Minio.BasePath + "/" + time.Now().Format("2006-01-02") + "/" + filename
+	}
+	// 设置超时10分钟
+	ctx, cancel := context.WithTimeout(context.Background(), time.Minute*10)
+	defer cancel()
+
+	//大文件自动切换为分片上传
+	info, err := m.Client.PutObject(ctx, global.Config.Minio.BucketName, filePathres, &buffer, file.Size, minio.PutObjectOptions{
+		ContentType: "application/octet-stream",
+	})
+	if err != nil {
+		global.Logger.Error("上传文件到minio失败", zap.Any("err", err.Error()))
+		return "", "", errors.New("上传文件到minio失败, err:" + err.Error())
+	}
+	//http://127.0.0.1:9000/planting-fun/uploads/2025-09-17/211476f3837fc7acbaebf0f901c1bd68.png
+	return global.Config.Minio.BucketUrl + "/" + info.Key, filePathres, nil
+
+}
+
+func (m *Minio) DeleteFile(key string) error {
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
+	defer cancel()
+
+	err := m.Client.RemoveObject(ctx, m.bucket, key, minio.RemoveObjectOptions{})
+	return err
+}
@@ -0,0 +1,31 @@
+package upload
+
+import (
+	"fmt"
+	"mime/multipart"
+	"sundynix-go/global"
+)
+
+// oss 对象存储接口
+type Oss interface {
+	UploadFile(file *multipart.FileHeader) (string, string, error)
+	DeleteFile(key string) error
+}
+
+// OssInstance 实例化oos方法
+func OssInstance() Oss {
+	switch global.Config.System.OssType {
+	case "local":
+		fmt.Println("local")
+	case "tencent-cos":
+		return &TencentCOS{}
+	case "minio":
+		minioClient, err := GetMinio(global.Config.Minio.Endpoint, global.Config.Minio.AccessKeyId, global.Config.Minio.AccessKeySecret, global.Config.Minio.BucketName, global.Config.Minio.UseSSL)
+		if err != nil {
+			global.Logger.Warn("minio初始化失败,请检查minio可用性或安全配置:" + err.Error())
+			panic("minio初始化失败,请检查minio可用性或安全配置")
+		}
+		return minioClient
+	}
+	return nil
+}
@@ -0,0 +1,60 @@
+package upload
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"mime/multipart"
+	"net/http"
+	"net/url"
+	"sundynix-go/global"
+	"time"
+
+	"github.com/tencentyun/cos-go-sdk-v5"
+	"go.uber.org/zap"
+)
+
+type TencentCOS struct{}
+
+// NewClient 创建一个腾讯云COS客户端
+func NewClient() *cos.Client {
+	urlStr, _ := url.Parse("https://" + global.Config.TencentCOS.Bucket + ".cos." + global.Config.TencentCOS.Region + ".myqcloud.com")
+	baseURL := &cos.BaseURL{BucketURL: urlStr}
+	client := cos.NewClient(baseURL, &http.Client{
+		Transport: &cos.AuthorizationTransport{
+			SecretID:  global.Config.TencentCOS.SecretID,
+			SecretKey: global.Config.TencentCOS.SecretKey,
+		},
+	})
+	return client
+}
+
+// UploadFile upload file to COS
+func (*TencentCOS) UploadFile(file *multipart.FileHeader) (string, string, error) {
+	client := NewClient()
+	f, openError := file.Open()
+	if openError != nil {
+		global.Logger.Error("function file.Open() failed", zap.Any("err", openError.Error()))
+		return "", "", errors.New("function file.Open() failed, err:" + openError.Error())
+	}
+	defer f.Close() // 创建文件 defer 关闭
+	fileKey := fmt.Sprintf("%d%s", time.Now().Unix(), file.Filename)
+
+	_, err := client.Object.Put(context.Background(), global.Config.TencentCOS.PathPrefix+"/"+fileKey, f, nil)
+	if err != nil {
+		panic(err)
+	}
+	return global.Config.TencentCOS.BaseURL + "/" + global.Config.TencentCOS.PathPrefix + "/" + fileKey, fileKey, nil
+}
+
+// DeleteFile delete file form COS
+func (*TencentCOS) DeleteFile(key string) error {
+	client := NewClient()
+	name := global.Config.TencentCOS.PathPrefix + "/" + key
+	_, err := client.Object.Delete(context.Background(), name)
+	if err != nil {
+		global.Logger.Error("function bucketManager.Delete() failed", zap.Any("err", err.Error()))
+		return errors.New("function bucketManager.Delete() failed, err:" + err.Error())
+	}
+	return nil
+}