103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
"server/internal/consts"
|
|
"server/internal/dao"
|
|
"server/internal/model"
|
|
"server/internal/model/do"
|
|
"server/internal/model/entity"
|
|
"server/internal/service"
|
|
"server/utility/ecode"
|
|
"server/utility/jwt"
|
|
|
|
"github.com/gogf/gf/v2/os/gtime"
|
|
"github.com/gogf/gf/v2/util/grand"
|
|
)
|
|
|
|
type sUser struct{}
|
|
|
|
func init() {
|
|
service.RegisterUser(New())
|
|
}
|
|
|
|
func New() service.IUser {
|
|
return &sUser{}
|
|
}
|
|
|
|
func (s *sUser) Login(ctx context.Context, in *model.UserLoginIn) (out *model.UserLoginOut, err error) {
|
|
// 根据 OpenId 查找用户
|
|
exist, err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Exist()
|
|
if err != nil {
|
|
return nil, ecode.Fail.Sub("查找用户失败")
|
|
}
|
|
|
|
var userId int64
|
|
if !exist {
|
|
// 用户不存在，创建新用户
|
|
user := &entity.Users{
|
|
WxOpenId: in.OpenId,
|
|
Username: grand.Digits(10),
|
|
FirstVisitAt: gtime.Now(),
|
|
}
|
|
result, err := dao.Users.Ctx(ctx).Insert(user)
|
|
if err != nil {
|
|
return nil, ecode.Fail.Sub("创建用户失败")
|
|
}
|
|
userId, err = result.LastInsertId()
|
|
if err != nil {
|
|
return nil, ecode.Fail.Sub("获取用户ID失败")
|
|
}
|
|
} else {
|
|
// 用户存在，更新最后登录时间
|
|
var user entity.Users
|
|
if err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId}).Scan(&user); err != nil {
|
|
return nil, ecode.Fail.Sub("查找用户失败")
|
|
}
|
|
userId = user.Id
|
|
if _, err := dao.Users.Ctx(ctx).Where(do.Users{Id: userId}).Update(do.Users{LastLoginAt: gtime.Now()}); err != nil {
|
|
return nil, ecode.Fail.Sub("更新登录时间失败")
|
|
}
|
|
}
|
|
|
|
// 生成 token
|
|
token, err := jwt.GenerateToken(&jwt.TokenIn{
|
|
UserId: userId,
|
|
Permission: consts.UserPermission,
|
|
})
|
|
if err != nil {
|
|
return nil, ecode.Fail.Sub("生成token失败")
|
|
}
|
|
|
|
out = &model.UserLoginOut{
|
|
Token: token,
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *sUser) Info(ctx context.Context, in *model.UserInfoIn) (out *model.UserInfoOut, err error) {
|
|
exist, err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId, Id: in.Id}).OmitEmptyWhere().Exist()
|
|
if err != nil {
|
|
return nil, ecode.Fail.Sub("查找用户失败")
|
|
}
|
|
if !exist {
|
|
return nil, ecode.Params.Sub("用户不存在")
|
|
}
|
|
var user entity.Users
|
|
if err := dao.Users.Ctx(ctx).Where(do.Users{WxOpenId: in.OpenId, Id: in.Id}).OmitEmptyWhere().Scan(&user); err != nil {
|
|
return nil, ecode.Fail.Sub("查找用户失败")
|
|
}
|
|
out = &model.UserInfoOut{
|
|
Id: user.Id,
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *sUser) Update(ctx context.Context, in *model.UserUpdateIn) (out *model.UpdateOut, err error) {
|
|
return
|
|
}
|
|
|
|
func (s *sUser) BindPhone(ctx context.Context, in *model.UserBindPhoneIn) (out *model.UpdateOut, err error) {
|
|
return
|
|
}
|