fix(onedrive,ali): upload progress

AlistGo · Aug 31, 2022 · b73dce3 · b73dce3
1 parent 7ac1d14
commit b73dce3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/drivers/aliyundrive/driver.go b/drivers/aliyundrive/driver.go
@@ -256,7 +256,7 @@ func (d *AliDrive) Put(ctx context.Context, dstDir model.Obj, stream model.FileS
 		file.ReadCloser = tempFile
 	}
 
-	for _, partInfo := range resp.PartInfoList {
+	for i, partInfo := range resp.PartInfoList {
 		req, err := http.NewRequest("PUT", partInfo.UploadUrl, io.LimitReader(file, DEFAULT))
 		if err != nil {
 			return err
@@ -266,6 +266,7 @@ func (d *AliDrive) Put(ctx context.Context, dstDir model.Obj, stream model.FileS
 			return err
 		}
 		res.Body.Close()
+		up(i * 100 / count)
 	}
 	var resp2 base.Json
 	_, err, e = d.request("https://api.aliyundrive.com/v2/file/complete", http.MethodPost, func(req *resty.Request) {

diff --git a/drivers/onedrive/util.go b/drivers/onedrive/util.go
@@ -194,7 +194,7 @@ func (d *Onedrive) upBig(ctx context.Context, dstDir model.Obj, stream model.Fil
 			return errors.New(string(data))
 		}
 		res.Body.Close()
-		up(int(finish / stream.GetSize()))
+		up(int(finish * 100 / stream.GetSize()))
 	}
 	return nil
 }