✨ init user created with avatar

utils/generate_avatar.go

@@ -0,0 +1,54 @@
+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
+}