消息推送配置接入指引

import (
    "context"
    "crypto/md5"
    "encoding/base64"
    "encoding/json"
    "fmt"
    "sort"
    "strings"
)

/**
比如：
   header := map[string]string{
                "x-nonce-str": "313932313532383034",
                "x-timestamp": "1737635474798",
                "x-appid":    "tt12321",
                "x-msg-type":  "verify_request",
        }
   bodyStr := "verify_body"
   token := "verify_token"

拼接好之后的 rawData为：x-appid=tt12321&x-msg-type=verify_request&x-nonce-str=123456&x-timestamp=456789verify_bodyverify_token

生成的 signatureStr 为：AoOtx/dFR5MFrCTqUmtmDg==
*/

// 生成数据签名
func generateSignature(ctx context.Context, headerMap map[string]string, bodyStr string, token string) string {
    // 1、对 headerMap 中的 key 按字典序从小到大排序（注意实际编码时，只对如上指定的 4个 key 进行排序）
    keyList := make([]string, 0, 5)
    for key := range headerMap {
       keyList = append(keyList, key)
    }
    sort.Slice(keyList, func(i, j int) bool {
       return keyList[i] < keyList[j]
    })
    // 2、将 key-value 按顺序拼接成字符串，格式为 key1=value1&key2=value2&...
    kvList := make([]string, 0, 5)
    for _, key := range keyList {
       kvList = append(kvList, key+"="+headerMap[key])
    }
    urlParams := strings.Join(kvList, "&")

    // 3、直接拼接(无需连接符)：urlParams + bodyStr + token
    rawData := urlParams + bodyStr + token
    md5Result := md5.Sum([]byte(rawData))
    signatureStr := base64.StdEncoding.EncodeToString(md5Result[:])

    return signatureStr
}

// 接收消息的函数（平台上设置的 Url）
func HandleMessage(ctx context.Context, c *app.RequestContext) {
    msgType := c.Request.Header.Get("x-msg-type")
    // 0、从请求头读取 Header
    headerMap := map[string]string{
       "x-appid":     c.Request.Header.Get("x-appid"),
       "x-msg-type":  msgType,
       "x-nonce-str": c.Request.Header.Get("x-nonce-str"),
       "x-timestamp": c.Request.Header.Get("x-timestamp"),
    }
    // 读取 body
    bodyStr := c.Request.Body()
    // 平台上设置的 Token
    token := "xxxxxxxx"
    // 1、生成签名
    signature := generateSignature(ctx, headerMap, string(bodyStr), token)

    // 2、验证签名
    logs.CtxInfo(ctx, "generate signature: %v", signature)
    logs.CtxInfo(ctx, "request signature: %v", c.Request.Header.Get("x-signature"))
    if signature != c.Request.Header.Get("x-signature") {
       service.SendLarkMessage(ctx, "验签失败")
       c.JSON(500, utils.H{
          "message": "signature not equal",
       })
       return
    }
    // 3、接收的是验证消息，则直接返回，无需处理后续逻辑
    if msgType == "verify_request" {
       c.JSON(200, utils.H{})
       return
    }
    
    // 接收的是游戏站礼包推送消息，处理请求 .....
    //（务必判断此 msgType 做好事件分离，后续抖音可能还会推送其他 msgType 到同样的服务器地址上）
    if msgType == "gift_delivery" {
      
       xxxxxxx // 具体执行逻辑
      
       c.JSON(200, utils.H{})
       return
    }

    // 接收的是抖音客服推送消息，处理请求 .....
    //（务必判断此 msgType 做好事件分离，后续抖音可能还会推送其他 msgType 到同样的服务器地址上）
    if msgType == "douyin_microgame_im" {
      
       xxxxxxx // 具体执行逻辑
      
       c.JSON(200, utils.H{})
       return
    }
    
    // 其他 msgType，不做处理