You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
2.1 KiB
94 lines
2.1 KiB
package model
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var (
|
|
ErrNotFound = errors.New("record not found")
|
|
)
|
|
|
|
// Insert 插入用户
|
|
func Insert(ctx context.Context, db *gorm.DB, user *User) (int64, error) {
|
|
result := db.WithContext(ctx).Create(user)
|
|
if result.Error != nil {
|
|
return 0, result.Error
|
|
}
|
|
return user.Id, nil
|
|
}
|
|
|
|
// FindOne 根据ID查询用户
|
|
func FindOne(ctx context.Context, db *gorm.DB, id int64) (*User, error) {
|
|
var user User
|
|
result := db.WithContext(ctx).First(&user, id)
|
|
if result.Error != nil {
|
|
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
|
return nil, ErrNotFound
|
|
}
|
|
return nil, result.Error
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
// FindOneByEmail 根据邮箱查询用户
|
|
func FindOneByEmail(ctx context.Context, db *gorm.DB, email string) (*User, error) {
|
|
var user User
|
|
result := db.WithContext(ctx).Where("email = ?", email).First(&user)
|
|
if result.Error != nil {
|
|
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
|
return nil, ErrNotFound
|
|
}
|
|
return nil, result.Error
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
// Update 更新用户
|
|
func Update(ctx context.Context, db *gorm.DB, user *User) error {
|
|
result := db.WithContext(ctx).Save(user)
|
|
return result.Error
|
|
}
|
|
|
|
// Delete 删除用户
|
|
func Delete(ctx context.Context, db *gorm.DB, id int64) error {
|
|
result := db.WithContext(ctx).Delete(&User{}, id)
|
|
return result.Error
|
|
}
|
|
|
|
// FindList 查询用户列表
|
|
func FindList(ctx context.Context, db *gorm.DB, page, pageSize int64, keyword string, status int64) ([]User, int64, error) {
|
|
var users []User
|
|
var total int64
|
|
|
|
query := db.WithContext(ctx).Model(&User{})
|
|
|
|
// 关键字搜索
|
|
if keyword != "" {
|
|
query = query.Where("username LIKE ? OR email LIKE ? OR phone LIKE ?",
|
|
"%"+keyword+"%", "%"+keyword+"%", "%"+keyword+"%")
|
|
}
|
|
|
|
// 状态筛选
|
|
if status > 0 {
|
|
query = query.Where("status = ?", status)
|
|
}
|
|
|
|
// 统计总数
|
|
if err := query.Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
// 分页查询
|
|
offset := (page - 1) * pageSize
|
|
if offset < 0 {
|
|
offset = 0
|
|
}
|
|
err := query.Offset(int(offset)).Limit(int(pageSize)).Find(&users).Error
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
return users, total, nil
|
|
}
|
|
|