Go Gin使用JWT实现认证机制

package main
 
import (
    "fmt"
    "github.com/dgrijalva/jwt-go"
    "time"
)
 
// 定义一个Claims结构体用于存储JWT的相关信息
type CustomClaims struct {
    ID       int
    Username string
    jwt.StandardClaims
}
 
// 生成JWT token
func GenerateToken(id int, username string) (string, error) {
    // 创建Claims
    claims := CustomClaims{
        id,
        username,
        jwt.StandardClaims{
            ExpiresAt: time.Now().Add(time.Hour * 2).Unix(), // 设置过期时间
            Issuer:    "gin-jwt-example",                    // 签发者
        },
    }
 
    // 创建token
    token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
 
    // 对token进行签名
    signedToken, err := token.SignedString([]byte("secret_key"))
    if err != nil {
        return "", err
    }
 
    return signedToken, nil
}
 
// 验证JWT token
func ValidateToken(tokenString string) (*CustomClaims, error) {
    token, err := jwt.ParseWithClaims(tokenString, &CustomClaims{}, func(token *jwt.Token) (interface{}, error) {
        return []byte("secret_key"), nil
    })
    if err != nil {
        return nil, err
    }
 
    if claims, ok := token.Claims.(*CustomClaims); ok && token.Valid {
        return claims, nil
    }
 
    return nil, fmt.Errorf("invalid token")
}
 
func main() {
    // 生成token
    tokenString, err := GenerateToken(1, "user1")
    if err != nil {
        panic(err)
    }
    fmt.Printf("Generated Token: %s\n", tokenString)
 
    // 验证token
    claims, err := ValidateToken(tokenString)
    if err != nil {
        panic(err)
    }
    fmt.Printf("Validated Token. ID: %d, Username: %s\n", claims.ID, claims.Username)
}

这段代码展示了如何在Go语言中使用jwt-go库生成和验证JWT token。首先定义了一个CustomClaims结构体来存储额外的信息，然后实现了生成和验证token的函数。最后，在main函数中演示了如何生成一个token并验证它。