61 lines
1.4 KiB
Go
61 lines
1.4 KiB
Go
// 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()
|
|
}
|