手机号适配双端

common/jwt/jwtx.go

@@ -1,68 +1,94 @@
 package jwtx
 
 import (
+	"encoding/json"
 	"errors"
-	"github.com/golang-jwt/jwt/v4"
-	"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
+const ExtraKey = "extra"
+
+type JwtClaims struct {
+	UserId   int64  `json:"userId"`
+	AgentId  int64  `json:"agentId"`
+	Platform string `json:"platform"`
+	// 用户身份类型：0-临时用户，1-正式用户
+	UserType int64 `json:"userType"`
+	// 是否代理：0-否，1-是
+	IsAgent int64 `json:"isAgent"`
+}
+
+// MapToJwtClaims 将 map[string]interface{} 转换为 JwtClaims 结构体
+func MapToJwtClaims(claimsMap map[string]interface{}) (*JwtClaims, error) {
+	// 使用JSON序列化/反序列化的方式自动转换
+	jsonData, err := json.Marshal(claimsMap)
+	if err != nil {
+		return nil, errors.New("序列化claims失败")
 	}
 
-	// 创建新的 JWT token
-	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
+	var claims JwtClaims
+	if err := json.Unmarshal(jsonData, &claims); err != nil {
+		return nil, errors.New("反序列化claims失败")
+	}
 
-	// 使用密钥对 token 签名
-	signedToken, err := token.SignedString([]byte(secret))
+	return &claims, nil
+}
+
+// GenerateJwtToken 生成JWT token
+func GenerateJwtToken(claims JwtClaims, secret string, expire int64) (string, error) {
+	now := time.Now().Unix()
+
+	// 将 claims 结构体转换为 map[string]interface{}
+	claimsBytes, err := json.Marshal(claims)
 	if err != nil {
 		return "", err
 	}
 
-	return signedToken, nil
+	var claimsMap map[string]interface{}
+	if err := json.Unmarshal(claimsBytes, &claimsMap); err != nil {
+		return "", err
+	}
+
+	jwtClaims := jwt.MapClaims{
+		"exp":    now + expire,
+		"iat":    now,
+		"userId": claims.UserId,
+		ExtraKey: claimsMap,
+	}
+	token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwtClaims)
+	return token.SignedString([]byte(secret))
 }
-func ParseJwtToken(tokenStr string, secret string) (int64, error) {
+
+func ParseJwtToken(tokenStr string, secret string) (*JwtClaims, 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")
+		return nil, errors.New("invalid JWT")
 	}
 
 	claims, ok := token.Claims.(jwt.MapClaims)
 	if !ok || !token.Valid {
-		return 0, errors.New("invalid JWT claims")
+		return nil, errors.New("invalid JWT claims")
 	}
 
-	// 从 claims 中提取 userId
-	userIdRaw, ok := claims["userId"]
-	if !ok {
-		return 0, errors.New("userId not found in JWT")
+	extraInfo, exists := claims[ExtraKey]
+	if !exists {
+		return nil, errors.New("extra 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")
+	// 尝试直接断言为 JwtClaims 结构体
+	if jwtClaims, ok := extraInfo.(JwtClaims); ok {
+		return &jwtClaims, nil
 	}
+
+	// 尝试从 map[string]interface{} 中解析
+	if claimsMap, ok := extraInfo.(map[string]interface{}); ok {
+		return MapToJwtClaims(claimsMap)
+	}
+
+	return nil, errors.New("unsupported extra type in JWT")
 }