Upload 2 files

internal/utils/req_client.go

@@ -0,0 +1,44 @@
+package utils
+
+import (
+	"crypto/tls"
+	"fmt"
+	"github.com/go-resty/resty/v2"
+	"time"
+)
+
+var (
+	RestySSEClient = resty.New().
+			SetTimeout(3 * time.Minute).
+			SetTLSClientConfig(&tls.Config{InsecureSkipVerify: true}).
+			SetDoNotParseResponse(true). // 告诉 Resty，不要自动读取/解析 Body，让我们自己来处理流
+			SetHeaders(map[string]string{
+			"Content-Type":    "application/json",
+			"User-Agent":      "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
+			"x-client-locale": "zh_CN", // 可以不传，但默认会返回英文回答
+		}).
+		OnAfterResponse(func(c *resty.Client, resp *resty.Response) error {
+			// 如果不是 200，尝试把 body 打印出来
+			if resp.StatusCode() != 200 {
+				return fmt.Errorf("monica API error: status %d, body: %s",
+					resp.StatusCode(), resp.String())
+			}
+			return nil
+		})
+
+	RestyDefaultClient = resty.New().
+				SetTimeout(time.Second * 30).
+				SetTLSClientConfig(&tls.Config{InsecureSkipVerify: true}).
+				SetHeaders(map[string]string{
+			"Content-Type": "application/json",
+			"User-Agent":   "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36",
+		}).
+		OnAfterResponse(func(c *resty.Client, resp *resty.Response) error {
+			// 如果不是 200，尝试把 body 打印出来
+			if resp.StatusCode() != 200 {
+				return fmt.Errorf("monica API error: status %d, body: %s",
+					resp.StatusCode(), resp.String())
+			}
+			return nil
+		})
+)

internal/utils/string.go

@@ -0,0 +1,21 @@
+package utils
+
+import (
+	"math/rand"
+	"time"
+)
+
+var randSource = rand.New(rand.NewSource(time.Now().UnixNano()))
+
+// RandStringUsingMathRand 生成指定长度的随机字符串
+func RandStringUsingMathRand(n int) string {
+	var letters = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789")
+
+	// 创建一个长度为 n 的切片，用来存放随机字符
+	result := make([]rune, n)
+	for i := 0; i < n; i++ {
+		result[i] = letters[randSource.Intn(len(letters))]
+	}
+
+	return string(result)
+}