GO制作微信機(jī)器人的流程分析
這些天在學(xué)習(xí)Go，也寫了幾篇關(guān)于閱讀Gin后端項(xiàng)目代碼的博客。但編程這種，一定要實(shí)際上手練習(xí)，要不然都是紙上談兵。于是就想上手自己實(shí)際寫一些代碼來練練手。思來想去，不知道能寫些什么來練手。后來突然想到，之前寫過用Python做微信聊天機(jī)器人（博客傳送門），當(dāng)時代碼沒有放到git上，后來重置了服務(wù)器導(dǎo)致代碼全部沒了?，F(xiàn)在正好苦于不知道做什么項(xiàng)目練手，可以用Go也實(shí)現(xiàn)一套微信聊天機(jī)器人。
說干就干，照著之前自己寫的博客，看了下當(dāng)時Python的代碼。轉(zhuǎn)而用Go優(yōu)化了下并實(shí)現(xiàn)。
0.回顧流程
根據(jù)之前Python寫的自動發(fā)消息的機(jī)器人可知，要想發(fā)消息就需要三個參數(shù)：company_id、secret、angent_id。 對于這三個參數(shù)如何獲取，可參考文章開頭的傳送門。整個發(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)大量時間。因此我使用了基于開源gin項(xiàng)目進(jìn)行二次開發(fā)的方法，實(shí)現(xiàn)這個機(jī)器人。
前幾天在學(xué)習(xí)Gin時，發(fā)現(xiàn)了一位老哥封裝了個Gin腳手架，可以達(dá)到開箱即用目的。項(xiàng)目地址: github傳送門。 里邊把讀取配置文件，編寫路由，連接數(shù)據(jù)庫等多個操作均進(jìn)行了實(shí)現(xiàn)。因此可以基于這個項(xiàng)目來進(jìn)行二次開發(fā)，做微信機(jī)器人。
在把項(xiàng)目clone下來后，可以先看下整個項(xiàng)目的布局，主要的業(yè)務(wù)核心代碼都放在了internal 下面。如果我們要實(shí)現(xiàn)一個主動給微信發(fā)消息的功能，那么多說了就是寫一個發(fā)送消息的方法，讓后端調(diào)用這個方法即可。
要想基于此項(xiàng)目來開發(fā)微信機(jī)器人，首先就要將三個參數(shù)配置上。項(xiàng)目中，對于各種參數(shù)均在config.yaml中配置，因?yàn)榭梢栽谶@個配置文件中增加這三個參數(shù)的配置：
然后在代碼的config/autoload目錄下新增一個weCaht.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{}并且，將此配置加入到項(xiàng)目的配置集合中。在config/config.go中添加如下代碼：
這樣操作，就可以通過代碼來讀取配置文件了。在其他包中，可以通過如下方式來訪問對應(yīng)的值
config.Config.WeChat.CompanyId //yaml中的company_id字段
2. Redis封裝
因?yàn)橐o微信發(fā)送消息，首先要獲取到token，而官方介紹此token的有效時長為2小時。在之前Python的項(xiàng)目中，是直接將token寫到了文件中，通過文件來讀取。在此項(xiàng)目中，我想直接使用redis來存儲。因?yàn)槭褂胷edis來存儲的話，可以設(shè)置key值時長，過了這個時長就自動清除，這樣就方便了許多。
而我們基于這個gin-layout項(xiàng)目中，已經(jīng)對redis做了一層封裝，具體代碼可查看data/redis.go，主要是通過對外暴露一個Rdb的結(jié)構(gòu)體，來操作redis
而目前我們這邊使用redis，只會用到對應(yīng)的set和get方法。因此我對這個項(xiàng)目中的redis又做了一層封裝。只對外暴露set，get，del方法。
首先將Rdb變量名改為小寫，這樣就代表不對外暴露，然后在此文件中添加如下代碼
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
}這樣，后續(xù)使用redis時候，只需要調(diào)用data.SetRedis(xxx) 即可。
然后就是修改配置文件，啟用redis，這里根據(jù)實(shí)際的redis配置來寫即可。
3.消息體封裝
在最終給微信服務(wù)器發(fā)送post請求時，對應(yīng)的請求體格式如下：
{
“touser”: “@all”,
“msgtype”: “text”,
“agentid”: “xxxxx”,
“text”: {“content”: “xxxx”}
}因此，接下來可以對這個結(jié)構(gòu)體做一個封裝。在model包下，新建一個send_msg.go文件
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
}這里針對message信息，專門對外暴露了一個方法來進(jìn)行設(shè)置。
4.核心代碼
在設(shè)置好redis，消息體封裝后，就可以編寫核心的代碼了。主要就是通過發(fā)送http請求，獲取token，再通過token發(fā)送post請求來發(fā)送消息。我們可以在service包下新建一個weChat.go的文件，里邊新建一個SendWeChat方法來進(jìn)行消息發(fā)送操作。
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"
)
/**
@description: 給企微發(fā)送消息
@param {string} message 消息內(nèi)容
@param {string} msgType 消息類型
@return {*}
*/
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
}從上面代碼中可以看出，首先是通過redis來獲取token，若沒有則請求api獲取token，并將其寫入到redis中，有效期為2小時。然后生成一個之前封裝的消息的結(jié)構(gòu)體，將AgentId和message進(jìn)行填充后，通過發(fā)送post請求，已達(dá)到發(fā)消息的目的。
5.本地測試
若想驗(yàn)證這個方法，可以通過對外提供一個接口，訪問此接口后調(diào)用發(fā)送消息的方法。
可以在controller目錄下新建一個weChat.go，在里邊實(shí)現(xiàn)一個get請求的方法，獲取請求中的msg參數(shù)，然后調(diào)用剛才實(shí)現(xiàn)的發(fā)送企微的方法。
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”)
}寫好后，將此方法綁定到路由上。在routers包下新建一個weChatRouter.go文件
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)
}
}這樣，后續(xù)可以通過wechat/send的url來請求這個接口。最后就是調(diào)用此綁定路由的方法，在routers/router.go中添加一行代碼即可
接下來啟動項(xiàng)目，比如發(fā)送一個msg=Hello,Golang 的請求
curl --location --request GET “http:// I P : {IP}: IP:{PORT}/wechat/send?msg=Hello,Golang”執(zhí)行這個命令，就可以得到本文開頭的截圖。
當(dāng)然，這個api接口主要是為了讓我們驗(yàn)證，實(shí)際項(xiàng)目運(yùn)行時，建議不要這么搞。因?yàn)檫@接口沒有任何鑒權(quán)的措施，如果對外暴露了出去，那么別人也可以肆意的調(diào)用這個接口給你的企微發(fā)送消息。
到此這篇關(guān)于利用go制作微信機(jī)器人的文章就介紹到這了,更多相關(guān)go微信機(jī)器人內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
7分鐘讀懂Go的臨時對象池pool以及其應(yīng)用場景
這篇文章主要給大家介紹了關(guān)于如何通過7分鐘讀懂Go的臨時對象池pool以及其應(yīng)用場景的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對大家學(xué)習(xí)或使用Go具有一定的參考學(xué)習(xí)價值，需要的朋友們下面來一起看看吧2018-11-11
這篇文章主要為大家介紹了Go語言中的速率限流策略全面詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-11-11
這篇文章主要介紹了詳解Go中Set的實(shí)現(xiàn)方式，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-09-09
這篇文章主要為大家介紹了Go快速開發(fā)一個RESTful API服務(wù)，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-06-06
這篇文章主要詳細(xì)介紹了Go 1.22 for循環(huán)的兩處重要更新,Go 1.22 版本于 2024 年 2 月 6 日發(fā)布,引入了幾個重要的特性和改進(jìn),在語言層面上,這個版本對 for 循環(huán)進(jìn)行了兩處更新,本文將會對 for 循環(huán)的兩個更新進(jìn)行介紹,需要的朋友可以參考下2024-02-02
在 Go 語言中,for 循環(huán)是實(shí)現(xiàn)迭代的主要方式,Go 中的 for 循環(huán)非常靈活,有多種使用方式,本文將給大家詳細(xì)的介紹一下Go 1.22中的for循環(huán)新特性,感興趣的朋友可以參考下2024-02-02
我們的代碼業(yè)務(wù)中很多地方需要我們自己進(jìn)行排序操作，本文主要介紹了詳解go語言中sort如何排序，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-03-03
Go channel發(fā)送方和接收方如何相互阻塞等待源碼解讀
這篇文章主要為大家介紹了Go channel發(fā)送方和接收方如何相互阻塞等待源碼解讀,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-12-12
這篇文章主要想來和大家討論一下,GO?語言處理并發(fā)的時候我們是選擇sync還是channel,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以了解下2023-08-08
Go語言實(shí)現(xiàn)百萬級WebSocket連接架構(gòu)設(shè)計及服務(wù)優(yōu)化
本文將詳細(xì)介紹如何在Go中構(gòu)建一個能夠支持百萬級WebSocket連接的服務(wù),包括系統(tǒng)架構(gòu)設(shè)計、性能優(yōu)化策略以及具體的實(shí)現(xiàn)步驟和代碼示例2024-01-01