优化kit/zip

Signed-off-by: lixxxww <941403820@qq.com>
This commit is contained in:
lixxxww
2024-01-23 11:31:44 +00:00
committed by Gitee
parent e51f21edee
commit 731f8224b0
2 changed files with 106 additions and 0 deletions

45
kit/file/file.go Normal file
View File

@@ -0,0 +1,45 @@
package utilFile
import (
"io"
"mime/multipart"
"net/http"
"os"
)
// DownloadFile 会将url下载到本地文件它会在下载时写入而不是将整个文件加载到内存中。
func DownloadFile(url, filepath string) error {
// Get the data
resp, err := http.Get(url)
if err != nil {
return err
}
defer resp.Body.Close()
// Create the file
out, err := os.Create(filepath)
if err != nil {
return err
}
defer out.Close()
// Write the body to file
_, err = io.Copy(out, resp.Body)
return err
}
func SaveUploadedFile(file *multipart.FileHeader, dst string) error {
src, err := file.Open()
if err != nil {
return err
}
defer src.Close()
out, err := os.Create(dst)
if err != nil {
return err
}
defer out.Close()
_, err = io.Copy(out, src)
return err
}