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并验证它。