59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
package utils
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"io"
|
|
"net/http"
|
|
"schisandra-cloud-album/global"
|
|
"time"
|
|
)
|
|
|
|
// 获取网络图片
|
|
var client = &http.Client{
|
|
Timeout: 5 * time.Second, // 超时时间
|
|
}
|
|
|
|
// GenerateAvatar 用于生成用户头像
|
|
func GenerateAvatar(userId string) (baseImg string) {
|
|
|
|
path := "https://api.multiavatar.com/" + userId + ".png"
|
|
|
|
// 创建请求
|
|
request, err := http.NewRequest("GET", path, nil)
|
|
if err != nil {
|
|
global.LOG.Error(err)
|
|
return ""
|
|
}
|
|
|
|
// 发送请求并获取响应
|
|
respImg, err := client.Do(request)
|
|
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)
|
|
|
|
// 读取图片数据
|
|
imgByte, err := io.ReadAll(respImg.Body)
|
|
if err != nil {
|
|
global.LOG.Error(err)
|
|
return ""
|
|
}
|
|
|
|
// 判断文件类型，生成一个前缀
|
|
mimeType := http.DetectContentType(imgByte)
|
|
switch mimeType {
|
|
case "image/png":
|
|
baseImg = "data:image/png;base64," + base64.StdEncoding.EncodeToString(imgByte)
|
|
default:
|
|
return ""
|
|
}
|
|
return baseImg
|
|
}
|