fix(baidu): custom upload part size (close #5757)

AlistGo · May 2, 2024 · 4cbbda8 · 4cbbda8 · pongfcnkl · May 28, 2024
1 parent 7bf5014
commit 4cbbda8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/drivers/baidu_netdisk/meta.go b/drivers/baidu_netdisk/meta.go
@@ -8,15 +8,16 @@ import (
 type Addition struct {
 	RefreshToken string `json:"refresh_token" required:"true"`
 	driver.RootPath
-	OrderBy        string `json:"order_by" type:"select" options:"name,time,size" default:"name"`
-	OrderDirection string `json:"order_direction" type:"select" options:"asc,desc" default:"asc"`
-	DownloadAPI    string `json:"download_api" type:"select" options:"official,crack" default:"official"`
-	ClientID       string `json:"client_id" required:"true" default:"iYCeC9g08h5vuP9UqvPHKKSVrKFXGa1v"`
-	ClientSecret   string `json:"client_secret" required:"true" default:"jXiFMOPVPCWlO2M5CwWQzffpNPaGTRBG"`
-	CustomCrackUA  string `json:"custom_crack_ua" required:"true" default:"netdisk"`
-	AccessToken    string
-	UploadThread   string `json:"upload_thread" default:"3" help:"1<=thread<=32"`
-	UploadAPI      string `json:"upload_api" default:"https://d.pcs.baidu.com"`
+	OrderBy              string `json:"order_by" type:"select" options:"name,time,size" default:"name"`
+	OrderDirection       string `json:"order_direction" type:"select" options:"asc,desc" default:"asc"`
+	DownloadAPI          string `json:"download_api" type:"select" options:"official,crack" default:"official"`
+	ClientID             string `json:"client_id" required:"true" default:"iYCeC9g08h5vuP9UqvPHKKSVrKFXGa1v"`
+	ClientSecret         string `json:"client_secret" required:"true" default:"jXiFMOPVPCWlO2M5CwWQzffpNPaGTRBG"`
+	CustomCrackUA        string `json:"custom_crack_ua" required:"true" default:"netdisk"`
+	AccessToken          string
+	UploadThread         string `json:"upload_thread" default:"3" help:"1<=thread<=32"`
+	UploadAPI            string `json:"upload_api" default:"https://d.pcs.baidu.com"`
+	CustomUploadPartSize int64  `json:"custom_upload_part_size" default:"0" help:"0 for auto"`
 }
 
 var config = driver.Config{

diff --git a/drivers/baidu_netdisk/util.go b/drivers/baidu_netdisk/util.go
@@ -249,6 +249,9 @@ const (
 )
 
 func (d *BaiduNetdisk) getSliceSize() int64 {
+	if d.CustomUploadPartSize != 0 {
+		return d.CustomUploadPartSize
+	}
 	switch d.vipType {
 	case 1:
 		return VipSliceSize