2

internal/erpserver/store/expansion.go

@@ -0,0 +1,24 @@
+package store
+
+import "management/internal/erpserver/store/system"
+
+// ======================================================================================
+// system
+
+type SystemExpansion interface {
+	User() system.UserStore
+	LoginLog() system.LoginLogStore
+	Role() system.RoleStore
+}
+
+func (store *datastore) User() system.UserStore {
+	return system.NewUserStore(store.core)
+}
+
+func (store *datastore) LoginLog() system.LoginLogStore {
+	return system.NewLoginLogStore(store.core)
+}
+
+func (store *datastore) Role() system.RoleStore {
+	return system.NewRoleStore(store.core)
+}

internal/erpserver/store/store.go

@@ -4,8 +4,6 @@ import (
 	"context"
 	"sync"
 
-	"management/internal/erpserver/store/system"
-
 	"gorm.io/gorm"
 )
 
@@ -20,7 +18,7 @@ type IStore interface {
 	DB(ctx context.Context) *gorm.DB
 	TX(ctx context.Context, fn func(ctx context.Context) error) error
 
-	User() system.UserStore
+	SystemExpansion
 }
 
 // transactionKey 用于在 context.Context 中存储事务上下文的键.
@@ -62,8 +60,3 @@ func (store *datastore) TX(ctx context.Context, fn func(ctx context.Context) err
 		},
 	)
 }
-
-// Users 返回一个实现了 UserStore 接口的实例.
-func (store *datastore) User() system.UserStore {
-	return system.NewUserStore(store.core)
-}

internal/erpserver/store/system/login_log.go

@@ -0,0 +1,58 @@
+package system
+
+import (
+	"context"
+
+	"management/internal/db/model/dto"
+	"management/internal/erpserver/model/system"
+
+	"gorm.io/gorm"
+)
+
+type LoginLogStore interface {
+	Create(ctx context.Context, obj *system.LoginLog) error
+	List(ctx context.Context, q dto.SearchDto) ([]*system.LoginLog, int64, error)
+}
+
+type loginLogStore struct {
+	db *gorm.DB
+}
+
+var _ LoginLogStore = (*loginLogStore)(nil)
+
+func NewLoginLogStore(db *gorm.DB) *loginLogStore {
+	return &loginLogStore{
+		db: db,
+	}
+}
+
+func (s *loginLogStore) Create(ctx context.Context, obj *system.LoginLog) error {
+	return s.db.WithContext(ctx).Create(obj).Error
+}
+
+func (s *loginLogStore) List(ctx context.Context, q dto.SearchDto) ([]*system.LoginLog, int64, error) {
+	query := s.db.WithContext(ctx).
+		Model(&system.LoginLog{}).
+		Where("created_at BETWEEN ? AND ?", q.SearchTimeBegin, q.SearchTimeEnd)
+	if q.SearchEmail != "" {
+		query = query.Where("email LIKE ?", "%"+q.SearchEmail+"%")
+	}
+
+	var count int64
+	err := query.Count(&count).Error
+	if err != nil {
+		return nil, 0, err
+	}
+
+	var logs []*system.LoginLog
+	err = query.
+		Order("id DESC").
+		Offset((q.Page - 1) * q.Rows).
+		Limit(q.Rows).
+		Find(&logs).
+		Error
+	if err != nil {
+		return nil, 0, err
+	}
+	return logs, count, nil
+}

internal/erpserver/store/system/role.go

@@ -0,0 +1,44 @@
+package system
+
+import (
+	"context"
+
+	"management/internal/erpserver/model/system"
+
+	"gorm.io/gorm"
+)
+
+type RoleStore interface {
+	Create(ctx context.Context, obj *system.Role) error
+	Update(ctx context.Context, obj *system.Role) error
+	Get(ctx context.Context, id int32) (*system.Role, error)
+}
+
+type roleStore struct {
+	db *gorm.DB
+}
+
+var _ RoleStore = (*roleStore)(nil)
+
+func NewRoleStore(db *gorm.DB) *roleStore {
+	return &roleStore{
+		db: db,
+	}
+}
+
+func (s *roleStore) Create(ctx context.Context, obj *system.Role) error {
+	return s.db.WithContext(ctx).Create(obj).Error
+}
+
+func (s *roleStore) Update(ctx context.Context, obj *system.Role) error {
+	return s.db.WithContext(ctx).Save(obj).Error
+}
+
+func (s *roleStore) Get(ctx context.Context, id int32) (*system.Role, error) {
+	var role system.Role
+	err := s.db.WithContext(ctx).Where("id =?", id).First(&role).Error
+	if err != nil {
+		return nil, err
+	}
+	return &role, nil
+}

internal/erpserver/store/system/user.go

@@ -3,6 +3,7 @@ package system
 import (
 	"context"
 
+	"management/internal/db/model/dto"
 	"management/internal/erpserver/model/system"
 
 	"gorm.io/gorm"
@@ -14,6 +15,8 @@ type UserStore interface {
 	Update(ctx context.Context, obj *system.User) error
 	Get(ctx context.Context, id int32) (*system.User, error)
 	GetByEmail(ctx context.Context, email string) (*system.User, error)
+	All(ctx context.Context) ([]*system.User, error)
+	List(ctx context.Context, q dto.SearchDto) ([]*system.User, int64, error)
 
 	UserExpansion
 }
@@ -61,3 +64,47 @@ func (s *userStore) GetByEmail(ctx context.Context, email string) (*system.User,
 	}
 	return &user, nil
 }
+
+func (s *userStore) All(ctx context.Context) ([]*system.User, error) {
+	var users []*system.User
+	err := s.db.WithContext(ctx).Find(&users).Error
+	if err != nil {
+		return nil, err
+	}
+	return users, nil
+}
+
+func (s *userStore) List(ctx context.Context, q dto.SearchDto) ([]*system.User, int64, error) {
+	query := s.db.WithContext(ctx).
+		Model(&system.User{}).
+		Where("created_at BETWEEN? AND?", q.SearchTimeBegin, q.SearchTimeEnd)
+	if q.SearchID != 0 {
+		query = query.Where("id = ?", q.SearchID)
+	}
+	if q.SearchName != "" {
+		query = query.Where("username LIKE ?", "%"+q.SearchName+"%")
+	}
+	if q.SearchEmail != "" {
+		query = query.Where("email LIKE ?", "%"+q.SearchEmail+"%")
+	}
+	if q.SearchStatus != 9999 {
+		query = query.Where("status = ?", q.SearchStatus)
+	}
+
+	var count int64
+	err := query.Count(&count).Error
+	if err != nil {
+		return nil, 0, err
+	}
+
+	var users []*system.User
+	err = query.
+		Order("id DESC").
+		Offset((q.Page - 1) * q.Rows).
+		Limit(q.Rows).
+		Find(&users).Error
+	if err != nil {
+		return nil, 0, err
+	}
+	return users, count, nil
+}