优化

2026-08-16 15:11:38 +08:00
parent f7f6db8695
commit a85477be80
40 changed files with 761 additions and 144 deletions
@@ -76,9 +76,14 @@ func (uc *MediaUsecase) Upload(ctx context.Context, userID uint, name, suppliedM
 	}
 	media := &MediaFile{Name: name, CategoryID: categoryID, URL: stored.URL, Tag: strings.TrimPrefix(filepath.Ext(name), "."), Key: key, Size: stored.Size, Mime: suppliedMIME, MD5: hex.EncodeToString(hash.Sum(nil)), UserID: userID}
 	if save {
-		if err = uc.CreateMedia(ctx, media); err != nil {
-			_ = uc.files.Delete(ctx, key)
-			return nil, err
+		count, countErr := uc.MediaKeyReferences(ctx, key)
+		if countErr != nil {
+			return nil, countErr
+		}
+		if count == 0 {
+			if err = uc.CreateMedia(ctx, media); err != nil {
+				return nil, err
+			}
 		}
 	}
 	return media, nil