在网络不稳定时会自动重试

Author: gaowanliang

Readme.md

@@ -8,7 +8,7 @@
 - 支持命令参数使用, 方便外部程序调用.
 - 支持自定义上传分块大小.
 - 支持多线程上传(多文件同时上传).
-- ~~支持根据文件大小动态调整重试次数~~
+- 支持根据文件大小动态调整重试次数
 - ~~支持跳过网盘中已存在的同名文件.~~
 - 支持通过Telegram Bot实时监控上传进度，方便使用全自动下载脚本时对上传的实时监控
 

api/restore/upload/onedriveRecoverableRestore.go

@@ -62,11 +62,19 @@ func (rs *RestoreService) recoverableUpload(userID string, bearerToken string, c
 
 		timeUnix = time.Now().UnixNano()
 		//3b. make a call to the upload url with the file part based on the offset. 使用基于偏移量的文件部分调用上载url。
-		resp, err := rs.uploadFilePart(uploadURL, filePath, bearerToken, *filePartInBytes, sOffset, isLastChunk)
-
+		var resp *http.Response
+		for errCount := 1; errCount < 10; errCount++ {
+			resp, err = rs.uploadFilePart(uploadURL, filePath, bearerToken, *filePartInBytes, sOffset, isLastChunk)
+			if err != nil {
+				sendMsg(fmt.Sprintf("向OneDrive账户 `%s` 上传 `%s` 时出现连接问题，正在重试，当前为第%d次重试", username, filePath, errCount))
+			} else {
+				break
+			}
+		}
 		if err != nil {
-			log.Panicf("Failed to Restore :%v", err)
+			log.Fatalf("Failed to Load Files from source :%v", err)
 		}
+
 		respMap := make(map[string]interface{})
 		err = json.NewDecoder(resp.Body).Decode(&respMap)
 		if err != nil {