feat(baidu_netdisk): Add shard upload timeout setting (#1682)

add timeout

Author: jenfonro

drivers/baidu_netdisk/driver.go

@@ -58,8 +58,13 @@ func (d *BaiduNetdisk) GetAddition() driver.Additional {
 }
 
 func (d *BaiduNetdisk) Init(ctx context.Context) error {
+	timeout := DEFAULT_UPLOAD_SLICE_TIMEOUT
+	if d.UploadSliceTimeout > 0 {
+		timeout = time.Second * time.Duration(d.UploadSliceTimeout)
+	}
+
 	d.upClient = base.NewRestyClient().
-		SetTimeout(UPLOAD_TIMEOUT).
+		SetTimeout(timeout).
 		SetRetryCount(UPLOAD_RETRY_COUNT).
 		SetRetryWaitTime(UPLOAD_RETRY_WAIT_TIME).
 		SetRetryMaxWaitTime(UPLOAD_RETRY_MAX_WAIT_TIME)

drivers/baidu_netdisk/meta.go

@@ -19,6 +19,7 @@ type Addition struct {
 	AccessToken           string
 	RefreshToken          string `json:"refresh_token" required:"true"`
 	UploadThread          string `json:"upload_thread" default:"3" help:"1<=thread<=32"`
+	UploadSliceTimeout    int    `json:"upload_timeout" type:"number" default:"60" help:"per-slice upload timeout in seconds"`
 	UploadAPI             string `json:"upload_api" default:"https://d.pcs.baidu.com"`
 	UseDynamicUploadAPI   bool   `json:"use_dynamic_upload_api" default:"true" help:"dynamically get upload api domain, when enabled, the 'Upload API' setting will be used as a fallback if failed to get"`
 	CustomUploadPartSize  int64  `json:"custom_upload_part_size" type:"number" default:"0" help:"0 for auto"`
@@ -27,12 +28,12 @@ type Addition struct {
 }
 
 const (
-	UPLOAD_FALLBACK_API        = "https://d.pcs.baidu.com" // 备用上传地址
-	UPLOAD_URL_EXPIRE_TIME     = time.Minute * 60          // 上传地址有效期(分钟)
-	UPLOAD_TIMEOUT             = time.Minute * 30          // 上传请求超时时间
-	UPLOAD_RETRY_COUNT         = 3
-	UPLOAD_RETRY_WAIT_TIME     = time.Second * 1
-	UPLOAD_RETRY_MAX_WAIT_TIME = time.Second * 5
+	UPLOAD_FALLBACK_API          = "https://d.pcs.baidu.com" // 备用上传地址
+	UPLOAD_URL_EXPIRE_TIME       = time.Minute * 60          // 上传地址有效期(分钟)
+	DEFAULT_UPLOAD_SLICE_TIMEOUT = time.Second * 60          // 上传分片请求默认超时时间
+	UPLOAD_RETRY_COUNT           = 3
+	UPLOAD_RETRY_WAIT_TIME       = time.Second * 1
+	UPLOAD_RETRY_MAX_WAIT_TIME   = time.Second * 5
 )
 
 var config = driver.Config{