feat: 禁止同名文件覆盖

Author: SugarMGP

internal/apiException/exceptions.go

@@ -24,6 +24,7 @@ var (
 	FileNotImageError   = NewError(200504, log.LevelInfo, "上传的文件不是图片")
 	ResourceNotFound    = NewError(200505, log.LevelInfo, "资源不存在")
 	NoPermission        = NewError(200506, log.LevelInfo, "权限不足")
+	FileAlreadyExists   = NewError(200507, log.LevelInfo, "该文件已存在")
 	BucketNotFound      = NewError(200508, log.LevelInfo, "存储桶不存在")
 
 	NotFound = NewError(200404, log.LevelWarn, http.StatusText(http.StatusNotFound))

internal/controllers/objectController/upload.go

@@ -112,6 +112,10 @@ func handleSingleUpload(
 	// 上传文件
 	objectKey := objectService.GenerateObjectKey(data.Location, name, ext)
 	err = bucket.SaveObject(reader, objectKey)
+	if errors.Is(err, oss.ErrFileAlreadyExists) {
+		element.Error = apiException.FileAlreadyExists.Error()
+		return element
+	}
 	if err != nil {
 		element.Error = apiException.ServerError.Error()
 		return element

pkg/oss/local.go

@@ -33,8 +33,14 @@ func (p *LocalStorageProvider) SaveObject(reader io.ReadSeeker, objectKey string
 	// 根据 objectKey 解析出文件的路径
 	relativePath := filepath.Join(p.path, objectKey)
 
+	// 检查文件是否已经存在
+	_, err := os.Stat(relativePath)
+	if err == nil {
+		return ErrFileAlreadyExists
+	}
+
 	// 创建文件夹，如果文件夹不存在
-	err := os.MkdirAll(filepath.Dir(relativePath), os.ModePerm)
+	err = os.MkdirAll(filepath.Dir(relativePath), os.ModePerm)
 	if err != nil {
 		return err
 	}

pkg/oss/provider.go

@@ -33,4 +33,6 @@ var (
 	ErrResourceNotExists = errors.New("resource not exists")
 	// ErrPathIsNotDir 路径不是目录
 	ErrPathIsNotDir = errors.New("path is not dir")
+	// ErrFileAlreadyExists 文件已存在
+	ErrFileAlreadyExists = errors.New("file already exists")
 )

pkg/oss/s3.go

@@ -10,6 +10,7 @@ import (
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/service/s3"
 	"github.com/aws/aws-sdk-go-v2/service/s3/types"
+	"github.com/aws/smithy-go"
 	"github.com/gabriel-vasile/mimetype"
 )
 
@@ -45,8 +46,14 @@ func (p *S3StorageProvider) SaveObject(reader io.ReadSeeker, objectKey string) e
 		Key:         aws.String(objectKey),
 		Body:        reader,
 		ContentType: aws.String(mime.String()),
+		IfNoneMatch: aws.String("*"),
 	})
-
+	if err != nil {
+		var ae smithy.APIError
+		if errors.As(err, &ae) && ae.ErrorCode() == "PreconditionFailed" {
+			return ErrFileAlreadyExists
+		}
+	}
 	return err
 }