三端用户手机号联通，增加临时用户

common/ctxdata/ctxData.go

@@ -5,14 +5,16 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"qnc-server/app/main/model"
+	jwtx "qnc-server/common/jwt"
 )
 
 const CtxKeyJwtUserId = "userId"
 
 // 定义错误类型
 var (
-	ErrNoUserIdInCtx = errors.New("上下文中没有用户ID") // 未登录
-	ErrInvalidUserId = errors.New("用户ID格式无效")   // 数据异常
+	ErrNoInCtx       = errors.New("上下文中没有相关数据")
+	ErrInvalidUserId = errors.New("用户ID格式无效") // 数据异常
 )
 
 // GetUidFromCtx 从 context 中获取用户 ID
@@ -20,7 +22,11 @@ func GetUidFromCtx(ctx context.Context) (int64, error) {
 	// 尝试从上下文中获取 jwtUserId
 	value := ctx.Value(CtxKeyJwtUserId)
 	if value == nil {
-		return 0, ErrNoUserIdInCtx
+		claims, err := GetClaimsFromCtx(ctx)
+		if err != nil {
+			return 0, err
+		}
+		return claims.UserId, nil
 	}
 
 	// 根据值的类型进行不同处理
@@ -47,12 +53,52 @@ func GetUidFromCtx(ctx context.Context) (int64, error) {
 	}
 }
 
+func GetClaimsFromCtx(ctx context.Context) (*jwtx.JwtClaims, error) {
+	value := ctx.Value(jwtx.ExtraKey)
+	if value == nil {
+		return nil, ErrNoInCtx
+	}
+
+	// 首先尝试直接断言为 *jwtx.JwtClaims
+	if claims, ok := value.(*jwtx.JwtClaims); ok {
+		return claims, nil
+	}
+
+	// 如果直接断言失败，尝试从 map[string]interface{} 中解析
+	if claimsMap, ok := value.(map[string]interface{}); ok {
+		return jwtx.MapToJwtClaims(claimsMap)
+	}
+
+	return nil, ErrNoInCtx
+}
+
 // IsNoUserIdError 判断是否是未登录错误
 func IsNoUserIdError(err error) bool {
-	return errors.Is(err, ErrNoUserIdInCtx)
+	return errors.Is(err, ErrNoInCtx)
 }
 
 // IsInvalidUserIdError 判断是否是用户ID格式错误
 func IsInvalidUserIdError(err error) bool {
 	return errors.Is(err, ErrInvalidUserId)
 }
+
+// GetPlatformFromCtx 从 context 中获取平台
+func GetPlatformFromCtx(ctx context.Context) (string, error) {
+	platform, platformOk := ctx.Value("platform").(string)
+	if !platformOk {
+		return "", fmt.Errorf("平台不存在: %s", platform)
+	}
+
+	switch platform {
+	case model.PlatformWxMini:
+		return model.PlatformWxMini, nil
+	case model.PlatformWxH5:
+		return model.PlatformWxH5, nil
+	case model.PlatformApp:
+		return model.PlatformApp, nil
+	case model.PlatformH5:
+		return model.PlatformH5, nil
+	default:
+		return "", fmt.Errorf("不支持的支付平台: %s", platform)
+	}
+}