refactor: 统一认证响应
This commit is contained in:
@@ -23,12 +23,10 @@ func LoginHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
|
|
||||||
l := auth.NewLoginLogic(r.Context(), svcCtx)
|
l := auth.NewLoginLogic(r.Context(), svcCtx)
|
||||||
resp, err := l.Login(&req)
|
resp, token, err := l.Login(&req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
httpx.ErrorCtx(r.Context(), w, err)
|
httpx.ErrorCtx(r.Context(), w, err)
|
||||||
} else {
|
} else {
|
||||||
token := resp.RefreshToken
|
|
||||||
resp.RefreshToken = ""
|
|
||||||
http.SetCookie(w, &http.Cookie{
|
http.SetCookie(w, &http.Cookie{
|
||||||
Name: "JToken",
|
Name: "JToken",
|
||||||
Value: token,
|
Value: token,
|
||||||
|
|||||||
@@ -23,12 +23,10 @@ func RegisterHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
|
|
||||||
l := auth.NewRegisterLogic(r.Context(), svcCtx)
|
l := auth.NewRegisterLogic(r.Context(), svcCtx)
|
||||||
resp, err := l.Register(&req)
|
resp, token, err := l.Register(&req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
httpx.ErrorCtx(r.Context(), w, err)
|
httpx.ErrorCtx(r.Context(), w, err)
|
||||||
} else {
|
} else {
|
||||||
token := resp.RefreshToken
|
|
||||||
resp.RefreshToken = ""
|
|
||||||
http.SetCookie(w, &http.Cookie{
|
http.SetCookie(w, &http.Cookie{
|
||||||
Name: "JToken",
|
Name: "JToken",
|
||||||
Value: token,
|
Value: token,
|
||||||
|
|||||||
@@ -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
|
// todo: add your logic here and delete this line
|
||||||
if len(req.Username) < 3 || len(req.Password) < 8 || len(req.Password) > 20 {
|
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{
|
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)
|
logx.Infof("res:%v", res)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logx.Errorf("rpc login err: %v", err)
|
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 == "" {
|
if res == nil || res.Id <= 0 || res.Username == "" || res.Token == "" {
|
||||||
logx.Errorf("rpc login returned empty payload, username=%s, resp=%+v", req.Username, res)
|
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{
|
return &types.LoginResp{
|
||||||
AccessToken: "",
|
|
||||||
RefreshToken: res.Token,
|
|
||||||
User: types.User{},
|
User: types.User{},
|
||||||
}, nil
|
}, res.Token, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -36,33 +36,33 @@ func NewRegisterLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Register
|
|||||||
|
|
||||||
var usernameRegex = regexp.MustCompile("^[a-zA-Z0-9_]+$")
|
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{
|
existingUser, err := l.svcCtx.UserRpc.GetUserByUsername(l.ctx, &pb.GetUserByUsernameReq{
|
||||||
Username: req.Username,
|
Username: req.Username,
|
||||||
})
|
})
|
||||||
if len(req.Username) < 3 {
|
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 {
|
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) {
|
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 {
|
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)
|
hashedPassword, err := pwdUtils.HashPassword(req.Password)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, errors.New("hash password failed")
|
return nil, "", errors.New("hash password failed")
|
||||||
}
|
}
|
||||||
|
|
||||||
requestId, err := contextj.RIdFrom(l.ctx)
|
requestId, err := contextj.RIdFrom(l.ctx)
|
||||||
logx.Infof("requestId: %s", requestId)
|
logx.Infof("requestId: %s", requestId)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logx.Errorf("contextj.RequestIdFrom failed: %v", err)
|
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{
|
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 {
|
if err != nil {
|
||||||
logx.Error("failed to register user: ", err)
|
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{
|
return &types.RegisterResp{
|
||||||
AccessToken: "",
|
|
||||||
RefreshToken: res.Res,
|
|
||||||
User: types.User{},
|
User: types.User{},
|
||||||
}, nil
|
}, res.Res, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -48,8 +48,6 @@ type LoginReq struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type LoginResp struct {
|
type LoginResp struct {
|
||||||
AccessToken string `json:"accessToken"`
|
|
||||||
RefreshToken string `json:"refreshToken"`
|
|
||||||
User User `json:"user"`
|
User User `json:"user"`
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -72,8 +70,6 @@ type RegisterReq struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type RegisterResp struct {
|
type RegisterResp struct {
|
||||||
AccessToken string `json:"accessToken"`
|
|
||||||
RefreshToken string `json:"refreshToken"`
|
|
||||||
User User `json:"user"`
|
User User `json:"user"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -302,12 +302,6 @@
|
|||||||
"schema": {
|
"schema": {
|
||||||
"type": "object",
|
"type": "object",
|
||||||
"properties": {
|
"properties": {
|
||||||
"accessToken": {
|
|
||||||
"type": "string"
|
|
||||||
},
|
|
||||||
"refreshToken": {
|
|
||||||
"type": "string"
|
|
||||||
},
|
|
||||||
"user": {
|
"user": {
|
||||||
"type": "object",
|
"type": "object",
|
||||||
"required": [
|
"required": [
|
||||||
@@ -447,12 +441,6 @@
|
|||||||
"schema": {
|
"schema": {
|
||||||
"type": "object",
|
"type": "object",
|
||||||
"properties": {
|
"properties": {
|
||||||
"accessToken": {
|
|
||||||
"type": "string"
|
|
||||||
},
|
|
||||||
"refreshToken": {
|
|
||||||
"type": "string"
|
|
||||||
},
|
|
||||||
"user": {
|
"user": {
|
||||||
"type": "object",
|
"type": "object",
|
||||||
"required": [
|
"required": [
|
||||||
|
|||||||
@@ -95,8 +95,6 @@ type (
|
|||||||
Vcode string `json:"vcode,omitempty"` // 验证码
|
Vcode string `json:"vcode,omitempty"` // 验证码
|
||||||
}
|
}
|
||||||
RegisterResp {
|
RegisterResp {
|
||||||
AccessToken string `json:"accessToken"`
|
|
||||||
RefreshToken string `json:"refreshToken"`
|
|
||||||
User User `json:"user"`
|
User User `json:"user"`
|
||||||
}
|
}
|
||||||
LoginReq {
|
LoginReq {
|
||||||
@@ -106,8 +104,6 @@ type (
|
|||||||
Remember bool `json:"remember,optional"`
|
Remember bool `json:"remember,optional"`
|
||||||
}
|
}
|
||||||
LoginResp {
|
LoginResp {
|
||||||
AccessToken string `json:"accessToken"`
|
|
||||||
RefreshToken string `json:"refreshToken"`
|
|
||||||
User User `json:"user"`
|
User User `json:"user"`
|
||||||
}
|
}
|
||||||
LogoutReq {}
|
LogoutReq {}
|
||||||
|
|||||||
Reference in New Issue
Block a user