GO制作微信機(jī)器人的流程分析

第GO制作微信機(jī)器人的流程分析目錄0.回顧流程1.項(xiàng)目基礎(chǔ)配置2.Redis封裝3.消息體封裝4.核心代碼5.本地測試這些天在學(xué)習(xí)Go，也寫了幾篇關(guān)于閱讀Gin后端項(xiàng)目代碼的博客。但編程這種，一定要實(shí)際上手練習(xí)，要不然都是紙上談兵。于是就想上手自己實(shí)際寫一些代碼來練練手。思來想去，不知道能寫些什么來練手。后來突然想到，之前寫過用Python做微信聊天機(jī)器人（博客傳送門），當(dāng)時(shí)代碼沒有放到git上，后來重置了服務(wù)器導(dǎo)致代碼全部沒了?，F(xiàn)在正好苦于不知道做什么項(xiàng)目練手，可以用Go也實(shí)現(xiàn)一套微信聊天機(jī)器人。

說干就干，照著之前自己寫的博客，看了下當(dāng)時(shí)Python的代碼。轉(zhuǎn)而用Go優(yōu)化了下并實(shí)現(xiàn)。

0.回顧流程

根據(jù)之前Python寫的自動(dòng)發(fā)消息的機(jī)器人可知，要想發(fā)消息就需要三個(gè)參數(shù)：company_id、secret、angent_id。對于這三個(gè)參數(shù)如何獲取，可參考文章開頭的傳送門。整個(gè)發(fā)送消息過程就是首先通過company_id和secret來調(diào)用接口獲取token，再通過token和angent_id來給對應(yīng)接口發(fā)送post請求，就可以把post請求體中的信息發(fā)送到微信上。

1.項(xiàng)目基礎(chǔ)配置

由于目前對Go的項(xiàng)目布局學(xué)習(xí)的還不是特別熟練，而且對于項(xiàng)目基礎(chǔ)部分如果從頭開始做的話，需要耗費(fèi)大量時(shí)間。因此我使用了基于開源gin項(xiàng)目進(jìn)行二次開發(fā)的方法，實(shí)現(xiàn)這個(gè)機(jī)器人。

前幾天在學(xué)習(xí)Gin時(shí)，發(fā)現(xiàn)了一位老哥封裝了個(gè)Gin腳手架，可以達(dá)到開箱即用目的。項(xiàng)目地址:github傳送門。里邊把讀取配置文件，編寫路由，連接數(shù)據(jù)庫等多個(gè)操作均進(jìn)行了實(shí)現(xiàn)。因此可以基于這個(gè)項(xiàng)目來進(jìn)行二次開發(fā)，做微信機(jī)器人。

在把項(xiàng)目clone下來后，可以先看下整個(gè)項(xiàng)目的布局，主要的業(yè)務(wù)核心代碼都放在了internal下面。如果我們要實(shí)現(xiàn)一個(gè)主動(dòng)給微信發(fā)消息的功能，那么多說了就是寫一個(gè)發(fā)送消息的方法，讓后端調(diào)用這個(gè)方法即可。

要想基于此項(xiàng)目來開發(fā)微信機(jī)器人，首先就要將三個(gè)參數(shù)配置上。項(xiàng)目中，對于各種參數(shù)均在config.yaml中配置，因?yàn)榭梢栽谶@個(gè)配置文件中增加這三個(gè)參數(shù)的配置：

然后在代碼的config/autoload目錄下新增一個(gè)weCaht.go文件，接收配置文件中的配置。

packageautoload

typeWeChatConfigstruct{

AgentIdstringini:"wechat"yaml:"agent_id"

Secretstringini:"wechat"yaml:"secret"

CompanyIdstringini:"wechat"yaml:"company_id"

varWeChat=WeChatConfig{}

并且，將此配置加入到項(xiàng)目的配置集合中。在config/config.go中添加如下代碼：

這樣操作，就可以通過代碼來讀取配置文件了。在其他包中，可以通過如下方式來訪問對應(yīng)的值

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

2.Redis封裝

因?yàn)橐o微信發(fā)送消息，首先要獲取到token，而官方介紹此token的有效時(shí)長為2小時(shí)。在之前Python的項(xiàng)目中，是直接將token寫到了文件中，通過文件來讀取。在此項(xiàng)目中，我想直接使用redis來存儲(chǔ)。因?yàn)槭褂胷edis來存儲(chǔ)的話，可以設(shè)置key值時(shí)長，過了這個(gè)時(shí)長就自動(dòng)清除，這樣就方便了許多。

而我們基于這個(gè)gin-layout項(xiàng)目中，已經(jīng)對redis做了一層封裝，具體代碼可查看data/redis.go，主要是通過對外暴露一個(gè)Rdb的結(jié)構(gòu)體，來操作redis

而目前我們這邊使用redis，只會(huì)用到對應(yīng)的set和get方法。因此我對這個(gè)項(xiàng)目中的redis又做了一層封裝。只對外暴露set，get，del方法。

首先將Rdb變量名改為小寫，這樣就代表不對外暴露，然后在此文件中添加如下代碼

