refactor: 统一认证响应
This commit is contained in:
@@ -29,10 +29,10 @@ func NewLoginLogic(ctx context.Context, svcCtx *svc.ServiceContext) *LoginLogic
|
||||
}
|
||||
}
|
||||
|
||||
func (l *LoginLogic) Login(req *types.LoginReq) (resp *types.LoginResp, err error) {
|
||||
func (l *LoginLogic) Login(req *types.LoginReq) (*types.LoginResp, string, error) {
|
||||
// todo: add your logic here and delete this line
|
||||
if len(req.Username) < 3 || len(req.Password) < 8 || len(req.Password) > 20 {
|
||||
return nil, errors.New("the information is illegal")
|
||||
return nil, "", errors.New("the information is illegal")
|
||||
}
|
||||
|
||||
res, err := l.svcCtx.UserRpc.Login(l.ctx, &usercenter.LoginReq{
|
||||
@@ -42,17 +42,15 @@ func (l *LoginLogic) Login(req *types.LoginReq) (resp *types.LoginResp, err erro
|
||||
logx.Infof("res:%v", res)
|
||||
if err != nil {
|
||||
logx.Errorf("rpc login err: %v", err)
|
||||
return nil, errors.New("login fail")
|
||||
return nil, "", errors.New("login fail")
|
||||
}
|
||||
|
||||
if res == nil || res.Id <= 0 || res.Username == "" || res.Token == "" {
|
||||
logx.Errorf("rpc login returned empty payload, username=%s, resp=%+v", req.Username, res)
|
||||
return nil, errors.New("login fail")
|
||||
return nil, "", errors.New("login fail")
|
||||
}
|
||||
|
||||
return &types.LoginResp{
|
||||
AccessToken: "",
|
||||
RefreshToken: res.Token,
|
||||
User: types.User{},
|
||||
}, nil
|
||||
User: types.User{},
|
||||
}, res.Token, nil
|
||||
}
|
||||
|
||||
@@ -36,33 +36,33 @@ func NewRegisterLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Register
|
||||
|
||||
var usernameRegex = regexp.MustCompile("^[a-zA-Z0-9_]+$")
|
||||
|
||||
func (l *RegisterLogic) Register(req *types.RegisterReq) (resp *types.RegisterResp, err error) {
|
||||
func (l *RegisterLogic) Register(req *types.RegisterReq) (*types.RegisterResp, string, error) {
|
||||
existingUser, err := l.svcCtx.UserRpc.GetUserByUsername(l.ctx, &pb.GetUserByUsernameReq{
|
||||
Username: req.Username,
|
||||
})
|
||||
if len(req.Username) < 3 {
|
||||
return nil, errors.New("username must be at least 3 characters long")
|
||||
return nil, "", errors.New("username must be at least 3 characters long")
|
||||
}
|
||||
if len(req.Username) > 20 {
|
||||
return nil, errors.New("username must be at most 20 characters long")
|
||||
return nil, "", errors.New("username must be at most 20 characters long")
|
||||
}
|
||||
if !usernameRegex.MatchString(req.Username) {
|
||||
return nil, errors.New("username can only contain letters, numbers, and underscores")
|
||||
return nil, "", errors.New("username can only contain letters, numbers, and underscores")
|
||||
}
|
||||
if err == nil && existingUser != nil {
|
||||
return nil, errors.New("user already exists")
|
||||
return nil, "", errors.New("user already exists")
|
||||
}
|
||||
|
||||
hashedPassword, err := pwdUtils.HashPassword(req.Password)
|
||||
if err != nil {
|
||||
return nil, errors.New("hash password failed")
|
||||
return nil, "", errors.New("hash password failed")
|
||||
}
|
||||
|
||||
requestId, err := contextj.RIdFrom(l.ctx)
|
||||
logx.Infof("requestId: %s", requestId)
|
||||
if err != nil {
|
||||
logx.Errorf("contextj.RequestIdFrom failed: %v", err)
|
||||
return nil, errors.New("contextj.RequestIdFrom failed")
|
||||
return nil, "", errors.New("contextj.RequestIdFrom failed")
|
||||
}
|
||||
|
||||
res, err := l.svcCtx.UserRpc.Register(l.ctx, &usercenter.RegisterReq{
|
||||
@@ -75,13 +75,10 @@ func (l *RegisterLogic) Register(req *types.RegisterReq) (resp *types.RegisterRe
|
||||
})
|
||||
if err != nil {
|
||||
logx.Error("failed to register user: ", err)
|
||||
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()))
|
||||
}
|
||||
|
||||
// 返回响应
|
||||
return &types.RegisterResp{
|
||||
AccessToken: "",
|
||||
RefreshToken: res.Res,
|
||||
User: types.User{},
|
||||
}, nil
|
||||
User: types.User{},
|
||||
}, res.Res, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user