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

jwt.StandardClaims

package main

import (
    "fmt"
    "log"
    "time"

    "github.com/dgrijalva/jwt-go"
)

// MyClaims 自定义声明，包含标准声明和额外信息
type MyClaims struct {
    UserID   string `json:"user_id"`
    Username string `json:"username"`
    jwt.StandardClaims
}

// SecretKey 用于签名和验证的密钥，生产环境请务必从安全的地方加载
var SecretKey = []byte("super_secret_key_that_no_one_should_know")

// GenerateToken 生成JWT令牌
func GenerateToken(userID, username string) (string, error) {
    // 设置令牌过期时间，比如1小时
    expirationTime := time.Now().Add(1 * time.Hour)
    claims := &MyClaims{
        UserID:   userID,
        Username: username,
        StandardClaims: jwt.StandardClaims{
            ExpiresAt: expirationTime.Unix(), // 过期时间
            IssuedAt:  time.Now().Unix(),     // 签发时间
            Issuer:    "my-golang-app",       // 签发者
        },
    }

    token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
    tokenString, err := token.SignedString(SecretKey)
    if err != nil {
        return "", fmt.Errorf("生成令牌失败: %w", err)
    }
    return tokenString, nil
}

// ValidateToken 验证JWT令牌
func ValidateToken(tokenString string) (*MyClaims, error) {
    claims := &MyClaims{}
    token, err := jwt.ParseWithClaims(tokenString, claims, func(token *jwt.Token) (interface{}, error) {
        // 检查签名方法是否是我们预期的
        if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
            return nil, fmt.Errorf("非法的签名方法: %v", token.Header["alg"])
        }
        return SecretKey, nil
    })

    if err != nil {
        // 这里可以细化错误处理，比如令牌过期、签名无效等
        if ve, ok := err.(*jwt.ValidationError); ok {
            if ve.Errors&jwt.ValidationErrorExpired != 0 {
                return nil, fmt.Errorf("令牌已过期")
            }
            if ve.Errors&jwt.ValidationErrorSignatureInvalid != 0 {
                return nil, fmt.Errorf("令牌签名无效")
            }
        }
        return nil, fmt.Errorf("验证令牌失败: %w", err)
    }

    if !token.Valid {
        return nil, fmt.Errorf("令牌无效")
    }

    return claims, nil
}

func main() {
    // 示例：生成令牌
    userID := "user123"
    username := "john_doe"
    token, err := GenerateToken(userID, username)
    if err != nil {
        log.Fatalf("生成令牌出错: %v", err)
    }
    fmt.Printf("生成的JWT令牌: %s\n", token)

    // 示例：验证令牌
    validatedClaims, err := ValidateToken(token)
    if err != nil {
        log.Fatalf("验证令牌出错: %v", err)
    }
    fmt.Printf("验证成功！用户ID: %s, 用户名: %s\n", validatedClaims.UserID, validatedClaims.Username)

    // 模拟令牌过期后的验证
    fmt.Println("\n模拟过期令牌验证...")
    // 故意将过期时间设为过去，用于测试
    expiredClaims := &MyClaims{
        UserID:   "expiredUser",
        Username: "expired",
        StandardClaims: jwt.StandardClaims{
            ExpiresAt: time.Now().Add(-1 * time.Hour).Unix(), // 设为过去
            IssuedAt:  time.Now().Unix(),
            Issuer:    "my-golang-app",
        },
    }
    expiredToken := jwt.NewWithClaims(jwt.SigningMethodHS256, expiredClaims)
    expiredTokenString, _ := expiredToken.SignedString(SecretKey)

    _, err = ValidateToken(expiredTokenString)
    if err != nil {
        fmt.Printf("验证过期令牌果然出错了: %v\n", err)
    }
}

jwt.NewWithClaims

token.SignedString

HS256

SecretKey

RS256

SecretKey

jwt-go

jwt.ParseWithClaims

jwt.ParseWithClaims

Keyfunc

if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok

StandardClaims

ExpiresAt

NotBefore

Issuer

jwt-go

ExpiresAt

NotBefore

Issuer

jwt.ValidationErrorExpired

jwt.ValidationErrorSignatureInvalid