126 lines
2.8 KiB
Go
126 lines
2.8 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"taotie-api/common"
|
||
"taotie-api/core"
|
||
"taotie-api/model/do/userdo"
|
||
"taotie-api/model/po"
|
||
"taotie-api/repo"
|
||
"taotie-api/utils/sctx"
|
||
"taotie-api/utils/sjwt"
|
||
"time"
|
||
|
||
"github.com/gookit/goutil/errorx"
|
||
)
|
||
|
||
type UserService struct {
|
||
cfg *core.Configuration
|
||
userRepo *repo.UserRepo[*po.TUser]
|
||
tenantRepo *repo.TenantRepo[*po.TTenant]
|
||
}
|
||
|
||
func NewUserService(mdb *repo.MongoDb, cfg *core.Configuration, userRepo *repo.UserRepo[*po.TUser], tenantRepo *repo.TenantRepo[*po.TTenant]) *UserService {
|
||
return &UserService{
|
||
cfg,
|
||
userRepo,
|
||
tenantRepo,
|
||
}
|
||
}
|
||
|
||
// Login 登录
|
||
func (s *UserService) Login(ctx context.Context, in *userdo.LoginIn) (*userdo.LoginOut, error) {
|
||
// 验证租户
|
||
ttenant, err := s.tenantRepo.FindOneByTenantNo(ctx, in.TenantNo)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if ttenant == nil {
|
||
return nil, common.ErrUserTenantNotFound
|
||
}
|
||
|
||
// 验证用户
|
||
tuser, err := s.userRepo.FindOneByTenantIdAndUserName(ctx, ttenant.Id(), in.UserName)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if tuser == nil {
|
||
return nil, common.ErrUserNotFound
|
||
}
|
||
if tuser.Password != in.Password {
|
||
return nil, common.ErrUserPasswordError
|
||
}
|
||
|
||
return &userdo.LoginOut{
|
||
TenantId: ttenant.Id(),
|
||
UserId: tuser.Id(),
|
||
UserName: tuser.UserName,
|
||
}, nil
|
||
}
|
||
|
||
// GenToken 生成token
|
||
func (s *UserService) GenToken(ctx context.Context, in *userdo.GenTokenIn) (*userdo.GenTokenOut, error) {
|
||
token, err := sjwt.GenerateToken(in.TenantNo, in.UserId, in.UserName, s.cfg.JWT.TimeOut, s.cfg.JWT.SignString)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return &userdo.GenTokenOut{
|
||
Token: token,
|
||
}, nil
|
||
}
|
||
|
||
func (s *UserService) Create(ctx context.Context, in *userdo.CreateIn) (*userdo.CreateOut, error) {
|
||
// 查重：租户名称
|
||
ttenant, err := s.tenantRepo.FindOneByTenantName(ctx, in.TenantName)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if ttenant != nil {
|
||
return nil, errorx.With(common.ErrSysDuplicate, "租户名称重复")
|
||
}
|
||
|
||
// 先设置临时用户到 context（避免 BaseRepo.Create 中 cuser 为 nil 导致 panic）
|
||
sctx.SetCurrentUser(ctx, &sctx.CurrentUser{
|
||
UserId: "",
|
||
UserName: in.UserName,
|
||
TenantId: "",
|
||
})
|
||
|
||
// 生成 tenantNo（基于时间戳）
|
||
tenantNo := fmt.Sprintf("T%d", time.Now().UnixMilli())
|
||
|
||
// 创建租户
|
||
ttenant = &po.TTenant{
|
||
TenantNo: tenantNo,
|
||
TenantName: in.TenantName,
|
||
}
|
||
tenantId, err := s.tenantRepo.Create(ctx, ttenant)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 更新 context 中的 TenantId
|
||
sctx.SetCurrentUser(ctx, &sctx.CurrentUser{
|
||
UserId: "",
|
||
UserName: in.UserName,
|
||
TenantId: tenantId,
|
||
})
|
||
|
||
// 创建用户
|
||
tuser := &po.TUser{
|
||
UserName: in.UserName,
|
||
Password: in.Password,
|
||
}
|
||
userId, err := s.userRepo.Create(ctx, tuser)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return &userdo.CreateOut{
|
||
TenantId: tenantId,
|
||
UserId: userId,
|
||
UserName: in.UserName,
|
||
}, nil
|
||
}
|