hospital-admin-api/api/dao/adminUser.go

package dao

import (
	"gorm.io/gorm"
	"hospital-admin-api/api/model"
	"hospital-admin-api/api/requests"
	"hospital-admin-api/global"
)

type AdminUserDao struct {
}

// GetAdminUserFirstById 获取用户数据-用户id
// roleId 用户id
func (r *AdminUserDao) GetAdminUserFirstById(userId int64) (m *model.AdminUser, err error) {
	err = global.Db.First(&m, userId).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetAdminUserFirstByAccess 获取用户数据-用户账号
func (r *AdminUserDao) GetAdminUserFirstByAccess(access string) (m *model.AdminUser, err error) {
	err = global.Db.Where("access = ?", access).First(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetAdminUserFirstByPhone 获取用户数据-手机号
func (r *AdminUserDao) GetAdminUserFirstByPhone(phone string) (m *model.AdminUser, err error) {
	err = global.Db.Where("phone = ?", phone).First(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// DeleteAdminUser 删除用户
func (r *AdminUserDao) DeleteAdminUser(tx *gorm.DB, maps interface{}) error {
	err := tx.Where(maps).Delete(&model.AdminUser{}).Error
	if err != nil {
		return err
	}
	return nil
}

// DeleteAdminUserById 删除用户-用户id
func (r *AdminUserDao) DeleteAdminUserById(tx *gorm.DB, userId int64) error {
	if err := tx.Delete(&model.AdminUser{}, userId).Error; err != nil {
		return err
	}
	return nil
}

// EditAdminUser 修改用户
func (r *AdminUserDao) EditAdminUser(tx *gorm.DB, maps interface{}, data interface{}) error {
	err := tx.Model(&model.AdminUser{}).Where(maps).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// EditAdminUserById 修改角色-用户id
func (r *AdminUserDao) EditAdminUserById(tx *gorm.DB, userId int64, data interface{}) error {
	err := tx.Model(&model.AdminUser{}).Where("user_id = ?", userId).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// GetAdminUserList 获取用户列表
func (r *AdminUserDao) GetAdminUserList(maps interface{}) (m []*model.AdminUser, err error) {
	err = global.Db.Where(maps).Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetAdminUserPageSearch 获取用户列表-分页
func (r *AdminUserDao) GetAdminUserPageSearch(getUserPage requests.GetUserPage, page, pageSize int) (m []*model.AdminUser, total int64, err error) {
	var totalRecords int64

	// 构建查询条件
	query := global.Db.Model(&model.AdminUser{}).Omit("password", "salt")

	// 岗位
	query = query.Preload("Post", func(db *gorm.DB) *gorm.DB {
		return db.Select("post_id, post_name")
	})

	// 部门
	query = query.Preload("Dept", func(db *gorm.DB) *gorm.DB {
		return db.Select("dept_id, dept_name")
	})

	// 角色
	query = query.Preload("Role", func(db *gorm.DB) *gorm.DB {
		return db.Select("role_id, role_name")
	})

	if getUserPage.PostName != "" {
		subQuery := global.Db.Model(&model.AdminPost{}).
			Select("post_id").
			Where("post_name LIKE ?", "%"+getUserPage.PostName+"%")

		query = query.Where(gorm.Expr("post_id IN (?)", subQuery))
	}

	if getUserPage.DeptName != "" {
		query = query.Preload("Dept", "dept_name like ?", "%"+getUserPage.DeptName+"%")
		subQuery := global.Db.Model(&model.AdminDept{}).
			Select("dept_id").
			Where("dept_name LIKE ?", "%"+getUserPage.DeptName+"%")

		query = query.Where(gorm.Expr("dept_id IN (?)", subQuery))
	}

	if getUserPage.NickName != "" {
		query = query.Where("nick_name = ?", getUserPage.NickName)
	}

	if getUserPage.Phone != "" {
		query = query.Where("phone = ?", getUserPage.Phone)
	}

	if getUserPage.Status != 0 {
		query = query.Where("status = ?", getUserPage.Status)
	}

	if getUserPage.DeptId != "" {
		query = query.Where("dept_id = ?", getUserPage.DeptId)
	}

	if getUserPage.PostId != "" {
		query = query.Where("post_id = ?", getUserPage.PostId)
	}

	// 查询总数量
	if err := query.Count(&totalRecords).Error; err != nil {
		return nil, 0, err
	}

	err = query.Scopes(model.Paginate(page, pageSize)).Find(&m).Error
	if err != nil {
		return nil, 0, err
	}
	return m, totalRecords, nil
}

// AddAdminUser 新增用户
func (r *AdminUserDao) AddAdminUser(tx *gorm.DB, model *model.AdminUser) (*model.AdminUser, error) {
	if err := tx.Create(model).Error; err != nil {
		return nil, err
	}
	return model, nil
}