package main
import (
   "fmt"
   "github.com/wonderivan/logger"
   "net/http"
   "time"
)
const (
   port     = 80 //后台服务器端口号80，自己的公网服务器 需要设置防火墙，打开80端口
   token    = "XXXXXXX"
)
// 需要自己修改token，以适应自己公众号的token
func main() {
   logger.SetLogger("./log.json")
   logger.Info(" ------------ start  main ------------")
   server := http.Server{
      Addr:           fmt.Sprintf(":%d", port), // 设置监听地址， ip:port
      Handler:        &httpHandler{},           // 具体使用哪个handler来处理
      ReadTimeout:    5 * time.Second,          // 读写超时 微信给出来 5秒
      WriteTimeout:   5 * time.Second,
      MaxHeaderBytes: 0,
   }
   logger.Info(fmt.Sprintf("Listen: %d", port))
   logger.Fatal(server.ListenAndServe())
}

package main
import (
   "fmt"
   "github.com/wonderivan/logger"
   "io"
   "net/http"
   "regexp"
   "time"
   "wechat/wx"
)
type WebController struct {
   Function func(http.ResponseWriter, *http.Request)
   Method   string
   Pattern  string
}
var mux []WebController // 自己定义的路由
// ^ 匹配输入字符串的开始位置
func init() {
   mux = append(mux, WebController{post, "POST", "^/"})
   mux = append(mux, WebController{get, "GET", "^/"})
}
type httpHandler struct{} 
//httpHandler 自己实现了 ServeHTTP 接口，可以通过源码看到，创建httpHandler 对象的时候，实际源码逻辑是会调用ServeHTTP进行处理
func (*httpHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
   t := time.Now()
   for _, webController := range mux { // 遍历路由
      // 匹配请求的   r.URL.Path  -> webController.Pattern
      if m, _ := regexp.MatchString(webController.Pattern, r.URL.Path); m { // 匹配URL
         logger.Info("webController.Pattern  == ", webController.Pattern)
         logger.Info("r.URL.Path  == ", r.URL.Path)
         if r.Method == webController.Method { // 匹配方法
            logger.Info("webController.Method  == ", webController.Method)
            webController.Function(w, r) // 调用对应的处理函数
            d := time.Now().Sub(t)
            l := fmt.Sprintf("[ACCESS] | % -10s | % -40s | % -16s", r.Method, r.URL.Path, d.String())
            logger.Info(l)
            return
         }
      }
   }
   d := time.Now().Sub(t)
   l := fmt.Sprintf("[ACCESS] | % -10s | % -40s | % -16s", r.Method, r.URL.Path, d.String())
   logger.Info(l)
   io.WriteString(w, "")
   return
}
// 处理token的认证
func get(w http.ResponseWriter, r *http.Request) {
   client, err := wx.NewClient(r, w, token)
   if err != nil {
      logger.Info(err)
      w.WriteHeader(403) // 校验失败
      return
   }
   if len(client.Query.Echostr) > 0 {
      logger.Info("Echostr == ", client.Query.Echostr)
      w.Write([]byte(client.Query.Echostr)) // 校验成功返回的是Echostr
      return
   }
   w.WriteHeader(403)
   return
}
// 微信平台过来消息， 处理 ，然后返回微信平台
func post(w http.ResponseWriter, r *http.Request) {
   client, err := wx.NewClient(r, w, token)
   if err != nil {
      logger.Info(err)
      w.WriteHeader(403)
      return
   }
   // 到这一步签名已经验证通过了
   client.Run()
   return
}