jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-06-07 02:27:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: create dir if not exist

Browse Source
This commit is contained in:
Simon Ding
2025-04-20 10:25:05 +08:00
parent 8f354fff41
commit ec3f9b2f96
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/storage/local.go
View File

@@ -6,6 +6,7 @@ import (
"io/ioutil" "io/ioutil"
"os" "os"
"path/filepath" "path/filepath"
"polaris/log"
"github.com/gabriel-vasile/mimetype" "github.com/gabriel-vasile/mimetype"
"github.com/pkg/errors" "github.com/pkg/errors"
@@ -30,6 +31,13 @@ func (l *LocalStorage) Copy(src, destDir string,walkFn WalkFn) error {
baseDest := filepath.Join(l.dir, destDir) baseDest := filepath.Join(l.dir, destDir)
uploadFunc := func(destPath string, destInfo fs.FileInfo, srcReader io.Reader, mimeType *mimetype.MIME) error { uploadFunc := func(destPath string, destInfo fs.FileInfo, srcReader io.Reader, mimeType *mimetype.MIME) error {
baseDir := filepath.Dir(destPath)
if _, err := os.Stat(baseDir); os.IsNotExist(err) {
if err := os.MkdirAll(baseDir, os.ModePerm); err != nil {
return errors.Wrapf(err, "create dir %s", baseDir)
}
log.Infof("create local dir %s", baseDir)
}
if writer, err := os.OpenFile(destPath, os.O_RDWR|os.O_CREATE|os.O_TRUNC, os.ModePerm); err != nil { if writer, err := os.OpenFile(destPath, os.O_RDWR|os.O_CREATE|os.O_TRUNC, os.ModePerm); err != nil {
return errors.Wrapf(err, "create file %s", destPath) return errors.Wrapf(err, "create file %s", destPath)
} else { } else {
@@ -40,6 +48,7 @@ func (l *LocalStorage) Copy(src, destDir string,walkFn WalkFn) error {
} }
} }
log.Infof("copy file %s to %s success", srcReader, destPath)
return nil return nil
} }
return b.Upload(baseDest, true, false, false, uploadFunc, func(s string) error { return b.Upload(baseDest, true, false, false, uploadFunc, func(s string) error {