2

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