GitHub/015
1
0
Fork 0
mirror of https://github.com/keven1024/015.git synced 2026-05-26 15:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(worker): ensure temporary files are removed on error during image compression and conversion

Browse Source
This commit is contained in:
keven1024
2026-02-27 18:47:16 +08:00
parent 123178084a
commit ceb2d026d2
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
worker/internal/tasks/image.go
View File

@@ -6,6 +6,7 @@ import (
"errors" "errors"
"fmt" "fmt"
"mime" "mime"
"os"
"path/filepath" "path/filepath"
"pkg/models" "pkg/models"
"pkg/utils" "pkg/utils"
@@ -37,6 +38,7 @@ func CompressImage(ctx context.Context, task *asynq.Task) error {
} }
compressedFileInfo, err := services.GenStandardFile(compressedPath, originalFileInfo.MimeType) compressedFileInfo, err := services.GenStandardFile(compressedPath, originalFileInfo.MimeType)
if err != nil { if err != nil {
defer os.Remove(compressedPath)
return err return err
} }
@@ -80,12 +82,14 @@ func ConvertImage(ctx context.Context, task *asynq.Task) error {
} }
return err return err
} }
mimeType := mime.TypeByExtension(payload.TargetExt) mimeType := mime.TypeByExtension(fmt.Sprintf(".%s", payload.TargetExt))
if mimeType == "" { if mimeType == "" {
defer os.Remove(convertedPath)
return ErrUnknown return ErrUnknown
} }
convertedFileInfo, err := services.GenStandardFile(convertedPath, mimeType) convertedFileInfo, err := services.GenStandardFile(convertedPath, mimeType)
if err != nil { if err != nil {
defer os.Remove(convertedPath)
return err return err
} }