first

common/jwt/jwtx.go

@@ -0,0 +1,68 @@
+package jwtx
+
+import (
+	"errors"
+	"github.com/golang-jwt/jwt/v4"
+	"strconv"
+	"time"
+)
+
+// 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")
+	}
+}