You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
1.6 KiB
75 lines
1.6 KiB
package logic
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
|
|
"healthapi/internal/model"
|
|
"healthapi/internal/svc"
|
|
"healthapi/internal/types"
|
|
"healthapi/pkg/errorx"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type GetUserProfileLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewGetUserProfileLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetUserProfileLogic {
|
|
return &GetUserProfileLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *GetUserProfileLogic) GetUserProfile() (resp *types.UserInfo, err error) {
|
|
// 从 context 获取用户ID（JWT中间件设置）
|
|
userID, err := GetUserIDFromCtx(l.ctx)
|
|
if err != nil {
|
|
return nil, errorx.ErrUnauthorized
|
|
}
|
|
|
|
var user model.User
|
|
if err := l.svcCtx.DB.First(&user, userID).Error; err != nil {
|
|
return nil, errorx.ErrUserNotFound
|
|
}
|
|
|
|
return &types.UserInfo{
|
|
ID: uint(user.ID),
|
|
Phone: user.Phone,
|
|
Email: user.Email,
|
|
Nickname: user.Nickname,
|
|
Avatar: user.Avatar,
|
|
SurveyCompleted: user.SurveyCompleted,
|
|
}, nil
|
|
}
|
|
|
|
// GetUserIDFromCtx 从上下文获取用户ID
|
|
func GetUserIDFromCtx(ctx context.Context) (uint, error) {
|
|
// go-zero JWT 中间件将用户信息存储在 context 中
|
|
val := ctx.Value("user_id")
|
|
if val == nil {
|
|
return 0, errorx.ErrUnauthorized
|
|
}
|
|
|
|
// 根据 JWT payload 类型进行转换
|
|
switch v := val.(type) {
|
|
case float64:
|
|
return uint(v), nil
|
|
case int64:
|
|
return uint(v), nil
|
|
case int:
|
|
return uint(v), nil
|
|
case uint:
|
|
return v, nil
|
|
case json.Number:
|
|
id, _ := v.Int64()
|
|
return uint(id), nil
|
|
default:
|
|
return 0, errorx.ErrUnauthorized
|
|
}
|
|
}
|
|
|