diff --git a/go.mod b/go.mod index cc44a66..a5f87b5 100644 --- a/go.mod +++ b/go.mod @@ -84,6 +84,7 @@ require ( github.com/google/pprof v0.0.0-20241009165004-a3522334989c // indirect github.com/gorilla/css v1.0.1 // indirect github.com/gorilla/securecookie v1.1.2 // indirect + github.com/h2non/filetype v1.1.3 // indirect github.com/hashicorp/errwrap v1.1.0 // indirect github.com/hashicorp/go-multierror v1.1.1 // indirect github.com/imroc/req/v3 v3.48.0 // indirect @@ -118,6 +119,7 @@ require ( github.com/quic-go/quic-go v0.48.0 // indirect github.com/refraction-networking/utls v1.6.7 // indirect github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect + github.com/rock-rabbit/rain v0.4.5 // indirect github.com/twitchyliquid64/golang-asm v0.15.1 // indirect github.com/ugorji/go/codec v1.2.12 // indirect github.com/xdg-go/pbkdf2 v1.0.0 // indirect @@ -136,6 +138,7 @@ require ( golang.org/x/net v0.30.0 // indirect golang.org/x/sync v0.8.0 // indirect golang.org/x/sys v0.26.0 // indirect + golang.org/x/time v0.7.0 // indirect golang.org/x/tools v0.26.0 // indirect google.golang.org/protobuf v1.35.1 // indirect gorm.io/datatypes v1.2.2 // indirect diff --git a/go.sum b/go.sum index adf628e..2e20c2d 100644 --- a/go.sum +++ b/go.sum @@ -174,6 +174,8 @@ github.com/gorilla/sessions v1.2.2 h1:lqzMYz6bOfvn2WriPUjNByzeXIlVzURcPmgMczkmTj github.com/gorilla/sessions v1.2.2/go.mod h1:ePLdVu+jbEgHH+KWw8I1z2wqd0BAdAQh/8LRvBeoNcQ= github.com/gorilla/sessions v1.4.0 h1:kpIYOp/oi6MG/p5PgxApU8srsSw9tuFbt46Lt7auzqQ= github.com/gorilla/sessions v1.4.0/go.mod h1:FLWm50oby91+hl7p/wRxDth9bWSuk0qVL2emc7lT5ik= +github.com/h2non/filetype v1.1.3 h1:FKkx9QbD7HR/zjK1Ia5XiBsq9zdLi5Kf3zGyFTAFkGg= +github.com/h2non/filetype v1.1.3/go.mod h1:319b3zT68BvV+WRj7cwy856M2ehB3HqNOt6sy1HndBY= github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4= github.com/hashicorp/errwrap v1.1.0 h1:OxrOeh75EUXMY8TBjag2fzXGZ40LB6IKw45YeGUDY2I= github.com/hashicorp/errwrap v1.1.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4= @@ -300,6 +302,8 @@ github.com/refraction-networking/utls v1.6.7 h1:zVJ7sP1dJx/WtVuITug3qYUq034cDq9B github.com/refraction-networking/utls v1.6.7/go.mod h1:BC3O4vQzye5hqpmDTWUqi4P5DDhzJfkV1tdqtawQIH0= github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE= github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo= +github.com/rock-rabbit/rain v0.4.5 h1:SNzv5oTN978STB477XQ/+3tctdqT+Cdf0pN+ru2mhM8= +github.com/rock-rabbit/rain v0.4.5/go.mod h1:3weEuxzk4ZxQZF158l2Vb+0Y+s2jGcIdjwJJFUvRB5c= github.com/rogpeppe/go-internal v1.12.0 h1:exVL4IDcn6na9z1rAb56Vxr+CgyK3nn3O+epU5NdKM8= github.com/rogpeppe/go-internal v1.12.0/go.mod h1:E+RYuTGaKKdloAfM02xzb0FW3Paa99yedzYV+kq4uf4= github.com/sirupsen/logrus v1.8.1/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0= @@ -470,6 +474,8 @@ golang.org/x/text v0.19.0 h1:kTxAhCbGbxhK0IwgSKiMO5awPoDQ0RpfiVYBfK860YM= golang.org/x/text v0.19.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY= golang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk= golang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM= +golang.org/x/time v0.7.0 h1:ntUhktv3OPE6TgYxXWv9vKvUSJyIFJlyohwbkEwPrKQ= +golang.org/x/time v0.7.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM= golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= golang.org/x/tools v0.0.0-20190425150028-36563e24a262/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q= golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= diff --git a/utils/generate_avatar.go b/utils/generate_avatar.go index 1c7010b..09a9836 100644 --- a/utils/generate_avatar.go +++ b/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 }