mirror of https://github.com/cloudreve/Cloudreve
Modify: create folder if not exist while importing files
parent
960c886496
commit
1afc750dae
|
@ -18,6 +18,7 @@ import (
|
||||||
"io"
|
"io"
|
||||||
"net/url"
|
"net/url"
|
||||||
"path"
|
"path"
|
||||||
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -42,10 +43,6 @@ type Driver struct {
|
||||||
HTTPClient request.Client
|
HTTPClient request.Client
|
||||||
}
|
}
|
||||||
|
|
||||||
func (handler Driver) List(ctx context.Context, path string, recursive bool) ([]response.Object, error) {
|
|
||||||
panic("implement me")
|
|
||||||
}
|
|
||||||
|
|
||||||
type key int
|
type key int
|
||||||
|
|
||||||
const (
|
const (
|
||||||
|
@ -103,6 +100,41 @@ func (handler *Driver) InitOSSClient() error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (handler Driver) List(ctx context.Context, path string, recursive bool) ([]response.Object, error) {
|
||||||
|
// 初始化客户端
|
||||||
|
if err := handler.InitOSSClient(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 列取文件
|
||||||
|
options := []oss.Option{
|
||||||
|
oss.Prefix(strings.TrimPrefix(path, "/")),
|
||||||
|
}
|
||||||
|
if recursive {
|
||||||
|
options = append(options, oss.Delimiter("/"))
|
||||||
|
}
|
||||||
|
listRes, err := handler.bucket.ListObjects(options...)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 处理列取结果
|
||||||
|
res := make([]response.Object, len(listRes.CommonPrefixes)+len(listRes.Objects))
|
||||||
|
// 处理目录
|
||||||
|
//for _,object := range listRes.CommonPrefixes{
|
||||||
|
// res = append(res,response.Object{
|
||||||
|
// Name: "",
|
||||||
|
// RelativePath: "",
|
||||||
|
// Source: "",
|
||||||
|
// Size: 0,
|
||||||
|
// IsDir: false,
|
||||||
|
// LastModify: time.Time{},
|
||||||
|
// })
|
||||||
|
//}
|
||||||
|
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
|
||||||
// Get 获取文件
|
// Get 获取文件
|
||||||
func (handler Driver) Get(ctx context.Context, path string) (response.RSCloser, error) {
|
func (handler Driver) Get(ctx context.Context, path string) (response.RSCloser, error) {
|
||||||
// 通过VersionID禁止缓存
|
// 通过VersionID禁止缓存
|
||||||
|
|
|
@ -146,13 +146,13 @@ func (job *ImportTask) Do() {
|
||||||
if parent, ok := pathCache[virtualPath]; ok {
|
if parent, ok := pathCache[virtualPath]; ok {
|
||||||
parentFolder = parent
|
parentFolder = parent
|
||||||
} else {
|
} else {
|
||||||
if exist, folder := fs.IsPathExist(virtualPath); exist {
|
folder, err := fs.CreateDirectory(context.Background(), virtualPath)
|
||||||
parentFolder = folder
|
if err != nil {
|
||||||
} else {
|
util.Log().Warning("导入任务无法创建用户目录[%s], %s",
|
||||||
util.Log().Warning("导入任务无法创插入文件[%s], 父目录不存在",
|
virtualPath, err)
|
||||||
object.RelativePath)
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
parentFolder = folder
|
||||||
}
|
}
|
||||||
|
|
||||||
// 插入文件记录
|
// 插入文件记录
|
||||||
|
|
Loading…
Reference in New Issue