ProjectHentai/alist
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(aliyundrive): make checksum cancellable (#3814)

Browse Source
This commit is contained in:
NewbieOrange 2023-03-12 13:59:40 +08:00 committed by GitHub
parent 4d9a29bddd
commit adc3a56552
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
drivers/aliyundrive/driver.go
View File

@ -226,7 +226,7 @@ func (d *AliDrive) Put(ctx context.Context, dstDir model.Obj, stream model.FileS
delete(reqBody, "pre_hash") delete(reqBody, "pre_hash")
h := sha1.New() h := sha1.New()
if localFile != nil { if localFile != nil {
if _, err = io.Copy(h, localFile); err != nil { if err = utils.CopyWithCtx(ctx, h, localFile, 0, nil); err != nil {
return err return err
} }
if _, err = localFile.Seek(0, io.SeekStart); err != nil { if _, err = localFile.Seek(0, io.SeekStart); err != nil {
@ -241,7 +241,7 @@ func (d *AliDrive) Put(ctx context.Context, dstDir model.Obj, stream model.FileS
_ = tempFile.Close() _ = tempFile.Close()
_ = os.Remove(tempFile.Name()) _ = os.Remove(tempFile.Name())
}() }()
if _, err = io.Copy(io.MultiWriter(tempFile, h), file); err != nil { if err = utils.CopyWithCtx(ctx, io.MultiWriter(tempFile, h), file, 0, nil); err != nil {
return err return err
} }
localFile = tempFile localFile = tempFile