GO制作微信机器人的流程分析

package autoload
type WeChatConfig struct {
AgentId string ini:"wechat" yaml:"agent_id"
Secret string ini:"wechat" yaml:"secret"
CompanyId string ini:"wechat" yaml:"company_id"
}
var WeChat = WeChatConfig{}

config.Config.WeChat.CompanyId //yaml中的company_id字段

func SetRedis(key string, value string, t int64) bool {
expire := time.Duration(t) * time.Second
if err := rdb.Set(ctx, key, value, expire).Err(); err != nil {
return false
}
return true
}

func GetRedis(key string) string {
result, err := rdb.Get(ctx, key).Result()
if err != nil {
return “”
}
return result
}

func DelRedis(key string) bool {
_, err := rdb.Del(ctx, key).Result()
if err != nil {
return false
}
return true
}

{
“touser”: “@all”,
“msgtype”: “text”,
“agentid”: “xxxxx”,
“text”: {“content”: “xxxx”}
}

package model
type wcSendcontent struct {
Content string JSON:"content"
}
type WcSendMsg struct {
ToUser string json:"touser"
MsgType string json:"msgtype"
AgentId string json:"agentid"
Text wcSendcontent json:"text"
}
func (t *WcSendMsg) SetMessage(message string) {
t.Text.Content = message
}

package service

import (
“bytes”
“encoding/json”
“errors”
“fmt”

c "github.com/wannanbigpig/gin-layout/config"
"github.com/wannanbigpig/gin-layout/data"
"github.com/wannanbigpig/gin-layout/internal/model"
log "github.com/wannanbigpig/gin-layout/internal/pkg/logger"
"github.com/wannanbigpig/gin-layout/pkg/utils"
"go.uber.org/zap"

)


func SendWeChat(message string, msgType string) error {
redis_key := “access_token”
// 尝试从redis中读取token
accessToken := data.GetRedis(redis_key)
http := &utils.HttpRequest{}
// 若redis中的token已过期，则重新请求api获取token
if accessToken == “” {
log.Logger.Info(“access token is null, will recall”)
getTokenUrl := fmt.Sprintf(“https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=%s&corpsecret=%s”,
c.Config.WeChat.CompanyId, c.Config.WeChat.Secret)
log.Logger.Info(“token_url”, zap.String(“url”, getTokenUrl))
http.Request(“GET”, getTokenUrl, nil)
ret := make(map[string]interface{})
if err := http.ParseJson(&ret); err != nil {
return err
}
marshal, _ := json.Marshal(ret)
log.Logger.Info(string(marshal))
accessToken = fmt.Sprintf("%v", ret[“access_token”])
// 写入redis 有效期2小时
data.SetRedis(redis_key, accessToken, 7200)
}
msg := &model.WcSendMsg{
ToUser: “@all”,
MsgType: msgType,
AgentId: c.Config.WeChat.AgentId,
}
msg.SetMessage(message)
sendMsgUrl := fmt.Sprintf(“https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=%v”, accessToken)
log.Logger.Info(“sendMsgUrl = " + string(sendMsgUrl))
header := map[string]string{“Content-Type”: “application/json”}
bytesData, _ := json.Marshal(msg)
http.Request(“POST”, sendMsgUrl, bytes.NewReader(bytesData), header)
log.Logger.Info(“bytes data = " + string(bytesData))
ret := make(map[string]interface{})
err := http.ParseJson(&ret)
if err != nil {
return err
}
if ret[“errcode”].(float64) != 0 {
errmsg := fmt.Sprintf(”%v”, ret[“errmsg”])
return errors.New(errmsg)
}
return nil
}

package wechat

import (
“github.com/gin-gonic/gin”
“github.com/wannanbigpig/gin-layout/internal/pkg/error_code”
log “github.com/wannanbigpig/gin-layout/internal/pkg/logger”
r “github.com/wannanbigpig/gin-layout/internal/pkg/response”
“github.com/wannanbigpig/gin-layout/internal/service”
)

func SendMsg(c *gin.Context) {
msg, ok := c.GetQuery(“msg”)
if !ok {
msg = “please input message”
}
log.Logger.Info("send wechat message: " + msg)
err := service.SendWeChat(msg, “text”)
if err != nil {
r.Resp().FailCode(c, error_code.FAILURE, err.Error())
return
}
r.Success(c, “success”)
}

package routers

import (
“github.com/gin-gonic/gin”
w “github.com/wannanbigpig/gin-layout/internal/controller/wechat”
)

func setWeChatRouter(r *gin.Engine) {
// version 1
v1 := r.Group(“wechat”)
{
v1.GET("/send", w.SendMsg)
}
}

curl --location --request GET “http:// I P : {IP}: IP:{PORT}/wechat/send?msg=Hello,Golang”