feat: add oss service factory implements, add support for setting custom upload handler, localstorage and minio oss

api/service/minio_service.go

@@ -1,54 +0,0 @@
-package service
-
-import (
-	"chatplus/core/types"
-	"chatplus/utils"
-	"context"
-	"fmt"
-	"github.com/minio/minio-go/v7"
-	"github.com/minio/minio-go/v7/pkg/credentials"
-	"net/url"
-	"path"
-	"strings"
-)
-
-type MinioService struct {
-	config *types.AppConfig
-	client *minio.Client
-}
-
-func NewMinioService(config *types.AppConfig) (*MinioService, error) {
-	minioClient, err := minio.New(config.MinioConfig.Endpoint, &minio.Options{
-		Creds:  credentials.NewStaticV4(config.MinioConfig.AccessKey, config.MinioConfig.AccessSecret, ""),
-		Secure: config.MinioConfig.UseSSL,
-	})
-	if err != nil {
-		return nil, err
-	}
-	return &MinioService{config: config, client: minioClient}, nil
-}
-
-func (s *MinioService) UploadMjImg(imageURL string) (string, error) {
-	parsedURL, err := url.Parse(imageURL)
-	if err != nil {
-		return "", err
-	}
-
-	filename := path.Base(parsedURL.Path)
-	imageBytes, err := utils.DownloadImage(imageURL, s.config.ProxyURL)
-	if err != nil {
-		return "", err
-	}
-
-	info, err := s.client.PutObject(
-		context.Background(),
-		s.config.MinioConfig.Bucket,
-		filename,
-		strings.NewReader(string(imageBytes)),
-		int64(len(imageBytes)),
-		minio.PutObjectOptions{ContentType: "image/png"})
-	if err != nil {
-		return "", err
-	}
-	return fmt.Sprintf("%s/%s/%s", s.config.MinioConfig.Domain, s.config.MinioConfig.Bucket, info.Key), nil
-}

api/service/oss/localstorage_service.go

@@ -0,0 +1,57 @@
+package oss
+
+import (
+	"chatplus/core/types"
+	"chatplus/utils"
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"path/filepath"
+)
+
+type LocalStorageService struct {
+	config   *types.LocalStorageConfig
+	proxyURL string
+}
+
+func NewLocalStorageService(config *types.AppConfig) LocalStorageService {
+	return LocalStorageService{
+		config:   &config.OSS.Local,
+		proxyURL: config.ProxyURL,
+	}
+}
+
+func (s LocalStorageService) PutFile(ctx *gin.Context) (string, error) {
+	file, err := ctx.FormFile("file")
+	if err != nil {
+		return "", fmt.Errorf("error with get form: %v", err)
+	}
+
+	filePath, err := utils.GenUploadPath(s.config.BasePath, file.Filename)
+	if err != nil {
+		return "", fmt.Errorf("error with generate filename: %s", err.Error())
+	}
+	// 将文件保存到指定路径
+	err = ctx.SaveUploadedFile(file, filePath)
+	if err != nil {
+		return "", fmt.Errorf("error with save upload file: %s", err.Error())
+	}
+
+	return utils.GenUploadUrl(s.config.BasePath, s.config.BaseURL, filePath), nil
+}
+
+func (s LocalStorageService) PutImg(imageURL string) (string, error) {
+	filename := filepath.Base(imageURL)
+	filePath, err := utils.GenUploadPath(s.config.BasePath, filename)
+	if err != nil {
+		return "", fmt.Errorf("error with generate image dir: %v", err)
+	}
+
+	err = utils.DownloadFile(imageURL, filePath, s.proxyURL)
+	if err != nil {
+		return "", fmt.Errorf("error with download image: %v", err)
+	}
+
+	return utils.GenUploadUrl(s.config.BasePath, s.config.BaseURL, filePath), nil
+}
+
+var _ Uploader = LocalStorageService{}

api/service/oss/minio_service.go

