GithubBackup/SaveAny-Bot
1
0
Fork 0
mirror of https://github.com/krau/SaveAny-Bot.git synced 2026-08-28 11:46:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(webdav): use write stream to upload file

Browse Source
This commit is contained in:
krau
2024-11-09 13:18:22 +08:00
parent a91cce9680
commit 9e226ae592
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
storage/webdav/webdav.go
+7 -4
View File
@@ -4,6 +4,7 @@ import (
"context" "context"
"errors" "errors"
"os" "os"
"path"
"path/filepath" "path/filepath"
"strings" "strings"
"time" "time"
@@ -32,17 +33,19 @@ func (w *Webdav) Init() {
} }
func (w *Webdav) Save(ctx context.Context, filePath, storagePath string) error { func (w *Webdav) Save(ctx context.Context, filePath, storagePath string) error {
storagePath = basePath + "/" + storagePath storagePath = path.Join(basePath, storagePath)
if err := Client.MkdirAll(filepath.Dir(storagePath), os.ModePerm); err != nil { if err := Client.MkdirAll(filepath.Dir(storagePath), os.ModePerm); err != nil {
logger.L.Errorf("Failed to create directory %s: %v", filepath.Dir(storagePath), err) logger.L.Errorf("Failed to create directory %s: %v", filepath.Dir(storagePath), err)
return errors.New("webdav: failed to create directory") return errors.New("webdav: failed to create directory")
} }
fileBytes, err := os.ReadFile(filePath) file, err := os.Open(filePath)
if err != nil { if err != nil {
logger.L.Errorf("Failed to read file %s: %v", filePath, err) logger.L.Errorf("Failed to open file %s: %v", filePath, err)
return err return err
} }
if err := Client.Write(storagePath, fileBytes, os.ModePerm); err != nil { defer file.Close()
if err := Client.WriteStream(storagePath, file, os.ModePerm); err != nil {
logger.L.Errorf("Failed to write file %s: %v", storagePath, err) logger.L.Errorf("Failed to write file %s: %v", storagePath, err)
return errors.New("webdav: failed to write file") return errors.New("webdav: failed to write file")
} }