fix: 登录和注册返回完整用户信息

This commit is contained in:
zetaloop
2026-04-03 19:12:19 +08:00
parent 15c940a3e9
commit 7ccb97b2e2
2 changed files with 54 additions and 2 deletions
@@ -7,11 +7,14 @@ import (
"context" "context"
"errors" "errors"
"juwan-backend/app/users/rpc/usercenter" "juwan-backend/app/users/rpc/usercenter"
"time"
"juwan-backend/app/users/api/internal/svc" "juwan-backend/app/users/api/internal/svc"
"juwan-backend/app/users/api/internal/types" "juwan-backend/app/users/api/internal/types"
"github.com/jinzhu/copier"
"github.com/zeromicro/go-zero/core/logx" "github.com/zeromicro/go-zero/core/logx"
"k8s.io/apimachinery/pkg/util/json"
) )
type LoginLogic struct { type LoginLogic struct {
@@ -50,7 +53,29 @@ func (l *LoginLogic) Login(req *types.LoginReq) (*types.LoginResp, string, error
return nil, "", errors.New("login fail") return nil, "", errors.New("login fail")
} }
userResp, err := l.svcCtx.UserRpc.GetUsersById(l.ctx, &usercenter.GetUsersByIdReq{Id: res.Id})
if err != nil {
logx.Errorf("GetUsersById err: %v", err)
return nil, "", errors.New("get user failed")
}
user := &types.User{}
err = copier.Copy(user, userResp.Users)
if err != nil {
logx.Errorf("copier.Copy err: %v", err)
return nil, "", errors.New("copy user failed")
}
var verificationStatus map[string]string
err = json.Unmarshal([]byte(userResp.Users.VerificationStatus), &verificationStatus)
if err != nil {
logx.Errorf("json.Unmarshal err: %v", err)
}
user.VerifiedRoles = userResp.Users.VerifiedRoles
user.VerificationStatus = verificationStatus
user.Role = userResp.Users.CurrentRole
user.CreatedAt = time.Unix(userResp.Users.CreatedAt, 0).Format(time.DateTime)
return &types.LoginResp{ return &types.LoginResp{
User: types.User{}, User: *user,
}, res.Token, nil }, res.Token, nil
} }
@@ -12,11 +12,14 @@ import (
"juwan-backend/common/utils/contextj" "juwan-backend/common/utils/contextj"
"juwan-backend/common/utils/pwdUtils" "juwan-backend/common/utils/pwdUtils"
"regexp" "regexp"
"time"
"juwan-backend/app/users/api/internal/svc" "juwan-backend/app/users/api/internal/svc"
"juwan-backend/app/users/api/internal/types" "juwan-backend/app/users/api/internal/types"
"github.com/jinzhu/copier"
"github.com/zeromicro/go-zero/core/logx" "github.com/zeromicro/go-zero/core/logx"
"k8s.io/apimachinery/pkg/util/json"
) )
type RegisterLogic struct { type RegisterLogic struct {
@@ -78,7 +81,31 @@ func (l *RegisterLogic) Register(req *types.RegisterReq) (*types.RegisterResp, s
return nil, "", errors.New(fmt.Sprintf("failed to register user: %v", err.Error())) return nil, "", errors.New(fmt.Sprintf("failed to register user: %v", err.Error()))
} }
userResp, err := l.svcCtx.UserRpc.GetUserByUsername(l.ctx, &pb.GetUserByUsernameReq{
Username: req.Username,
})
if err != nil {
logx.Errorf("GetUserByUsername err: %v", err)
return nil, "", errors.New("get user failed")
}
user := &types.User{}
err = copier.Copy(user, userResp.Users)
if err != nil {
logx.Errorf("copier.Copy err: %v", err)
return nil, "", errors.New("copy user failed")
}
var verificationStatus map[string]string
err = json.Unmarshal([]byte(userResp.Users.VerificationStatus), &verificationStatus)
if err != nil {
logx.Errorf("json.Unmarshal err: %v", err)
}
user.VerifiedRoles = userResp.Users.VerifiedRoles
user.VerificationStatus = verificationStatus
user.Role = userResp.Users.CurrentRole
user.CreatedAt = time.Unix(userResp.Users.CreatedAt, 0).Format(time.DateTime)
return &types.RegisterResp{ return &types.RegisterResp{
User: types.User{}, User: *user,
}, res.Res, nil }, res.Res, nil
} }