@@ -0,0 +1,78 @@
+package oss
+
+import (
+	"chatplus/core/types"
+	"chatplus/utils"
+	"context"
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"github.com/minio/minio-go/v7"
+	"github.com/minio/minio-go/v7/pkg/credentials"
+	"path/filepath"
+	"strings"
+	"time"
+)
+
+type MinioService struct {
+	config   *types.MinioConfig
+	client   *minio.Client
+	proxyURL string
+}
+
+func NewMinioService(appConfig *types.AppConfig) (MinioService, error) {
+	config := &appConfig.OSS.Minio
+	minioClient, err := minio.New(config.Endpoint, &minio.Options{
+		Creds:  credentials.NewStaticV4(config.AccessKey, config.AccessSecret, ""),
+		Secure: config.UseSSL,
+	})
+	if err != nil {
+		return MinioService{}, err
+	}
+	return MinioService{config: config, client: minioClient, proxyURL: appConfig.ProxyURL}, nil
+}
+
+func (s MinioService) PutImg(imageURL string) (string, error) {
+	imageData, err := utils.DownloadImage(imageURL, s.proxyURL)
+	if err != nil {
+		return "", fmt.Errorf("error with download image: %v", err)
+	}
+	fileExt := filepath.Ext(filepath.Base(imageURL))
+	filename := fmt.Sprintf("%d%s", time.Now().UnixNano(), fileExt)
+	info, err := s.client.PutObject(
+		context.Background(),
+		s.config.Bucket,
+		filename,
+		strings.NewReader(string(imageData)),
+		int64(len(imageData)),
+		minio.PutObjectOptions{ContentType: "image/png"})
+	if err != nil {
+		return "", err
+	}
+	return fmt.Sprintf("%s/%s/%s", s.config.Domain, s.config.Bucket, info.Key), nil
+}
+
+func (s MinioService) PutFile(ctx *gin.Context) (string, error) {
+	file, err := ctx.FormFile("file")
+	if err != nil {
+		return "", fmt.Errorf("error with get form: %v", err)
+	}
+	// Open the uploaded file
+	fileReader, err := file.Open()
+	if err != nil {
+		return "", fmt.Errorf("error opening file: %v", err)
+	}
+	defer fileReader.Close()
+
+	fileExt := filepath.Ext(file.Filename)
+	filename := fmt.Sprintf("%d%s", time.Now().UnixNano(), fileExt)
+	info, err := s.client.PutObject(ctx, s.config.Bucket, filename, fileReader, file.Size, minio.PutObjectOptions{
+		ContentType: file.Header.Get("Content-Type"),
+	})
+	if err != nil {
+		return "", fmt.Errorf("error uploading to MinIO: %v", err)
+	}
+
+	return fmt.Sprintf("%s/%s/%s", s.config.Domain, s.config.Bucket, info.Key), nil
+}
+
+var _ Uploader = MinioService{}

api/service/oss/uploader.go

@@ -0,0 +1,8 @@
+package oss
+
+import "github.com/gin-gonic/gin"
+
+type Uploader interface {
+	PutFile(ctx *gin.Context) (string, error)
+	PutImg(imageURL string) (string, error)
+}

api/service/oss/uploader_manager.go

@@ -0,0 +1,37 @@
+package oss
+
+import (
+	"chatplus/core/types"
+	"strings"
+)
+
+type UploaderManager struct {
+	active         string
+	uploadServices map[string]Uploader
+}
+
+const uploaderLocal = "LOCAL"
+const uploaderMinio = "MINIO"
+
+func NewUploaderManager(config *types.AppConfig) (*UploaderManager, error) {
+	services := make(map[string]Uploader)
+	if config.OSS.Minio.AccessKey != "" {
+		minioService, err := NewMinioService(config)
+		if err != nil {
+			return nil, err
+		}
+		services[uploaderMinio] = minioService
+	}
+	if config.OSS.Local.BasePath != "" {
+		services[uploaderLocal] = NewLocalStorageService(config)
+	}
+	active := uploaderLocal
+	if config.OSS.Active != "" {
+		active = strings.ToUpper(config.OSS.Active)
+	}
+	return &UploaderManager{uploadServices: services, active: active}, nil
+}
+
+func (m *UploaderManager) GetActiveService() Uploader {
+	return m.uploadServices[m.active]
+}