167 lines
4.0 KiB
Go
167 lines
4.0 KiB
Go
/**
|
||
* package utils
|
||
*
|
||
* JWT Token生成和验证工具包
|
||
*
|
||
* 功能概述：
|
||
* 1. 生成JWT Token（包含用户ID和过期时间）
|
||
* 2. 解析JWT Token（验证签名和过期时间）
|
||
* 3. 验证Token有效性（提取用户ID）
|
||
*
|
||
* 使用场景：
|
||
* - 用户登录后生成Token
|
||
* - API请求时验证Token
|
||
* - 从Token中提取用户信息
|
||
*/
|
||
package utils
|
||
|
||
import (
|
||
"errors"
|
||
"time"
|
||
|
||
"github.com/golang-jwt/jwt/v5"
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
// jwtSecret JWT签名密钥（从配置文件读取）
|
||
var jwtSecret []byte
|
||
|
||
/**
|
||
* init
|
||
*
|
||
* 功能：初始化JWT签名密钥
|
||
*
|
||
* 步骤：
|
||
* 1. 从配置文件读取JWT密钥
|
||
* 2. 如果配置文件中没有，使用默认密钥（仅用于开发环境）
|
||
* 3. 将密钥转换为字节数组存储
|
||
*
|
||
* 注意：生产环境必须使用配置文件中的强随机密钥
|
||
*/
|
||
func init() {
|
||
secret := viper.GetString("jwt.secret")
|
||
if secret == "" {
|
||
secret = "xk-websocket-secret-key-2025" // 默认密钥，生产环境应使用配置
|
||
}
|
||
jwtSecret = []byte(secret)
|
||
}
|
||
|
||
/**
|
||
* Claims
|
||
*
|
||
* JWT Token的载荷结构
|
||
*
|
||
* 字段说明：
|
||
* - UserID: 用户ID（业务数据）
|
||
* - RegisteredClaims: JWT标准声明（过期时间、签发时间等）
|
||
*/
|
||
type Claims struct {
|
||
UserID string `json:"user_id"` // 用户ID
|
||
jwt.RegisteredClaims // JWT标准声明
|
||
}
|
||
|
||
/**
|
||
* GenerateToken
|
||
*
|
||
* 功能：生成JWT Token
|
||
*
|
||
* 步骤：
|
||
* 1. 设置Token过期时间（默认7天）
|
||
* 2. 创建Claims对象，包含用户ID和标准声明
|
||
* 3. 使用HS256算法签名Token
|
||
* 4. 返回Token字符串
|
||
*
|
||
* @param userID 用户ID
|
||
* @returns token字符串和错误
|
||
*/
|
||
func GenerateToken(userID string) (string, error) {
|
||
// 步骤1: 设置Token过期时间（7天后过期）
|
||
expirationTime := time.Now().Add(7 * 24 * time.Hour)
|
||
|
||
// 步骤2: 创建Claims对象，包含用户ID和标准声明
|
||
claims := &Claims{
|
||
UserID: userID, // 业务数据：用户ID
|
||
RegisteredClaims: jwt.RegisteredClaims{
|
||
ExpiresAt: jwt.NewNumericDate(expirationTime), // 过期时间
|
||
IssuedAt: jwt.NewNumericDate(time.Now()), // 签发时间
|
||
NotBefore: jwt.NewNumericDate(time.Now()), // 生效时间（立即生效）
|
||
},
|
||
}
|
||
|
||
// 步骤3: 使用HS256算法创建Token并签名
|
||
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
||
tokenString, err := token.SignedString(jwtSecret)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
|
||
// 步骤4: 返回Token字符串
|
||
return tokenString, nil
|
||
}
|
||
|
||
/**
|
||
* ParseToken
|
||
*
|
||
* 功能：解析JWT Token
|
||
*
|
||
* 步骤：
|
||
* 1. 创建空的Claims对象
|
||
* 2. 使用密钥解析Token并验证签名
|
||
* 3. 检查Token是否有效（签名正确、未过期）
|
||
* 4. 返回Claims对象
|
||
*
|
||
* @param tokenString token字符串
|
||
* @returns Claims和错误
|
||
*/
|
||
func ParseToken(tokenString string) (*Claims, error) {
|
||
// 步骤1: 创建空的Claims对象
|
||
claims := &Claims{}
|
||
|
||
// 步骤2: 解析Token并验证签名
|
||
// 使用密钥验证Token的签名是否有效
|
||
token, err := jwt.ParseWithClaims(tokenString, claims, func(token *jwt.Token) (interface{}, error) {
|
||
return jwtSecret, nil // 返回签名密钥
|
||
})
|
||
|
||
if err != nil {
|
||
return nil, err // 解析失败（签名错误、格式错误等）
|
||
}
|
||
|
||
// 步骤3: 检查Token是否有效（签名正确、未过期）
|
||
if !token.Valid {
|
||
return nil, errors.New("invalid token")
|
||
}
|
||
|
||
// 步骤4: 返回解析后的Claims对象
|
||
return claims, nil
|
||
}
|
||
|
||
/**
|
||
* ValidateToken
|
||
*
|
||
* 功能：验证Token有效性并提取用户ID
|
||
*
|
||
* 步骤：
|
||
* 1. 调用ParseToken解析Token
|
||
* 2. 如果解析成功，从Claims中提取用户ID
|
||
* 3. 返回用户ID
|
||
*
|
||
* 使用场景：
|
||
* - 中间件中验证Token
|
||
* - API处理器中获取当前用户ID
|
||
*
|
||
* @param tokenString token字符串
|
||
* @returns 用户ID和错误
|
||
*/
|
||
func ValidateToken(tokenString string) (string, error) {
|
||
// 步骤1: 解析Token
|
||
claims, err := ParseToken(tokenString)
|
||
if err != nil {
|
||
return "", err // Token无效或已过期
|
||
}
|
||
|
||
// 步骤2: 从Claims中提取用户ID
|
||
return claims.UserID, nil
|
||
}
|
||
|