Golang实现JWT认证 生成验证Token全流程

github.com/golang-jwt/jwt/v5

package main

import (
    "fmt"
    "time"

    "github.com/golang-jwt/jwt/v5"
)

// 定义JWT的Claims结构体
type UserClaims struct {
    UserID   string `json:"user_id"`
    Username string `json:"username"`
    jwt.RegisteredClaims
}

// 假设这是你的密钥，实际应用中应该从环境变量或配置中心读取
var jwtSecret = []byte("my_super_secret_key_that_should_be_long_and_complex_and_random")

// GenerateToken 用于生成JWT
func GenerateToken(userID, username string) (string, error) {
    // 设置Token过期时间，比如1小时
    expirationTime := time.Now().Add(1 * time.Hour)

    claims := &UserClaims{
        UserID:   userID,
        Username: username,
        RegisteredClaims: jwt.RegisteredClaims{
            ExpiresAt: jwt.NewNumericDate(expirationTime), // 过期时间
            IssuedAt:  jwt.NewNumericDate(time.Now()),     // 签发时间
            NotBefore: jwt.NewNumericDate(time.Now()),     // 生效时间
            Issuer:    "your-app-name",                    // 签发者
            Subject:   userID,                             // 主题
        },
    }

    // 创建一个新的Token
    token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)

    // 使用密钥签名Token
    tokenString, err := token.SignedString(jwtSecret)
    if err != nil {
        return "", fmt.Errorf("签名Token失败: %w", err)
    }
    return tokenString, nil
}

// ValidateToken 用于验证JWT并解析Claims
func ValidateToken(tokenString string) (*UserClaims, error) {
    token, err := jwt.ParseWithClaims(tokenString, &UserClaims{}, func(token *jwt.Token) (interface{}, error) {
        // 确保签名方法是预期的，这里我们用HS256
        if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
            return nil, fmt.Errorf("非预期的签名方法: %v", token.Header["alg"])
        }
        return jwtSecret, nil
    })

    if err != nil {
        // 这里可以细化错误类型，比如过期、签名无效等
        if ve, ok := err.(*jwt.ValidationError); ok {
            if ve.Errors&jwt.ValidationErrorMalformed != 0 {
                return nil, fmt.Errorf("Token格式不正确")
            } else if ve.Errors&jwt.ValidationErrorExpired != 0 {
                return nil, fmt.Errorf("Token已过期")
            } else if ve.Errors&jwt.ValidationErrorNotValidYet != 0 {
                return nil, fmt.Errorf("Token尚未生效")
            } else {
                return nil, fmt.Errorf("Token无效: %w", err)
            }
        }
        return nil, fmt.Errorf("解析Token失败: %w", err)
    }

    if claims, ok := token.Claims.(*UserClaims); ok && token.Valid {
        return claims, nil
    }
    return nil, fmt.Errorf("Token验证失败或Claims解析失败")
}

// 简单示例用法 (可以放到main函数里测试)
/*
func main() {
    // 生成Token
    token, err := GenerateToken("user123", "Alice")
    if err != nil {
        fmt.Println("Error generating token:", err)
        return
    }
    fmt.Println("Generated Token:", token)

    // 验证Token
    claims, err := ValidateToken(token)
    if err != nil {
        fmt.Println("Error validating token:", err)
        return
    }
    fmt.Printf("Token Valid! UserID: %s, Username: %s\n", claims.UserID, claims.Username)

    // 模拟Token过期（实际应用中不会这么做，这里为了演示）
    fmt.Println("\n--- 模拟Token过期 ---")
    // 临时修改过期时间，生成一个立即过期的token
    oldSecret := jwtSecret
    jwtSecret = []byte("another_temp_secret") // 用不同的密钥避免干扰
    tempTokenFunc := func(userID, username string) (string, error) {
        expirationTime := time.Now().Add(-1 * time.Minute) // 设为过去时间
        claims := &UserClaims{
            UserID:   userID,
            Username: username,
            RegisteredClaims: jwt.RegisteredClaims{
                ExpiresAt: jwt.NewNumericDate(expirationTime),
                IssuedAt:  jwt.NewNumericDate(time.Now()),
                NotBefore: jwt.NewNumericDate(time.Now()),
                Issuer:    "your-app-name",
                Subject:   userID,
            },
        }
        token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
        return token.SignedString(jwtSecret)
    }

    expiredToken, err := tempTokenFunc("user456", "Bob")
    if err != nil {
        fmt.Println("Error generating expired token:", err)
        return
    }
    fmt.Println("Generated Expired Token:", expiredToken)

    _, err = ValidateToken(expiredToken)
    if err != nil {
        fmt.Println("Error validating expired token (expected):", err)
    }
    jwtSecret = oldSecret // 恢复密钥
}
*/

ValidateToken

.env