tyc-server/common/jwt/jwtx.go

package jwtx

import (
	"errors"
	"strconv"
	"time"

	"github.com/golang-jwt/jwt/v4"
)

// Token 生成逻辑的函数，接收 userId、过期时间和密钥，返回生成的 token
func GenerateJwtToken(userId int64, secret string, expireTime int64) (string, error) {
	// 获取当前时间戳
	now := time.Now().Unix()
	// 定义 JWT Claims
	claims := jwt.MapClaims{
		"exp":    now + expireTime, // token 过期时间
		"iat":    now,              // 签发时间
		"userId": userId,           // 用户ID
	}

	// 创建新的 JWT token
	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)

	// 使用密钥对 token 签名
	signedToken, err := token.SignedString([]byte(secret))
	if err != nil {
		return "", err
	}

	return signedToken, nil
}
func ParseJwtToken(tokenStr string, secret string) (int64, error) {
	token, err := jwt.Parse(tokenStr, func(token *jwt.Token) (interface{}, error) {
		return []byte(secret), nil
	})

	if err != nil || !token.Valid {
		return 0, errors.New("invalid JWT")
	}

	claims, ok := token.Claims.(jwt.MapClaims)
	if !ok || !token.Valid {
		return 0, errors.New("invalid JWT claims")
	}

	// 从 claims 中提取 userId
	userIdRaw, ok := claims["userId"]
	if !ok {
		return 0, errors.New("userId not found in JWT")
	}

	// 处理不同类型的 userId，确保它被转换为 int64
	switch userId := userIdRaw.(type) {
	case float64:
		return int64(userId), nil
	case int64:
		return userId, nil
	case string:
		// 如果 userId 是字符串，可以尝试将其转换为 int64
		parsedId, err := strconv.ParseInt(userId, 10, 64)
		if err != nil {
			return 0, errors.New("invalid userId in JWT")
		}
		return parsedId, nil
	default:
		return 0, errors.New("unsupported userId type in JWT")
	}
}