118 lines
2.7 KiB
Go
118 lines
2.7 KiB
Go
package system
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"time"
|
|
|
|
"management/internal/db/model/dto"
|
|
db "management/internal/db/sqlc"
|
|
"management/internal/erpserver/model/req"
|
|
"management/internal/global/know"
|
|
"management/internal/pkg/crypto"
|
|
"management/internal/pkg/session"
|
|
)
|
|
|
|
// UserBiz 定义处理用户请求所需的方法.
|
|
type UserBiz interface {
|
|
Create(ctx context.Context, req *db.CreateSysUserParams) (*db.SysUser, error)
|
|
|
|
UserExpansion
|
|
}
|
|
|
|
// UserExpansion 定义用户操作的扩展方法.
|
|
type UserExpansion interface {
|
|
Login(ctx context.Context, req *req.Login) error
|
|
}
|
|
|
|
// userBiz 是 UserBiz 接口的实现.
|
|
type userBiz struct {
|
|
store db.Store
|
|
session session.ISession
|
|
}
|
|
|
|
// 确保 userBiz 实现了 UserBiz 接口.
|
|
var _ UserBiz = (*userBiz)(nil)
|
|
|
|
func NewUser(store db.Store, session session.ISession) *userBiz {
|
|
return &userBiz{
|
|
store: store,
|
|
session: session,
|
|
}
|
|
}
|
|
|
|
func (b *userBiz) Create(ctx context.Context, req *db.CreateSysUserParams) (*db.SysUser, error) {
|
|
return b.store.CreateSysUser(ctx, req)
|
|
}
|
|
|
|
func (b *userBiz) Login(ctx context.Context, req *req.Login) error {
|
|
log := &db.CreateSysUserLoginLogParams{
|
|
CreatedAt: time.Now(),
|
|
Email: req.Email,
|
|
IsSuccess: false,
|
|
RefererUrl: req.Referrer,
|
|
Url: req.Url,
|
|
Os: req.Os,
|
|
Ip: req.Ip,
|
|
Browser: req.Browser,
|
|
}
|
|
|
|
user, err := b.store.GetSysUserByEmail(ctx, req.Email)
|
|
if err != nil {
|
|
log.Message = err.Error()
|
|
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
|
return err
|
|
}
|
|
log.UserUuid = user.Uuid
|
|
log.Username = user.Username
|
|
|
|
err = crypto.BcryptComparePassword(user.HashedPassword, req.Password+user.Salt)
|
|
if err != nil {
|
|
log.Message = "compare password failed"
|
|
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
|
return errors.New(log.Message)
|
|
}
|
|
|
|
// 登陆成功
|
|
|
|
if user.RoleID == 0 {
|
|
log.Message = "账号没有配置角色, 请联系管理员"
|
|
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
|
return errors.New(log.Message)
|
|
}
|
|
|
|
sysRole, err := b.store.GetSysRole(ctx, user.RoleID)
|
|
if err != nil {
|
|
log.Message = "账号配置的角色错误, 请联系管理员"
|
|
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
|
return errors.New(log.Message)
|
|
}
|
|
|
|
auth := dto.AuthorizeUser{
|
|
ID: user.ID,
|
|
Uuid: user.Uuid,
|
|
Email: user.Email,
|
|
Username: user.Username,
|
|
RoleID: sysRole.ID,
|
|
RoleName: sysRole.Name,
|
|
OS: log.Os,
|
|
IP: log.Ip,
|
|
Browser: log.Browser,
|
|
}
|
|
|
|
gob, err := json.Marshal(auth)
|
|
if err != nil {
|
|
log.Message = err.Error()
|
|
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
|
return err
|
|
}
|
|
|
|
b.session.Put(ctx, know.StoreName, gob)
|
|
|
|
log.IsSuccess = true
|
|
log.Message = "登陆成功"
|
|
_ = b.store.CreateSysUserLoginLog(ctx, log)
|
|
return nil
|
|
}
|