From 70fc2be02f11e67c4b4a07a009b326b34ce2a9a4 Mon Sep 17 00:00:00 2001 From: keven1024 Date: Fri, 27 Feb 2026 17:06:44 +0800 Subject: [PATCH] feat(worker): enhance GenStandardFile to set file expiration and store expiration info in Redis --- worker/internal/services/file.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/worker/internal/services/file.go b/worker/internal/services/file.go index 9fa01bf..9eb4f04 100644 --- a/worker/internal/services/file.go +++ b/worker/internal/services/file.go @@ -48,6 +48,11 @@ func GenStandardFile(filePath string, mimeType string) (GenStandardFileReturn, e if err := os.Rename(filePath, newPath); err != nil { return GenStandardFileReturn{}, err } + expire := cast.ToInt64(u.GetEnvWithDefault("upload.remove_expire", "2")) * 3600 + err = services.SetFileRemoveTask(fileId, time.Duration(expire)*time.Second) + if err != nil { + return GenStandardFileReturn{}, err + } models.SetRedisFileInfo(fileId, models.RedisFileInfo{ FileInfo: models.FileInfo{ FileSize: fileSize, @@ -56,12 +61,8 @@ func GenStandardFile(filePath string, mimeType string) (GenStandardFileReturn, e }, FileType: models.FileTypeUpload, CreatedAt: time.Now().Unix(), + Expire: expire, }) - expire := cast.ToInt64(u.GetEnvWithDefault("upload.remove_expire", "2")) * 3600 - err = services.SetFileRemoveTask(fileId, time.Duration(expire)*time.Second) - if err != nil { - return GenStandardFileReturn{}, err - } return GenStandardFileReturn{ FileId: fileId, FileInfo: models.FileInfo{