From 2348039c169d81ce6f1123642b32eb291bc177b1 Mon Sep 17 00:00:00 2001
From: fumiama <s862105088@gmail.com>
Date: Sun, 2 Jan 2022 15:06:35 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=8F=EF=B8=8F=20=20=E5=8A=A0=E5=BF=AB=20re?=
 =?UTF-8?q?g=20=E5=A4=84=E7=90=86=E9=80=9F=E5=BA=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 utils/file/updater.go | 48 +++++++++++++++++++++++++++++++++----------
 1 file changed, 37 insertions(+), 11 deletions(-)

diff --git a/utils/file/updater.go b/utils/file/updater.go
index 40f45aa1..a064dc0b 100644
--- a/utils/file/updater.go
+++ b/utils/file/updater.go
@@ -8,6 +8,7 @@ import (
 	"net/http"
 	"os"
 	"sync"
+	"time"
 	"unsafe"
 
 	reg "github.com/fumiama/go-registry"
@@ -19,8 +20,11 @@ const (
 )
 
 var (
-	registry = reg.NewRegReader("reilia.eastasia.azurecontainer.io:32664", "fumiama")
-	lzmu     sync.Mutex
+	registry      = reg.NewRegReader("reilia.eastasia.azurecontainer.io:32664", "fumiama")
+	connmu, getmu sync.Mutex
+	processes     sync.WaitGroup
+	connerr       error
+	isconnected   bool
 )
 
 // GetLazyData 获取懒加载数据
@@ -32,24 +36,46 @@ func GetLazyData(path string, isReturnDataBytes, isDataMustEqual bool) ([]byte,
 	var resp *http.Response
 	var filemd5 *[16]byte
 	var ms string
+	var err error
 
 	u := dataurl + path[5:]
-	lzmu.Lock()
-	logrus.Infoln("[file]检查懒加载文件:", path)
-	err := registry.Connect()
-	if err != nil {
-		logrus.Errorln("[file]无法连接到md5验证服务器，请自行确保下载文件的正确性:", err)
+
+	if !isconnected {
+		connmu.Lock()
+		if !isconnected && connerr == nil {
+			logrus.Infoln("[file]连接md5验证服务器...")
+			connerr = registry.ConnectIn(time.Second * 4)
+			if connerr == nil {
+				isconnected = true
+				processes.Add(1)
+				go func() {
+					processes.Wait()
+					_ = registry.Close()
+					isconnected = false
+				}()
+			} else {
+				logrus.Warnln("[file]连接md5验证服务器失败:", connerr)
+			}
+		}
+		connmu.Unlock()
 	} else {
+		processes.Add(1)
+	}
+
+	if connerr != nil {
+		logrus.Warnln("[file]无法连接到md5验证服务器，请自行确保下载文件", path, "的正确性")
+	} else {
+		getmu.Lock()
 		ms, err = registry.Get(path)
+		getmu.Unlock()
+		processes.Done()
 		if err != nil || len(ms) != 16 {
-			logrus.Errorln("[file]获取md5失败，请自行确保下载文件", path, "的正确性:", err)
+			logrus.Warnln("[file]获取md5失败，请自行确保下载文件", path, "的正确性:", err)
 		} else {
 			filemd5 = (*[16]byte)(*(*unsafe.Pointer)(unsafe.Pointer(&ms)))
-			logrus.Infoln("[file]从验证服务器获得文件md5:", hex.EncodeToString(filemd5[:]))
+			logrus.Infoln("[file]从验证服务器获得文件", path, "md5:", hex.EncodeToString(filemd5[:]))
 		}
 	}
-	_ = registry.Close()
-	lzmu.Unlock()
 
 	if IsExist(path) {
 		data, err = os.ReadFile(path)