// Code generated by hertz generator. package auth import ( "cn/oneao/base-go/db" "cn/oneao/base-go/ent/userinfo" "cn/oneao/base-go/utils/jwtx" "cn/oneao/base-go/utils/response" "context" "fmt" "time" auth "cn/oneao/base-go/biz/model/auth" "github.com/cloudwego/hertz/pkg/app" "github.com/cloudwego/hertz/pkg/protocol/consts" ) // LoginDefault . // @router /auth/login/default [POST] func LoginDefault(ctx context.Context, c *app.RequestContext) { var err error var req auth.LoginDefaultReq err = c.BindAndValidate(&req) if err != nil { c.String(consts.StatusBadRequest, err.Error()) return } // 假设 ctx 已有,req.Account / req.Password 已有 start := time.Now() loginUser, err := db.Client.UserInfo. Query(). Where(userinfo.AccountEQ(req.Account), userinfo.PasswordEQ(req.Password)). Only(ctx) duration := time.Since(start) // 耗时 fmt.Printf("登录查询: (耗时: %v)\n", duration) if err != nil { c.String(consts.StatusInternalServerError, "账号或密码错误") return } token, err := jwtx.CreateAccessToken(loginUser.ID) refreshToken, err := jwtx.CreateRefreshToken(loginUser.ID) resp := new(auth.LoginDefaultResp) resp.Token = token resp.RefreshToken = refreshToken resp.UserId = loginUser.ID resp.Gender = loginUser.Gender resp.SeqNo = loginUser.SeqNo resp.Username = loginUser.Username response.Success(c).Data(resp).Send() }