55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
package utils
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"errors"
|
|
"io"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
// 获取网络图片
|
|
var client = &http.Client{
|
|
Timeout: 5 * time.Second, // 超时时间
|
|
}
|
|
|
|
// GenerateAvatar 用于生成用户头像
|
|
func GenerateAvatar(userId string) (baseImg string, err error) {
|
|
|
|
path := "https://api.multiavatar.com/" + userId + ".png"
|
|
|
|
// 创建请求
|
|
request, err := http.NewRequest("GET", path, nil)
|
|
if err != nil {
|
|
return "", errors.New("image request error")
|
|
}
|
|
|
|
// 发送请求并获取响应
|
|
respImg, err := client.Do(request)
|
|
if err != nil {
|
|
return "", errors.New("failed to fetch image")
|
|
}
|
|
defer func(Body io.ReadCloser) {
|
|
err := Body.Close()
|
|
if err != nil {
|
|
return
|
|
}
|
|
}(respImg.Body)
|
|
|
|
// 读取图片数据
|
|
imgByte, err := io.ReadAll(respImg.Body)
|
|
if err != nil {
|
|
return "", errors.New("failed to read image data")
|
|
}
|
|
|
|
// 判断文件类型，生成一个前缀
|
|
mimeType := http.DetectContentType(imgByte)
|
|
switch mimeType {
|
|
case "image/png":
|
|
baseImg = "data:image/png;base64," + base64.StdEncoding.EncodeToString(imgByte)
|
|
default:
|
|
return "", errors.New("unsupported image type")
|
|
}
|
|
return baseImg, nil
|
|
}
|