nyne/nysoure
1
0
Fork 0
mirror of https://github.com/wgh136/nysoure.git synced 2025-09-27 12:17:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor downloadFile to use buffered writer for improved performance

Browse Source
This commit is contained in:
nyne
2025-09-03 09:13:57 +08:00
parent a3fc1cd801
commit 4e709dd952
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
server/service/file.go
View File

@@ -1,6 +1,7 @@
package service
import (
"bufio"
"context"
"crypto/md5"
"encoding/hex"
@@ -521,6 +522,7 @@ func downloadFile(ctx context.Context, url string, path string) error {
return model.NewInternalServerError("failed to open file for writing")
}
defer file.Close()
writer := bufio.NewWriter(file)
buf := make([]byte, 64*1024)
for {
@@ -530,12 +532,15 @@ func downloadFile(ctx context.Context, url string, path string) error {
default:
n, readErr := resp.Body.Read(buf)
if n > 0 {
if _, writeErr := file.Write(buf[:n]); writeErr != nil {
if _, writeErr := writer.Write(buf[:n]); writeErr != nil {
return model.NewInternalServerError("failed to write to file")
}
}
if readErr != nil {
if readErr == io.EOF {
if err := writer.Flush(); err != nil {
return model.NewInternalServerError("failed to flush writer")
}
return nil // Download completed successfully
}
if ctx.Err() != nil {