Golang实现微信公众号后台接入的示例代码

ngork.exe http 80

package main

import (
	"fmt"
	"net/http"
	"wechat-oa/route"
)

func main() {
	fmt.Println("====== 微信公众号服务器程序 ======")

	http.HandleFunc("/", route.WechatServer) // 处理微信服务器请求

	err := http.ListenAndServe(":80", nil)
	if err != nil {
		fmt.Println("wechat server start error", err)
	}
}

// 微信公众号平台配置
const (
	appId          = "xxxxxxxxxx" // 公众号开发识别码
	appSecret      = "xxxxxxxxxx" // 公众号开发者密码
	encodingAESKey = "xxxxxxxxxx" // 消息加密秘钥
	token          = "xxxxxxxxxx" // 跟微信公众平台的token一样即可
)

func WechatServer(w http.ResponseWriter, req *http.Request) {
	defer req.Body.Close()

	fmt.Println("服务器请求: ", req) //打印http的请求url
	// 1、验证消息是否来自微信服务器
	hashcode,signature,echostr := handleValid(req)
	if hashcode != signature { // 校验
		return
	}
	// 2、请求处理
	if req.Method == "GET" {
		_, _ = w.Write([]byte(echostr)) // 验证成功返回echostr
	} else if req.Method == "POST" {
		handleMsg(w,req)
	}else {
		_, _ = w.Write([]byte("error"))
	}
}

func handleValid(r *http.Request) (hashcode string,signature string,echostr string){
	//1.尝试获取4个字段
	nonce := r.URL.Query().Get("nonce")
	timestamp := r.URL.Query().Get("timestamp")
	signature = r.URL.Query().Get("signature")
	echostr = r.URL.Query().Get("echostr")
	//2. 赋值一个token

	//3.token，timestamp，nonce按字典排序的字符串list
	strs := sort.StringSlice{token, timestamp, nonce} // 使用本地的token生成校验
	sort.Strings(strs)
	str := ""
	for _, s := range strs {
		str += s
	}
	// 4. 哈希算法加密list得到hashcode
	h := sha1.New()
	h.Write([]byte(str))
	hashcode = fmt.Sprintf("%x", h.Sum(nil)) // h.Sum(nil) 做hash
	return
}

// 处理POST请求
// 微信服务器会将用户发送的数据以POST请求的方式转发给开发者服务器
func handleMsg(rw http.ResponseWriter, req *http.Request) {
	wc := wechat.NewWechat()
	//这里本地内存保存access_token，也可选择redis，memcache或者自定义cache
	memory := cache.NewMemory()
	cfg := &offConfig.Config{
		AppID:          appId,
		AppSecret:      appSecret,
		Token:          token,
		EncodingAESKey: encodingAESKey,
		Cache:          memory,
	}
	// 底层根据AppID和AppSecret获取access_token并保存到cache中
	// access_token每两个小时刷新一次
	officialAccount := wc.GetOfficialAccount(cfg)
	// 传入request和responseWriter
	server := officialAccount.GetServer(req, rw)
	// 设置接收消息的处理方法
	server.SetMessageHandler(controller.HandleMsg)
	//处理消息接收以及回复
	err := server.Serve()
	if err != nil {
		fmt.Println(err)
		return
	}
	//发送回复的消息
	_ = server.Send()
}

// 处理用户发送的消息
func HandleMsg(msg *message.MixMessage) *message.Reply{
	fmt.Println("接收用户消息:",msg.Content)
	switch msg.MsgType {
	case message.MsgTypeText: // 文本消息
		return &message.Reply{
			MsgType: message.MsgTypeText,
			MsgData: message.NewText(msg.Content),
		}
	case message.MsgTypeEvent: // 事件推送
		return service.HandleEventMsg(msg)
	default:
		return &message.Reply{
			MsgType: message.MsgTypeText,
			MsgData: message.NewText("感谢关注沐风丶,更多功能正在开发中..."),
		}
	}
}

// 处理事件推送
func HandleEventMsg(msg *message.MixMessage) *message.Reply{
	switch msg.Event {
	case message.EventSubscribe: // 用户订阅
		fmt.Printf("%s 订阅了你 \n",msg.FromUserName)
		return &message.Reply{
			MsgType: message.MsgTypeText,
			MsgData: message.NewText("感谢关注沐风丶,(*￣︶￣)❀你真好看"),
		}
	case message.EventUnsubscribe: // 用户取消订阅
		fmt.Printf("%s 取消了订阅 \n",msg.FromUserName)
		return &message.Reply{
			MsgType: message.MsgTypeText,
			MsgData: message.NewText(""),
		}
	default:
		return &message.Reply{
			MsgType: message.MsgTypeText,
			MsgData: message.NewText("default"),
		}
	}
}