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

302 lines
7.7 KiB
Go

package dao
import (
"errors"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"hospital-admin-api/api/model"
"hospital-admin-api/api/requests"
"hospital-admin-api/global"
"strings"
"time"
)
type UserPatientDao struct {
}
// GetUserPatientById 获取患者数据-患者id
func (r *UserPatientDao) GetUserPatientById(patientId int64) (m *model.UserPatient, err error) {
err = global.Db.First(&m, patientId).Error
if err != nil {
return nil, err
}
return m, nil
}
// GetUserPatientPreloadById 获取患者数据-加载全部关联-患者id
func (r *UserPatientDao) GetUserPatientPreloadById(patientId int64) (m *model.UserPatient, err error) {
err = global.Db.Preload(clause.Associations).First(&m, patientId).Error
if err != nil {
return nil, err
}
return m, nil
}
// DeleteUserPatient 删除患者
func (r *UserPatientDao) DeleteUserPatient(tx *gorm.DB, maps interface{}) error {
err := tx.Where(maps).Delete(&model.UserPatient{}).Error
if err != nil {
return err
}
return nil
}
// DeleteUserPatientById 删除患者-患者id
func (r *UserPatientDao) DeleteUserPatientById(tx *gorm.DB, patientId int64) error {
if err := tx.Delete(&model.UserPatient{}, patientId).Error; err != nil {
return err
}
return nil
}
// EditUserPatient 修改患者
func (r *UserPatientDao) EditUserPatient(tx *gorm.DB, maps interface{}, data interface{}) error {
err := tx.Model(&model.UserPatient{}).Where(maps).Updates(data).Error
if err != nil {
return err
}
return nil
}
// EditUserPatientById 修改患者-患者id
func (r *UserPatientDao) EditUserPatientById(tx *gorm.DB, patientId int64, data interface{}) error {
err := tx.Model(&model.UserPatient{}).Where("patient_id = ?", patientId).Updates(data).Error
if err != nil {
return err
}
return nil
}
// GetUserPatientList 获取患者列表
func (r *UserPatientDao) GetUserPatientList(maps interface{}) (m []*model.UserPatient, err error) {
err = global.Db.Where(maps).Find(&m).Error
if err != nil {
return nil, err
}
return m, nil
}
// GetUserPatientPageSearch 获取患者列表-分页
func (r *UserPatientDao) GetUserPatientPageSearch(req requests.GetUserPatientPage, page, pageSize int) (m []*model.UserPatient, total int64, err error) {
var totalRecords int64
// 构建查询条件
query := global.Db.Model(&model.UserPatient{}).Omit("open_id", "union_id", "wx_session_key")
// 用户表
query = query.Preload("User", func(db *gorm.DB) *gorm.DB {
return db.Omit("user_password", "salt")
})
// 家庭成员表
query = query.Preload("PatientFamily")
// 手机号
if req.Mobile != "" {
subQuery := global.Db.Model(&model.User{}).
Select("user_id").
Where("mobile = ?", req.Mobile)
query = query.Where(gorm.Expr("user_id IN (?)", subQuery))
}
// 用户名称-用户-就诊人
if req.UserName != "" {
// 用户
query = query.Where("user_name LIKE ?", "%"+req.UserName+"%")
// 就诊人
patientFamilySubQuery := global.Db.Model(&model.PatientFamily{}).
Select("patient_id").
Where("card_name LIKE ?", "%"+req.UserName+"%")
query = query.Or("patient_id IN (?)", patientFamilySubQuery).Or(query)
}
// 用户状态
if req.Status != nil {
query = query.Where("status = ?", req.Status)
}
// 注册时间
if req.CreatedAt != "" {
cancelTime := strings.Split(req.CreatedAt, "&")
if len(cancelTime) == 2 {
startTime, _ := time.Parse("2006-01-02", cancelTime[0])
endTime, _ := time.Parse("2006-01-02", cancelTime[1])
if startTime == endTime {
endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
}
query = query.Where("created_at BETWEEN ? AND ?", startTime, endTime)
}
}
// 排序
query = query.Order("created_at desc")
// 查询总数量
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
}
// AddUserPatient 新增患者
func (r *UserPatientDao) AddUserPatient(tx *gorm.DB, model *model.UserPatient) (*model.UserPatient, error) {
if err := tx.Create(model).Error; err != nil {
return nil, err
}
return model, nil
}
// GetUserPatientExportListSearch 获取患者列表-导出
func (r *UserPatientDao) GetUserPatientExportListSearch(req requests.UserPatientExportList) (m []*model.UserPatient, err error) {
// 构建查询条件
query := global.Db.Model(&model.UserPatient{}).Omit("open_id", "union_id", "wx_session_key")
// 用户表
query = query.Preload("User", func(db *gorm.DB) *gorm.DB {
return db.Omit("user_password", "salt")
})
// 家庭成员表
query = query.Preload("PatientFamily")
// 当前搜索数据
if req.Type == 1 {
// 手机号
if req.Mobile != "" {
subQuery := global.Db.Model(&model.User{}).
Select("user_id").
Where("mobile = ?", req.Mobile)
query = query.Where(gorm.Expr("user_id IN (?)", subQuery))
}
// 用户名称-用户-就诊人
if req.UserName != "" {
// 用户
query = query.Where("user_name LIKE ?", "%"+req.UserName+"%")
// 就诊人
patientFamilySubQuery := global.Db.Model(&model.PatientFamily{}).
Select("patient_id").
Where("card_name LIKE ?", "%"+req.UserName+"%")
query = query.Or("patient_id IN (?)", patientFamilySubQuery).Or(query)
}
// 用户状态
if req.Status != nil {
query = query.Where("status = ?", req.Status)
}
// 注册时间
if req.CreatedAt != "" {
cancelTime := strings.Split(req.CreatedAt, "&")
if len(cancelTime) == 2 {
startTime, _ := time.Parse("2006-01-02", cancelTime[0])
endTime, _ := time.Parse("2006-01-02", cancelTime[1])
if startTime == endTime {
endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
}
query = query.Where("created_at BETWEEN ? AND ?", startTime, endTime)
}
}
}
// 当前选中数据
if req.Type == 2 {
if req.Id == "" {
return nil, errors.New("未提供需导出数据编号")
}
id := strings.Split(req.Id, ",")
query = query.Where("patient_id IN (?)", id)
}
// 排序
query = query.Order("created_at desc")
err = query.Find(&m).Error
if err != nil {
return nil, err
}
return m, nil
}
// GetUserPatientByUserId 获取患者数据-id
func (r *UserPatientDao) GetUserPatientByUserId(userId int64) (m *model.UserPatient, err error) {
err = global.Db.Where("user_id = ?", userId).First(&m).Error
if err != nil {
return nil, err
}
return m, nil
}
// GetUserPatientListSearch 获取患者列表-限制条数
func (r *UserPatientDao) GetUserPatientListSearch(req requests.GetUserPatientList) (m []*model.UserPatient, err error) {
// 构建查询条件
query := global.Db.Model(&model.UserPatient{}).Omit("open_id", "union_id", "wx_session_key")
// 用户
query = query.Preload("User", func(db *gorm.DB) *gorm.DB {
return db.Omit("user_password", "salt")
})
// 用户名称
if req.UserName != "" {
query = query.Where("user_name LIKE ?", "%"+req.UserName+"%")
}
// 手机号
if req.Mobile != "" {
subQuery := global.Db.Model(&model.User{}).
Select("user_id").
Where("mobile = ?", req.Mobile)
query = query.Where(gorm.Expr("user_id IN (?)", subQuery))
}
// 用户状态
if req.Status != nil {
query = query.Where("status = ?", req.Status)
}
// 注册时间
if req.CreatedAt != "" {
cancelTime := strings.Split(req.CreatedAt, "&")
if len(cancelTime) == 2 {
startTime, _ := time.Parse("2006-01-02", cancelTime[0])
endTime, _ := time.Parse("2006-01-02", cancelTime[1])
if startTime == endTime {
endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
}
query = query.Where("created_at BETWEEN ? AND ?", startTime, endTime)
}
}
// 排序
query = query.Order("created_at desc")
err = query.Limit(10).Find(&m).Error
if err != nil {
return nil, err
}
return m, nil
}