111 lines
2.7 KiB
Go
111 lines
2.7 KiB
Go
package system
|
|
|
|
import (
|
|
"context"
|
|
|
|
"management/internal/db/model/dto"
|
|
"management/internal/erpserver/model/system"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// UserStore 定义了 user 模块在 store 层所实现的方法.
|
|
type UserStore interface {
|
|
Create(ctx context.Context, obj *system.User) error
|
|
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
|
|
}
|
|
|
|
// UserExpansion 定义了用户操作的附加方法.
|
|
type UserExpansion interface{}
|
|
|
|
// userStore 是 UserStore 接口的实现.
|
|
type userStore struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// 确保 userStore 实现了 UserStore 接口.
|
|
var _ UserStore = (*userStore)(nil)
|
|
|
|
// NewUserStore 创建 userStore 的实例.
|
|
func NewUserStore(db *gorm.DB) *userStore {
|
|
return &userStore{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (s *userStore) Create(ctx context.Context, obj *system.User) error {
|
|
return s.db.WithContext(ctx).Create(obj).Error
|
|
}
|
|
|
|
func (s *userStore) Update(ctx context.Context, obj *system.User) error {
|
|
return s.db.WithContext(ctx).Save(obj).Error
|
|
}
|
|
|
|
func (s *userStore) Get(ctx context.Context, id int32) (*system.User, error) {
|
|
var user system.User
|
|
err := s.db.WithContext(ctx).Where("id = ?", id).First(&user).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func (s *userStore) GetByEmail(ctx context.Context, email string) (*system.User, error) {
|
|
var user system.User
|
|
err := s.db.WithContext(ctx).Where("email = ?", email).First(&user).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
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
|
|
}
|