parent
841832bb65
commit
99e7eecab7
@ -0,0 +1,10 @@
|
|||||||
|
package filesystem
|
||||||
|
|
||||||
|
import "errors"
|
||||||
|
|
||||||
|
var (
|
||||||
|
UnknownPolicyTypeError = errors.New("未知存储策略类型")
|
||||||
|
FileSizeTooBigError = errors.New("单个文件尺寸太大")
|
||||||
|
FileExtensionNotAllowedError = errors.New("不允许上传此类型的文件")
|
||||||
|
InsufficientCapacityError = errors.New("容量空间不足")
|
||||||
|
)
|
@ -0,0 +1,37 @@
|
|||||||
|
package local
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"github.com/HFO4/cloudreve/pkg/util"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Handler struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
// Put 将文件流保存到指定目录
|
||||||
|
func (handler Handler) Put(ctx context.Context, file io.ReadCloser, dst string) error {
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
// 如果目标目录不存在,创建
|
||||||
|
basePath := filepath.Dir(dst)
|
||||||
|
if !util.Exists(basePath) {
|
||||||
|
fmt.Println("创建", basePath)
|
||||||
|
err := os.MkdirAll(basePath, 0666)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
out, err := os.Create(dst)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer out.Close()
|
||||||
|
|
||||||
|
_, err = io.Copy(out, file)
|
||||||
|
return err
|
||||||
|
}
|
@ -1 +0,0 @@
|
|||||||
package local
|
|
Loading…
Reference in new issue