优化

internal/biz/media_upload.go

@@ -7,6 +7,7 @@ import (
 	"errors"
 	"fmt"
 	"io"
+	"path"
 	"path/filepath"
 	"sort"
 	"strings"
@@ -15,6 +16,20 @@ import (
 	"github.com/google/uuid"
 )
 
+func (uc *MediaUsecase) chunkPrefix(uploadID uint) string {
+	directory := ".chunks"
+	if uc.settings != nil {
+		if configured := strings.Trim(uc.settings.MediaSettings().ChunkDir, "/\\ "); configured != "" {
+			directory = configured
+		}
+	}
+	return path.Join(directory, fmt.Sprintf("%d", uploadID))
+}
+
+func (uc *MediaUsecase) chunkKey(uploadID uint, index int) string {
+	return path.Join(uc.chunkPrefix(uploadID), fmt.Sprintf("%08d", index))
+}
+
 func (uc *MediaUsecase) InitUpload(ctx context.Context, userID uint, name, hash string, size, chunkSize int64, total int) (*UploadSession, *MediaFile, []int, error) {
 	if err := validateMediaName(name); err != nil {
 		return nil, nil, nil, err
@@ -57,7 +72,7 @@ func (uc *MediaUsecase) SaveChunk(ctx context.Context, userID, uploadID uint, in
 		return errors.New("上传会话状态不允许收片")
 	}
 	hash := md5.New()
-	key := fmt.Sprintf(".chunks/%d/%08d", uploadID, index)
+	key := uc.chunkKey(uploadID, index)
 	stored, err := uc.files.Put(ctx, key, io.TeeReader(reader, hash))
 	if err != nil {
 		return err
@@ -95,7 +110,7 @@ func (uc *MediaUsecase) CompleteUpload(ctx context.Context, userID, uploadID uin
 		if chunk.Index != index {
 			return fail(errors.New("分片序号不连续"))
 		}
-		names = append(names, fmt.Sprintf(".chunks/%d/%08d", uploadID, index))
+		names = append(names, uc.chunkKey(uploadID, index))
 	}
 	ext := strings.ToLower(filepath.Ext(session.FileName))
 	key := time.Now().Format("20060102") + "/" + uuid.NewString() + ext
@@ -114,7 +129,7 @@ func (uc *MediaUsecase) CompleteUpload(ctx context.Context, userID, uploadID uin
 	}
 	_ = uc.CompleteUploadSession(ctx, uploadID, key, media.ID)
 	_ = uc.DeleteChunks(ctx, uploadID)
-	_ = uc.files.DeletePrefix(ctx, fmt.Sprintf(".chunks/%d", uploadID))
+	_ = uc.files.DeletePrefix(ctx, uc.chunkPrefix(uploadID))
 	return media, nil
 }
 func (uc *MediaUsecase) CancelUpload(ctx context.Context, userID, uploadID uint) error {
@@ -123,7 +138,7 @@ func (uc *MediaUsecase) CancelUpload(ctx context.Context, userID, uploadID uint)
 		return errors.New("上传会话不存在或无权操作")
 	}
 	_ = uc.DeleteChunks(ctx, uploadID)
-	_ = uc.files.DeletePrefix(ctx, fmt.Sprintf(".chunks/%d", uploadID))
+	_ = uc.files.DeletePrefix(ctx, uc.chunkPrefix(uploadID))
 	return uc.DeleteUploadSession(ctx, uploadID)
 }
 func (uc *MediaUsecase) CleanupStale(ctx context.Context, ttlHours int) error {
@@ -138,7 +153,7 @@ func (uc *MediaUsecase) CleanupStale(ctx context.Context, ttlHours int) error {
 		if err = uc.DeleteUploadData(ctx, id); err != nil {
 			return err
 		}
-		if err = uc.files.DeletePrefix(ctx, fmt.Sprintf(".chunks/%d", id)); err != nil {
+		if err = uc.files.DeletePrefix(ctx, uc.chunkPrefix(id)); err != nil {
 			return err
 		}
 	}