59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
package ctxdata
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
)
|
|
|
|
const CtxKeyJwtUserId = "userId"
|
|
|
|
// 定义错误类型
|
|
var (
|
|
ErrNoUserIdInCtx = errors.New("上下文中没有用户ID") // 未登录
|
|
ErrInvalidUserId = errors.New("用户ID格式无效") // 数据异常
|
|
)
|
|
|
|
// GetUidFromCtx 从 context 中获取用户 ID
|
|
func GetUidFromCtx(ctx context.Context) (int64, error) {
|
|
// 尝试从上下文中获取 jwtUserId
|
|
value := ctx.Value(CtxKeyJwtUserId)
|
|
if value == nil {
|
|
return 0, ErrNoUserIdInCtx
|
|
}
|
|
|
|
// 根据值的类型进行不同处理
|
|
switch v := value.(type) {
|
|
case json.Number:
|
|
// 如果是 json.Number 类型，转换为 int64
|
|
uid, err := v.Int64()
|
|
if err != nil {
|
|
return 0, fmt.Errorf("%w: %v", ErrInvalidUserId, err)
|
|
}
|
|
return uid, nil
|
|
case int64:
|
|
// 如果已经是 int64 类型，直接返回
|
|
return v, nil
|
|
case float64:
|
|
// 有些JSON解析器可能会将数字解析为float64
|
|
return int64(v), nil
|
|
case int:
|
|
// 处理int类型
|
|
return int64(v), nil
|
|
default:
|
|
// 其他类型都视为无效
|
|
return 0, fmt.Errorf("%w: 期望类型 json.Number 或 int64, 实际类型 %T", ErrInvalidUserId, value)
|
|
}
|
|
}
|
|
|
|
// IsNoUserIdError 判断是否是未登录错误
|
|
func IsNoUserIdError(err error) bool {
|
|
return errors.Is(err, ErrNoUserIdInCtx)
|
|
}
|
|
|
|
// IsInvalidUserIdError 判断是否是用户ID格式错误
|
|
func IsInvalidUserIdError(err error) bool {
|
|
return errors.Is(err, ErrInvalidUserId)
|
|
}
|