生成表结构、

utility/wechat/wechat.go

@@ -0,0 +1,165 @@
+package wechat
+
+import (
+	"context"
+	"github.com/go-resty/resty/v2"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/glog"
+	"io"
+	"os"
+	"server/utility/ecode"
+	"sync"
+	"time"
+)
+
+type weChatClient struct {
+	AppId        string
+	AppSecret    string
+	TicketExpire int
+	Token        string
+	accessToken  string
+	expiresIn    int
+	lastUpdated  time.Time
+	mu           sync.RWMutex
+}
+
+var (
+	instance *weChatClient
+	once     sync.Once
+)
+
+func GetWeChatClient() *weChatClient {
+	return instance
+}
+func init() {
+	once.Do(func() {
+		ctx := context.Background()
+		instance = &weChatClient{
+			AppId:        g.Config().MustGet(ctx, "wechat.appId").String(),
+			AppSecret:    g.Config().MustGet(ctx, "wechat.appSecret").String(),
+			TicketExpire: g.Config().MustGet(ctx, "wechat.ticketExpire").Int(),
+		}
+
+		go instance.autoRefreshToken(ctx)
+	})
+}
+
+func (c *weChatClient) getAccessToken() error {
+	result := struct {
+		AccessToken string `json:"access_token"`
+		ExpiresIn   int    `json:"expires_in"`
+	}{}
+	ctx := context.Background()
+	resp, err := resty.New().R().
+		SetQueryParams(g.MapStrStr{
+			"grant_type": "client_credential",
+			"appid":      c.AppId,
+			"secret":     c.AppSecret,
+		}).
+		SetResult(&result).
+		Get("https://api.weixin.qq.com/cgi-bin/token")
+
+	if err != nil {
+		glog.Errorf(ctx, "发起 get access_token 请求出现异常: %+v", err)
+		return ecode.Fail.Sub("发起 get access_token 请求出现异常")
+	}
+	if resp.StatusCode() != 200 {
+		glog.Errorf(ctx, "获取微信 access_token 响应异常: %+v", resp.Status())
+		return ecode.Fail.Sub("获取微信 access_token 失败")
+	}
+
+	c.mu.Lock()
+	defer c.mu.Unlock()
+
+	c.accessToken = result.AccessToken
+	c.expiresIn = result.ExpiresIn
+	c.lastUpdated = time.Now()
+	glog.Infof(ctx, "获取微信 access_token 成功: %+v", c.accessToken)
+	return nil
+}
+func (c *weChatClient) autoRefreshToken(ctx context.Context) {
+	for {
+		// 初次获取
+		if err := c.getAccessToken(); err != nil {
+			time.Sleep(1 * time.Minute)
+			continue
+		}
+
+		// 等待 expiresIn - 300 秒（即提前 5 分钟刷新）
+		c.mu.RLock()
+		expiresIn := c.expiresIn
+		c.mu.RUnlock()
+
+		refreshAfter := time.Duration(expiresIn-300) * time.Second
+		time.Sleep(refreshAfter)
+	}
+}
+func (c *weChatClient) GetTicket(sceneID string) (string, error) {
+	body := map[string]interface{}{
+		"expire_seconds": c.TicketExpire,
+		"action_name":    "QR_SCENE",
+		"action_info": map[string]interface{}{
+			"scene": map[string]interface{}{
+				"scene_str": sceneID,
+			},
+		},
+	}
+	result := struct {
+		Ticket        string `json:"ticket"`
+		ExpireSeconds int    `json:"expire_seconds"`
+		Url           string `json:"url"`
+	}{}
+
+	resp, err := resty.New().R().
+		SetQueryParams(g.MapStrStr{"access_token": c.accessToken}).
+		SetBody(body).
+		SetResult(&result).
+		Post("https://api.weixin.qq.com/cgi-bin/qrcode/create")
+
+	if err != nil {
+		glog.Errorf(context.Background(), "发起 get ticket 请求出现异常: %+v", err)
+		return "", ecode.Fail.Sub("发起 get ticket 请求出现异常")
+	}
+	if resp.StatusCode() != 200 {
+		glog.Errorf(context.Background(), "获取微信 ticket 响应异常: %+v", resp.Status())
+		return "", ecode.Fail.Sub("获取微信 ticket 失败")
+	}
+
+	glog.Infof(context.Background(), "获取微信 ticket 成功: %+v", result.Ticket)
+	return result.Ticket, nil
+}
+
+func (c *weChatClient) GetQrCode(ticket string, filename string) (imagePath string, err error) {
+	qrCodeUrl := "https://mp.weixin.qq.com/cgi-bin/showqrcode"
+	resp, err := resty.New().R().
+		SetDoNotParseResponse(true).
+		SetQueryParams(g.MapStrStr{"ticket": ticket}).
+		Get(qrCodeUrl)
+
+	if err != nil {
+		glog.Errorf(context.Background(), "获取二维码图片请求异常: %+v", err)
+		return "", ecode.Fail.Sub("获取二维码图片请求失败")
+	}
+	if resp.StatusCode() != 200 {
+		glog.Errorf(context.Background(), "获取二维码图片响应异常: %+v", resp.Status())
+		return "", ecode.Fail.Sub("获取二维码图片失败")
+	}
+
+	imagePath = filename
+
+	data, readErr := io.ReadAll(resp.RawBody())
+	if readErr != nil {
+		glog.Errorf(context.Background(), "读取二维码图片失败: %+v", readErr)
+		return "", ecode.Fail.Sub("读取二维码图片失败")
+	}
+	defer resp.RawBody().Close()
+
+	writeErr := os.WriteFile(imagePath, data, 0644)
+	if writeErr != nil {
+		glog.Errorf(context.Background(), "保存二维码图片失败: %+v", writeErr)
+		return "", ecode.Fail.Sub("保存二维码图片失败")
+	}
+
+	glog.Infof(context.Background(), "二维码图片保存成功: %s", imagePath)
+	return imagePath, nil
+}