✨ update generate_avatar.go

utils/generate_avatar.go

@@ -5,54 +5,99 @@ import (
 	"io"
 	"net/http"
 	"schisandra-cloud-album/global"
+	"strconv"
 	"time"
 )
 
-// 获取网络图片
-var client = &http.Client{
-	Timeout: 5 * time.Second, // 超时时间
-}
+const (
+	numParts = 4 // 分成4块
+)
 
-// GenerateAvatar 用于生成用户头像
 func GenerateAvatar(userId string) (baseImg string) {
-
 	path := "https://api.multiavatar.com/" + userId + ".png"
 
-	// 创建请求
-	request, err := http.NewRequest("GET", path, nil)
+	// 创建 HTTP 客户端并设置超时时间
+	client := &http.Client{
+		Timeout: 10 * time.Second,
+	}
+
+	// 发送 HEAD 请求获取图片大小
+	headReq, err := http.NewRequest("HEAD", path, nil)
 	if err != nil {
 		global.LOG.Error(err)
 		return ""
 	}
 
-	// 发送请求并获取响应
-	respImg, err := client.Do(request)
+	respHead, err := client.Do(headReq)
 	if err != nil {
 		global.LOG.Error(err)
 		return ""
 	}
-	defer func(Body io.ReadCloser) {
-		err = Body.Close()
-		if err != nil {
-			global.LOG.Error(err)
-			return
-		}
-	}(respImg.Body)
+	defer respHead.Body.Close()
 
-	// 读取图片数据
-	imgByte, err := io.ReadAll(respImg.Body)
-	if err != nil {
-		global.LOG.Error(err)
+	// 获取图片大小
+	contentLength := respHead.ContentLength
+	if contentLength <= 0 {
 		return ""
 	}
 
+	imgChunks := make([][]byte, numParts) // 存储每个部分的图片数据
+
+	// 启动多个 goroutine 下载分块
+	for i := 0; i < numParts; i++ {
+		wg.Add(1)
+		go func(part int) {
+			defer wg.Done()
+			start := (contentLength / int64(numParts)) * int64(part)
+			end := start + (contentLength / int64(numParts)) - 1
+			if part == numParts-1 {
+				end = contentLength - 1 // 最后一部分下载到文件末尾
+			}
+
+			// 创建 RANGE 请求
+			rangeHeader := "bytes=" + strconv.FormatInt(start, 10) + "-" + strconv.FormatInt(end, 10)
+			request, err := http.NewRequest("GET", path, nil)
+			if err != nil {
+				global.LOG.Error(err)
+				return
+			}
+			request.Header.Set("Range", rangeHeader)
+
+			respImg, err := client.Do(request)
+			if err != nil {
+				global.LOG.Error(err)
+				return
+			}
+			defer respImg.Body.Close()
+
+			// 读取图片数据
+			imgByte, err := io.ReadAll(respImg.Body)
+			if err != nil {
+				global.LOG.Error(err)
+				return
+			}
+
+			// 存储分块
+			imgChunks[part] = imgByte
+		}(i)
+	}
+
+	wg.Wait() // 等待所有 goroutine 完成
+
+	// 合并所有部分
+	var fullImg []byte
+	for _, chunk := range imgChunks {
+		fullImg = append(fullImg, chunk...)
+	}
+
 	// 判断文件类型，生成一个前缀
-	mimeType := http.DetectContentType(imgByte)
+	mimeType := http.DetectContentType(fullImg)
 	switch mimeType {
 	case "image/png":
-		baseImg = "data:image/png;base64," + base64.StdEncoding.EncodeToString(imgByte)
+		baseImg = "data:image/png;base64," + base64.StdEncoding.EncodeToString(fullImg)
 	default:
 		return ""
 	}
+
 	return baseImg
 }