优化结构

internal/biz/system/media_upload.go

@@ -97,7 +97,7 @@ func (uc *MediaUsecase) InitUpload(ctx context.Context, userID uint, name, hash
 	completed, err := uc.FindCompletedSession(ctx, userID, strings.ToLower(hash))
 	if err == nil && completed.MediaID != 0 {
 		if media, findErr := uc.FindMedia(ctx, completed.MediaID); findErr == nil {
-			copy := &MediaFile{Name: name, URL: media.URL, Tag: strings.TrimPrefix(filepath.Ext(name), "."), Key: media.Key, Size: media.Size, MD5: media.MD5, Mime: media.Mime, UserID: userID}
+			copy := &MediaFile{Name: name, CategoryID: media.CategoryID, URL: media.URL, Tag: strings.TrimPrefix(filepath.Ext(name), "."), Key: media.Key, Size: media.Size, MD5: media.MD5, Mime: media.Mime, UserID: userID}
 			if strings.TrimSpace(copy.Mime) == "" {
 				copy.Mime = stdmime.TypeByExtension(strings.ToLower(filepath.Ext(name)))
 				if copy.Mime == "" {
@@ -294,8 +294,10 @@ func (uc *MediaUsecase) CleanupStale(ctx context.Context, ttlHours int) error {
 	// scheduler. Keep the endpoint-independent background behavior compatible.
 	ids, _ := uc.StaleUploadSessionIDs(ctx, time.Now().Add(-time.Duration(ttlHours)*time.Hour))
 	for _, id := range ids {
+		if err := uc.files.DeletePrefix(ctx, uc.chunkPrefix(id)); err != nil {
+			continue
+		}
 		_ = uc.DeleteUploadData(ctx, id)
-		_ = uc.files.DeletePrefix(ctx, uc.chunkPrefix(id))
 	}
 	return nil
 }