100行Golang打造灵活JWT库，精简高效，助你轻松实现验证

2023-11-01 14:49:53

导语： JWT（JSON Web Token）已成为现代应用程序中广泛采用的安全机制，它能够简化身份验证和授权流程。如果您正在寻找一个既精简又高效的Golang JWT库，那么您来对地方了！本文将为您展示如何仅用不到100行代码，便打造一套灵活的Golang JWT库，让您轻松实现验证。

什么是JWT？

JWT是一种紧凑的URL安全格式，用于在各方之间传输声明。它通常用于在用户和服务器之间传递身份信息，实现无状态的身份验证。JWT由三部分组成：标头、有效载荷和签名。

打造灵活的JWT库：分步指南

步骤1：导入必要的库

import (
    "crypto/hmac"
    "crypto/sha256"
    "encoding/base64"
    "encoding/json"
    "time"
)

步骤2：创建JWT结构体

type JWT struct {
    Header      map[string]interface{}
    Payload     map[string]interface{}
    Signature   string
    SigningKey  []byte
    SigningAlgo string
}

步骤3：生成JWT

func (jwt *JWT) Generate() (string, error) {
    // 编码header和payload
    headerBytes, err := json.Marshal(jwt.Header)
    if err != nil {
        return "", err
    }
    payloadBytes, err := json.Marshal(jwt.Payload)
    if err != nil {
        return "", err
    }

    // 创建签名
    signature := hmac.New(sha256.New, jwt.SigningKey)
    signature.Write(headerBytes)
    signature.Write([]byte("."))
    signature.Write(payloadBytes)
    jwt.Signature = base64.StdEncoding.EncodeToString(signature.Sum(nil))

    // 组合JWT
    return base64.StdEncoding.EncodeToString(headerBytes) + "." +
        base64.StdEncoding.EncodeToString(payloadBytes) + "." + jwt.Signature, nil
}

步骤4：验证JWT

func (jwt *JWT) Verify() error {
    // 拆分JWT
    parts := strings.Split(jwt.Signature, ".")
    if len(parts) != 3 {
        return errors.New("invalid JWT format")
    }

    // 验证签名
    signature := hmac.New(sha256.New, jwt.SigningKey)
    signature.Write([]byte(parts[0]))
    signature.Write([]byte("."))
    signature.Write([]byte(parts[1]))
    expectedSignature := base64.StdEncoding.EncodeToString(signature.Sum(nil))
    if expectedSignature != parts[2] {
        return errors.New("invalid signature")
    }

    // 验证payload
    payload := map[string]interface{}{}
    err := json.Unmarshal([]byte(parts[1]), &payload)
    if err != nil {
        return err
    }

    // 验证过期时间
    if exp, ok := payload["exp"]; ok {
        if exp.(float64) < float64(time.Now().Unix()) {
            return errors.New("token expired")
        }
    }

    return nil
}

步骤5：示例用法

// 创建JWT实例
jwt := JWT{
    Header: map[string]interface{}{
        "alg": "HS256",
        "typ": "JWT",
    },
    Payload: map[string]interface{}{
        "name": "John Doe",
        "exp":  time.Now().Add(time.Hour * 24).Unix(),
    },
    SigningKey:  []byte("secretKey"),
    SigningAlgo: "HS256",
}

// 生成JWT
token, err := jwt.Generate()
if err != nil {
    // 处理错误
}

// 验证JWT
err = jwt.Verify()
if err != nil {
    // 处理错误
}