feat: 优化文件上传逻辑，减少临时空间占用

1Panel-dev · Jun 19, 2024 · 572c3a3 · 572c3a3
1 parent b6d916d
commit 572c3a3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/backend/app/api/v1/file.go b/backend/app/api/v1/file.go
@@ -593,6 +593,10 @@ func (b *BaseApi) Size(c *gin.Context) {
 }
 
 func mergeChunks(fileName string, fileDir string, dstDir string, chunkCount int, overwrite bool) error {
+	defer func() {
+		_ = os.RemoveAll(fileDir)
+	}()
+
 	op := files.NewFileOp()
 	dstDir = strings.TrimSpace(dstDir)
 	mode, _ := files.GetParentMode(dstDir)
@@ -629,9 +633,10 @@ func mergeChunks(fileName string, fileDir string, dstDir string, chunkCount int,
 		if err != nil {
 			return err
 		}
+		_ = os.Remove(chunkPath)
 	}
 
-	return files.NewFileOp().DeleteDir(fileDir)
+	return nil
 }
 
 // @Tags File