funcSetRedis(keystring,valuestring,tint64)bool{

expire:=time.Duration(t)*time.Second

iferr:=rdb.Set(ctx,key,value,expire).Err();err!=nil{

returnfalse

returntrue

funcGetRedis(keystring)string{

result,err:=rdb.Get(ctx,key).Result()

iferr!=nil{

return“”

returnresult

funcDelRedis(keystring)bool{

_,err:=rdb.Del(ctx,key).Result()

iferr!=nil{

returnfalse

returntrue

}

這樣，后續(xù)使用redis時(shí)候，只需要調(diào)用data.SetRedis(xxx)即可。

然后就是修改配置文件，啟用redis，這里根據(jù)實(shí)際的redis配置來寫即可。

3.消息體封裝

在最終給微信服務(wù)器發(fā)送post請求時(shí)，對應(yīng)的請求體格式如下：

{

“touser”:“@all”,

“msgtype”:“text”,

“agentid”:“xxxxx”,

“text”:{“content”:“xxxx”}

}

因此，接下來可以對這個(gè)結(jié)構(gòu)體做一個(gè)封裝。在model包下，新建一個(gè)send_msg.go文件

packagemodel

typewcSendcontentstruct{

Contentstringjson:"content"

typeWcSendMsgstruct{

ToUserstringjson:"touser"

MsgTypestringjson:"msgtype"

AgentIdstringjson:"agentid"

TextwcSendcontentjson:"text"

func(t*WcSendMsg)SetMessage(messagestring){

t.Text.Content=message

}

這里針對message信息，專門對外暴露了一個(gè)方法來進(jìn)行設(shè)置。

4.核心代碼

在設(shè)置好redis，消息體封裝后，就可以編寫核心的代碼了。主要就是通過發(fā)送http請求，獲取token，再通過token發(fā)送post請求來發(fā)送消息。我們可以在service包下新建一個(gè)weChat.go的文件，里邊新建一個(gè)SendWeChat方法來進(jìn)行消息發(fā)送操作。

packageservice

import(

“bytes”

“encoding/json”

“errors”

“fmt”

c"/wannanbigpig/gin-layout/config"

"/wannanbigpig/gin-layout/data"

"/wannanbigpig/gin-layout/internal/model"

log"/wannanbigpig/gin-layout/internal/pkg/logger"

"/wannanbigpig/gin-layout/pkg/utils"

"/zap"

)

@description:給企微發(fā)送消息

@param{string}message消息內(nèi)容

@param{string}msgType消息類型

@return{*}

funcSendWeChat(messagestring,msgTypestring)error{

redis_key:=“access_token”

//嘗試從redis中讀取token

accessToken:=data.GetRedis(redis_key)

http:=utils.HttpRequest{}

//若redis中的token已過期，則重新請求api獲取token

ifaccessToken==“”{

log.Logger.Info(“accesstokenisnull,willrecall”)

getTokenUrl:=fmt.Sprintf(“/cgi-bin/gettokencorpid=%scorpsecret=%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{})

iferr:=http.ParseJson(ret);err!=nil{

returnerr

marshal,_:=json.Marshal(ret)

log.Logger.Info(string(marshal))

accessToken=fmt.Sprintf("%v",ret[“access_token”])

//寫入redis有效期2小時(shí)

data.SetRedis(redis_key,accessToken,7200)

msg:=model.WcSendMsg{

ToUser:“@all”,

MsgType:msgType,

AgentId:c.Config.WeChat.AgentId,

msg.SetMessage(message)

sendMsgUrl:=fmt.Sprintf(“/cgi-bin/message/sendaccess_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(“bytesdata="+string(bytesData))

ret:=make(map[string]interface{})

err:=http.ParseJson(ret)

iferr!=nil{

returnerr

ifret[“errcode”].(float64)!=0{

errmsg:=fmt.Sprintf(”%v”,ret[“errmsg”])

returnerrors.New(errmsg)

returnnil

}

從上面代碼中可以看出，首先是通過redis來獲取token，若沒有則請求api獲取token，并將其寫入到redis中，有效期為2小時(shí)。然后生成一個(gè)之前封裝的消息的結(jié)構(gòu)體，將AgentId和message進(jìn)行填充后，通過發(fā)送post請求，已達(dá)到發(fā)消息的目的。

5.本地測試

若想驗(yàn)證這個(gè)方法，可以通過對外提供一個(gè)接口，訪問此接口后調(diào)用發(fā)送消息的方法。

可以在controller目錄下新建一個(gè)weChat.go，在里邊實(shí)現(xiàn)一個(gè)get請求的方法，獲取請求中的msg參數(shù)，然后調(diào)用剛才實(shí)現(xiàn)的發(fā)送企微的方法。

packagewechat

import(

“/gin-gonic/gin”

“/wannanbigpig/gin-layout/internal/pkg/error_code”

log“/wannanbigpig/gin-layout/internal/pkg/logger”

r“/wannanbigpig/gin-layout/internal/pkg/response”

“/wannanbigpig/gin-layout/internal/service”

funcSendMsg(c*gin.Context){

msg,ok:=c.GetQuery(“msg”)

if!ok{

msg=“pleaseinputmessage”

log.Logger.Info("sendwechatmessage:"+msg)

err:=service.SendWeChat(msg,“text”)

iferr!=nil{

r.Resp().FailCode(c,error_code.FAILURE,err.Error())

re