promote

common/jwt/jwtx.go

@@ -2,9 +2,10 @@ package jwtx
 
 import (
 	"errors"
-	"github.com/golang-jwt/jwt/v4"
 	"strconv"
 	"time"
+
+	"github.com/golang-jwt/jwt/v4"
 )
 
 // Token 生成逻辑的函数，接收 userId、过期时间和密钥，返回生成的 token

common/jwt/jwtxV2.go

@@ -0,0 +1,64 @@
+package jwtx
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+
+	"github.com/golang-jwt/jwt/v4"
+)
+
+// Token 生成逻辑的函数，接收 userId、过期时间和密钥，返回生成的 token
+func GenerateJwtTokenV2(claims map[string]interface{}, secret string, expireTime int64) (string, error) {
+	// 获取当前时间戳
+	now := time.Now().Unix()
+	// 定义 JWT Claims
+	claimsMap := jwt.MapClaims{
+		"exp":    now + expireTime, // token 过期时间
+		"iat":    now,              // 签发时间
+		"claims": claims,           // 自定义claims
+	}
+
+	// 创建新的 JWT token
+	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claimsMap)
+
+	// 使用密钥对 token 签名
+	signedToken, err := token.SignedString([]byte(secret))
+	if err != nil {
+		return "", err
+	}
+
+	return signedToken, nil
+}
+func ParseJwtTokenV2(tokenStr string, secret string) (map[string]interface{}, error) {
+	token, err := jwt.Parse(tokenStr, func(token *jwt.Token) (interface{}, error) {
+		return []byte(secret), nil
+	})
+
+	if err != nil || !token.Valid {
+		return nil, errors.New("invalid JWT")
+	}
+
+	claims, ok := token.Claims.(jwt.MapClaims)
+	if !ok || !token.Valid {
+		return nil, errors.New("invalid JWT claims")
+	}
+
+	// 从 claims 中提取 userId
+	claimsRaw, ok := claims["claims"]
+	if !ok {
+		return nil, errors.New("claims not found in JWT")
+	}
+
+	return claimsRaw.(map[string]interface{}), nil
+}
+
+func GetJwtClaims(ctx context.Context) (map[string]interface{}, error) {
+	// 尝试从上下文中获取 jwtUserId
+	claims, ok := ctx.Value("claims").(map[string]interface{})
+	if !ok {
+		return nil, fmt.Errorf("无法获取jwt claims: %v", claims)
+	}
+	return claims, nil
+}