refactor: 统一认证响应

This commit is contained in:
zetaloop
2026-04-03 17:46:08 +08:00
parent 3eb44d8a73
commit 5ff573f8fc
7 changed files with 22 additions and 51 deletions
@@ -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
} }
-4
View File
@@ -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"`
} }
-12
View File
@@ -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": [
-4
View File
@@ -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 {}