v2

2025-03-28 17:51:34 +08:00
parent da612380e0
commit 5c8802d2f0
68 changed files with 3422 additions and 630 deletions
--- a/internal/erpserver/biz/v1/system/user.go
+++ b/internal/erpserver/biz/v1/system/user.go
@@ -0,0 +1,117 @@
+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
+}