github
/
alist
mirror of https://github.com/Xhofe/alist
1
0
Fork 0
Code Issues Releases Wiki Activity

feat(fs): add put return after finished

refactor/fs
Noah Hsu 2022-06-24 14:21:28 +08:00
parent 956a5ae906
commit 3f49271db6
2 changed files with 24 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
internal/fs/fs.go
View File

@ -77,8 +77,16 @@ func Remove(ctx context.Context, path string) error {
return err
}
func Put(ctx context.Context, dstDirPath string, file model.FileStreamer) error {
err := put(ctx, dstDirPath, file)
func PutDirectly(ctx context.Context, dstDirPath string, file model.FileStreamer) error {
err := putDirectly(ctx, dstDirPath, file)
if err != nil {
log.Errorf("failed put %s: %+v", dstDirPath, err)
}
return err
}
func PutAsTask(dstDirPath string, file model.FileStreamer) error {
err := putAsTask(dstDirPath, file)
if err != nil {
log.Errorf("failed put %s: %+v", dstDirPath, err)
}

16
internal/fs/put.go
View File

@ -15,8 +15,8 @@ var UploadTaskManager = task.NewTaskManager[uint64](3, func(tid *uint64) {
atomic.AddUint64(tid, 1)
})
// Put add as a put task
func put(ctx context.Context, dstDirPath string, file model.FileStreamer) error {
// putAsTask add as a put task and return immediately
func putAsTask(dstDirPath string, file model.FileStreamer) error {
account, dstDirActualPath, err := operations.GetAccountAndActualPath(dstDirPath)
if account.Config().NoUpload {
return errors.WithStack(errs.UploadNotSupported)
@ -32,3 +32,15 @@ func put(ctx context.Context, dstDirPath string, file model.FileStreamer) error
}))
return nil
}
// putDirect put the file and return after finish
func putDirectly(ctx context.Context, dstDirPath string, file model.FileStreamer) error {
account, dstDirActualPath, err := operations.GetAccountAndActualPath(dstDirPath)
if account.Config().NoUpload {
return errors.WithStack(errs.UploadNotSupported)
}
if err != nil {
return errors.WithMessage(err, "failed get account")
}
return operations.Put(ctx, account, dstDirActualPath, file, nil)
}