go - 生成 JWT token 时类型不匹配导致错误

Question

我正在尝试生成 JWT token ，但在将所有内容转换为字节数组时遇到了问题。

func GenerateToken(uid, cid int64) string{
    header := `{"alg": "HS256","typ": "JWT"}`
    header = base64.URLEncoding.EncodeToString([]byte(header))
    var b structs.JwtBody
    b.UID = uid
    b.CID = cid
    body, _ := json.Marshal(b)
    key := []byte(secret)
    h := hmac.New(sha256.New, key)
    h.Write([]byte(header + "." + body))
    signature := base64.URLEncoding.EncodeToString(h.Sum(nil))
    jwt := header + "." + body + "." + signature
    return jwt
}

我收到此行的错误，因为类型不匹配。

h.Write([]byte(header + "." + body))

我该如何解决这个问题？

Answer 1

将 h.Write 行更改为以下内容:

h.Write([]byte(header))
h.Write([]byte("."))
h.Write(body)

将散列计算拆分为三个单独的调用应该(稍微)快于将各个部分连接在一起并编写。