Clone/aliyunpan
1
0
Fork 0
mirror of https://github.com/tickstep/aliyunpan.git synced 2025-01-23 22:42:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add proof code for upload to support rapid upload

Browse Source
This commit is contained in:
tickstep 2021-10-24 22:03:47 +08:00
parent 60f3e1a0a4
commit 971369e968
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
internal/functions/panupload/upload_task_unit.go
View File

@ -15,6 +15,7 @@ package panupload
import ( import (
"fmt" "fmt"
"os"
"path" "path"
"path/filepath" "path/filepath"
"strings" "strings"
@ -126,7 +127,7 @@ func (utu *UploadTaskUnit) rapidUpload() (isContinue bool, result *taskframework
result = &taskframework.TaskUnitRunResult{} result = &taskframework.TaskUnitRunResult{}
fmt.Printf("[%s] 检测秒传中, 请稍候...\n", utu.taskInfo.Id()) fmt.Printf("[%s] 检测秒传中, 请稍候...\n", utu.taskInfo.Id())
if utu.LocalFileChecksum.UploadOpEntity.RapidUpload { if utu.LocalFileChecksum.UploadOpEntity.RapidUpload {
fmt.Printf("[%s] 秒传成功, 保存到网盘路径: %s\n\n", utu.taskInfo.Id(), utu.SavePath) fmt.Printf("[%s] 秒传成功, 保存到网盘路径: %s\n", utu.taskInfo.Id(), utu.SavePath)
result.Succeed = true result.Succeed = true
return false, result return false, result
} else { } else {
@ -284,7 +285,7 @@ func (utu *UploadTaskUnit) Run() (result *taskframework.TaskUnitRunResult) {
} else { } else {
msg = result.ResultMessage msg = result.ResultMessage
} }
fmt.Printf("%s [%s] 文件上传结果：%s 耗时 %s\n", time.Now().Format("2006-01-02 15:04:06"), utu.taskInfo.Id(), msg, time.Now().Sub(timeStart)) fmt.Printf("%s [%s] 文件上传结果：%s 耗时 %.2fs\n", time.Now().Format("2006-01-02 15:04:06"), utu.taskInfo.Id(), msg, time.Now().Sub(timeStart).Seconds())
}() }()
// 准备文件 // 准备文件
utu.prepareFile() utu.prepareFile()
@ -296,6 +297,9 @@ func (utu *UploadTaskUnit) Run() (result *taskframework.TaskUnitRunResult) {
var saveFilePath string var saveFilePath string
var testFileMeta = &UploadedFileMeta{} var testFileMeta = &UploadedFileMeta{}
var uploadOpEntity *aliyunpan.CreateFileUploadResult var uploadOpEntity *aliyunpan.CreateFileUploadResult
var proofCode = ""
var localFileInfo os.FileInfo
var localFile *os.File
switch utu.Step { switch utu.Step {
case StepUploadPrepareUpload: case StepUploadPrepareUpload:
@ -378,6 +382,12 @@ StepUploadPrepareUpload:
sha1Str = aliyunpan.DefaultZeroSizeFileContentHash sha1Str = aliyunpan.DefaultZeroSizeFileContentHash
} }
// proof code
localFile, _ = os.Open(utu.LocalFileChecksum.Path)
localFileInfo,_ = localFile.Stat()
proofCode = aliyunpan.CalcProofCode(utu.PanClient.GetAccessToken(), rio.NewFileReaderAtLen64(localFile), localFileInfo.Size())
localFile.Close()
appCreateUploadFileParam = &aliyunpan.CreateFileUploadParam{ appCreateUploadFileParam = &aliyunpan.CreateFileUploadParam{
DriveId: utu.DriveId, DriveId: utu.DriveId,
Name: filepath.Base(utu.LocalFileChecksum.Path), Name: filepath.Base(utu.LocalFileChecksum.Path),
@ -385,6 +395,8 @@ StepUploadPrepareUpload:
ContentHash: sha1Str, ContentHash: sha1Str,
ParentFileId: rs.FileId, ParentFileId: rs.FileId,
BlockSize: utu.BlockSize, BlockSize: utu.BlockSize,
ProofCode: proofCode,
ProofVersion: "v1",
} }
uploadOpEntity, apierr = utu.PanClient.CreateUploadFile(appCreateUploadFileParam) uploadOpEntity, apierr = utu.PanClient.CreateUploadFile(appCreateUploadFileParam)