license add mutex

api/restore/upload/onedriveRecoverableRestore.go

@@ -75,9 +75,9 @@ func (rs *RestoreService) recoverableUpload(userID string, bearerToken string, c
 		//3b. make a call to the upload url with the file part based on the offset. 使用基于偏移量的文件部分调用上载url。
 		var resp *http.Response
 		for errCount := 1; errCount < 10; errCount++ {
-			bearerToken = http2.GetBearer() //解决长时下载时，Bearer超时的问题，这里采用超时一次就重新获取一次token的方案
 			resp, err = rs.uploadFilePart(uploadURL, filePath, bearerToken, *filePartInBytes, sOffset, isLastChunk)
 			if err != nil {
+				bearerToken = http2.GetBearer() //解决长时上传时，Bearer超时的问题，这里采用超时一次就重新获取一次token的方案
 				sendMsg("close|" + fmt.Sprintf(locText("failToLink"), username, filePath, errCount))
 				// close 用作输出时定位，带有 close 在输出时不会被刷新走
 				// close= 表示文件传输结束，此时会同步删除tg发出的消息

graph/net/http/OAuth2.go

@@ -3,14 +3,14 @@ package http
 import (
 	"encoding/json"
 	"fmt"
+	"github.com/buger/jsonparser"
 	"io/ioutil"
 	"log"
 	"net/http"
 	"os"
 	"regexp"
 	"strings"
-
-	"github.com/buger/jsonparser"
+	"sync"
 )
 
 type Certificate struct {
@@ -30,6 +30,7 @@ var GraphURL = "https://graph.microsoft.com/v1.0/me/"
 var TokenURL = "https://login.microsoftonline.com/common/oauth2/v2.0/token"
 var Host = "login.microsoftonline.com"
 var isCN = false
+var mutex sync.Mutex
 
 func ChangeCNURL() {
 	GraphURL = "https://microsoftgraph.chinacloudapi.cn/v1.0/me/"
@@ -200,6 +201,7 @@ func getAccessToken(oauth2URL string, ms int, lang string) string {
 }
 
 func refreshAccessToken(path string, Thread int, BlockSize int, Language string, TimeOut int, BotKey string, UserID string) string {
+	mutex.Lock() //使用互斥锁防止线程数过高时信息被覆盖问题
 	filePtr, err := os.Open(path)
 	if err != nil {
 		log.Panicln(err)
@@ -260,6 +262,7 @@ func refreshAccessToken(path string, Thread int, BlockSize int, Language string,
 	// 创建Json编码器
 	encoder := json.NewEncoder(filePtr)
 	err = encoder.Encode(info)
+	defer mutex.Unlock()
 	if err != nil {
 		log.Panicln(err.Error())
 	}