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

fix: improve max storage size handling in CreateUploadTask by using GetFileSize for better error management

Browse Source
This commit is contained in:
keven1024
2025-06-22 00:48:53 +08:00
parent 1ac21b3dd0
commit f4740f4373
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/internal/controllers/file.go
View File

@@ -14,7 +14,6 @@ import (
"github.com/hibiken/asynq"
"github.com/labstack/echo/v4"
"github.com/spf13/cast"
)
func CreateUploadTask(c echo.Context) error {
@@ -41,8 +40,10 @@ func CreateUploadTask(c echo.Context) error {
"chunk_size": fileInfo.ChunkSize,
})
}
maxStorageSize := cast.ToInt64(utils.GetEnv("MAX_LOCALSTORAGE_SIZE"))
maxStorageSize, err := utils.GetFileSize(utils.GetEnv("MAX_LOCALSTORAGE_SIZE"))
if err != nil {
return utils.HTTPErrorHandler(c, err)
}
fileInfoMap, err := models.GetRedisFileInfoAll()
if err != nil {
return utils.HTTPErrorHandler(c, err)
@@ -56,7 +57,7 @@ func CreateUploadTask(c echo.Context) error {
}
totalSize += fileInfo.FileSize
}
if totalSize+r.FileSize > maxStorageSize {
if totalSize+r.FileSize > int64(maxStorageSize) {
return utils.HTTPErrorHandler(c, errors.New("存储空间不足"))
}