49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
// Package jwtutil 封装 JWT Token 的签发与解析
|
||
package jwtutil
|
||
|
||
import (
|
||
"errors"
|
||
"time"
|
||
|
||
"github.com/golang-jwt/jwt/v5"
|
||
|
||
"nl-game-api-gin/internal/config"
|
||
)
|
||
|
||
// Claims 自定义 JWT 载荷：用户ID与角色
|
||
type Claims struct {
|
||
UserID int `json:"user_id"` // 用户ID
|
||
Role int `json:"role"` // 角色：1超管 2普通
|
||
jwt.RegisteredClaims
|
||
}
|
||
|
||
// Generate 为用户签发 Token
|
||
func Generate(userID, role int) (string, error) {
|
||
claims := Claims{
|
||
UserID: userID,
|
||
Role: role,
|
||
RegisteredClaims: jwt.RegisteredClaims{
|
||
// 过期时间从配置文件读取
|
||
ExpiresAt: jwt.NewNumericDate(time.Now().Add(time.Duration(config.C.JWT.ExpireHours) * time.Hour)),
|
||
IssuedAt: jwt.NewNumericDate(time.Now()),
|
||
},
|
||
}
|
||
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
||
return token.SignedString([]byte(config.C.JWT.Secret))
|
||
}
|
||
|
||
// Parse 解析并校验 Token，返回载荷
|
||
func Parse(tokenStr string) (*Claims, error) {
|
||
token, err := jwt.ParseWithClaims(tokenStr, &Claims{}, func(t *jwt.Token) (any, error) {
|
||
return []byte(config.C.JWT.Secret), nil
|
||
})
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
claims, ok := token.Claims.(*Claims)
|
||
if !ok || !token.Valid {
|
||
return nil, errors.New("无效的 Token")
|
||
}
|
||
return claims, nil
|
||
}
|