2

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
+}