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 OrderInquiryDao struct {
}

// GetOrderInquiryById 获取问诊订单数据-问诊订单id
func (r *OrderInquiryDao) GetOrderInquiryById(orderInquiryId int64) (m *model.OrderInquiry, err error) {
	err = global.Db.First(&m, orderInquiryId).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetOrderInquiryPreloadById 获取问诊订单数据-加载全部关联-问诊订单id
func (r *OrderInquiryDao) GetOrderInquiryPreloadById(orderInquiryId int64) (m *model.OrderInquiry, err error) {
	err = global.Db.Preload(clause.Associations).First(&m, orderInquiryId).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// DeleteOrderInquiry 删除问诊订单
func (r *OrderInquiryDao) DeleteOrderInquiry(tx *gorm.DB, maps interface{}) error {
	err := tx.Where(maps).Delete(&model.OrderInquiry{}).Error
	if err != nil {
		return err
	}
	return nil
}

// EditOrderInquiry 修改问诊订单
func (r *OrderInquiryDao) EditOrderInquiry(tx *gorm.DB, maps interface{}, data interface{}) error {
	err := tx.Model(&model.OrderInquiry{}).Where(maps).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// EditOrderInquiryById 修改问诊订单-问诊订单id
func (r *OrderInquiryDao) EditOrderInquiryById(tx *gorm.DB, orderInquiryId int64, data interface{}) error {
	err := tx.Model(&model.OrderInquiry{}).Where("order_inquiry_id = ?", orderInquiryId).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// GetOrderInquiryList 获取问诊订单列表
func (r *OrderInquiryDao) GetOrderInquiryList(maps interface{}) (m []*model.OrderInquiry, err error) {
	err = global.Db.Where(maps).Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// AddOrderInquiry 新增问诊订单
func (r *OrderInquiryDao) AddOrderInquiry(tx *gorm.DB, model *model.OrderInquiry) (*model.OrderInquiry, error) {
	if err := tx.Create(model).Error; err != nil {
		return nil, err
	}
	return model, nil
}

// GetOrderInquiryPageSearch 获取问诊订单列表-分页
func (r *OrderInquiryDao) GetOrderInquiryPageSearch(req requests.GetOrderInquiryPage, page, pageSize int) (m []*model.OrderInquiry, total int64, err error) {
	var totalRecords int64

	// 构建查询条件
	query := global.Db.Model(&model.OrderInquiry{})

	// 医生
	query = query.Preload("UserDoctor", func(db *gorm.DB) *gorm.DB {
		return db.Omit("open_id", "union_id", "wx_session_key")
	})

	// 医生姓名
	if req.DoctorName != "" {
		subQuery := global.Db.Model(&model.UserDoctor{}).
			Select("doctor_id").
			Where("user_name LIKE ?", "%"+req.DoctorName+"%")

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

	// 用户
	query = query.Preload("User", func(db *gorm.DB) *gorm.DB {
		return db.Select("user_id", "user_name", "mobile")
	})

	// 订单类型
	if req.InquiryType != nil {
		query = query.Where("inquiry_type = ?", req.InquiryType)
	}

	// 订单问诊方式
	if req.InquiryMode != nil {
		query = query.Where("inquiry_mode = ?", req.InquiryMode)
	}

	// 问诊订单状态
	if req.InquiryStatus != nil {
		query = query.Where("inquiry_status = ?", req.InquiryStatus)
	}

	// 问诊订单退款状态
	if req.InquiryRefundStatus != nil {
		query = query.Where("inquiry_refund_status = ?", req.InquiryRefundStatus)
	}

	// 支付渠道
	if req.InquiryPayChannel != nil {
		query = query.Where("inquiry_pay_channel = ?", req.InquiryPayChannel)
	}

	// 支付状态
	if req.InquiryPayStatus != nil {
		query = query.Where("inquiry_pay_status = ?", req.InquiryPayStatus)
	}

	// 系统订单编号
	if req.InquiryNo != "" {
		query = query.Where("inquiry_no = ?", req.InquiryNo)
	}

	// 第三方支付流水号
	if req.EscrowTradeNo != "" {
		query = query.Where("escrow_trade_no = ?", req.EscrowTradeNo)
	}

	// 支付时间
	if req.PayTime != "" {
		payTime := strings.Split(req.PayTime, "&")
		if len(payTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", payTime[0])
			endTime, _ := time.Parse("2006-01-02", payTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("pay_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 接诊时间
	if req.ReceptionTime != "" {
		receptionTime := strings.Split(req.ReceptionTime, "&")
		if len(receptionTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", receptionTime[0])
			endTime, _ := time.Parse("2006-01-02", receptionTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("reception_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 订单完成时间
	if req.CompleteTime != "" {
		completeTime := strings.Split(req.CompleteTime, "&")
		if len(completeTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", completeTime[0])
			endTime, _ := time.Parse("2006-01-02", completeTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("complete_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 订单结束时间
	if req.FinishTime != "" {
		finishTime := strings.Split(req.FinishTime, "&")
		if len(finishTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", finishTime[0])
			endTime, _ := time.Parse("2006-01-02", finishTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("finish_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 是否提现
	if req.IsWithdrawal != nil {
		query = query.Where("is_withdrawal = ?", req.IsWithdrawal)
	}

	// 订单取消时间
	if req.CancelTime != "" {
		cancelTime := strings.Split(req.CancelTime, "&")
		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("cancel_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 创建时间
	if req.CreatedAt != "" {
		createdAt := strings.Split(req.CreatedAt, "&")
		if len(createdAt) == 2 {
			startTime, _ := time.Parse("2006-01-02", createdAt[0])
			endTime, _ := time.Parse("2006-01-02", createdAt[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.PatientName != "" {
		query = query.Where("patient_name LIKE ?", "%"+req.PatientName+"%")
	}

	// 手机号-医生/患者
	if req.Mobile != "" {
		// 患者
		patientSubQuery := global.Db.Model(&model.User{}).
			Select("user_id").
			Where("mobile = ?", req.Mobile)

		// 医生
		doctorUserSubQuery := global.Db.Model(&model.User{}).
			Select("user_id").
			Where("mobile = ?", req.Mobile)

		doctorSubQuery := global.Db.Model(&model.UserDoctor{}).
			Select("doctor_id").
			Where(gorm.Expr("user_id IN (?)", doctorUserSubQuery))

		query = query.Where("user_id IN (?)", patientSubQuery).Or("doctor_id IN (?)", doctorSubQuery)
	}

	// 排序
	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
}

// GetOrderInquiryRecordPageSearch 获取问诊记录列表-分页
func (r *OrderInquiryDao) GetOrderInquiryRecordPageSearch(req requests.GetOrderInquiryRecordPage, page, pageSize int) (m []*model.OrderInquiry, total int64, err error) {
	var totalRecords int64

	// 构建查询条件
	query := global.Db.Model(&model.OrderInquiry{})

	// 医生
	query = query.Preload("UserDoctor", func(db *gorm.DB) *gorm.DB {
		return db.Omit("open_id", "union_id", "wx_session_key")
	})

	// 医生姓名
	if req.DoctorName != "" {
		subQuery := global.Db.Model(&model.UserDoctor{}).
			Select("doctor_id").
			Where("user_name LIKE ?", "%"+req.DoctorName+"%")

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

	// 用户
	query = query.Preload("User", func(db *gorm.DB) *gorm.DB {
		return db.Select("user_id", "user_name", "mobile")
	})

	// 订单类型
	if req.InquiryType != nil {
		query = query.Where("inquiry_type = ?", req.InquiryType)
	}

	// 订单问诊方式
	if req.InquiryMode != nil {
		query = query.Where("inquiry_mode = ?", req.InquiryMode)
	}

	// 问诊订单状态
	if req.InquiryStatus != nil {
		query = query.Where("inquiry_status = ?", req.InquiryStatus)
	}

	// 问诊订单退款状态
	if req.InquiryRefundStatus != nil {
		query = query.Where("inquiry_refund_status = ?", req.InquiryRefundStatus)
	}

	// 支付渠道
	if req.InquiryPayChannel != nil {
		query = query.Where("inquiry_pay_channel = ?", req.InquiryPayChannel)
	}

	// 支付状态
	if req.InquiryPayStatus != nil {
		query = query.Where("inquiry_pay_status = ?", req.InquiryPayStatus)
	}

	// 系统订单编号
	if req.InquiryNo != "" {
		query = query.Where("inquiry_no = ?", req.InquiryNo)
	}

	// 第三方支付流水号
	if req.EscrowTradeNo != "" {
		query = query.Where("escrow_trade_no = ?", req.EscrowTradeNo)
	}

	// 支付时间
	if req.PayTime != "" {
		payTime := strings.Split(req.PayTime, "&")
		if len(payTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", payTime[0])
			endTime, _ := time.Parse("2006-01-02", payTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("pay_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 接诊时间
	if req.ReceptionTime != "" {
		receptionTime := strings.Split(req.ReceptionTime, "&")
		if len(receptionTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", receptionTime[0])
			endTime, _ := time.Parse("2006-01-02", receptionTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("reception_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 订单完成时间
	if req.CompleteTime != "" {
		completeTime := strings.Split(req.CompleteTime, "&")
		if len(completeTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", completeTime[0])
			endTime, _ := time.Parse("2006-01-02", completeTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("complete_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 订单结束时间
	if req.FinishTime != "" {
		finishTime := strings.Split(req.FinishTime, "&")
		if len(finishTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", finishTime[0])
			endTime, _ := time.Parse("2006-01-02", finishTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("finish_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 是否提现
	if req.IsWithdrawal != nil {
		query = query.Where("is_withdrawal = ?", req.IsWithdrawal)
	}

	// 订单取消时间
	if req.CancelTime != "" {
		cancelTime := strings.Split(req.CancelTime, "&")
		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("cancel_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 创建时间
	if req.CreatedAt != "" {
		createdAt := strings.Split(req.CreatedAt, "&")
		if len(createdAt) == 2 {
			startTime, _ := time.Parse("2006-01-02", createdAt[0])
			endTime, _ := time.Parse("2006-01-02", createdAt[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.PatientName != "" {
		query = query.Where("patient_name LIKE ?", "%"+req.PatientName+"%")
	}

	// 手机号-医生/患者
	if req.Mobile != "" {
		// 患者
		patientSubQuery := global.Db.Model(&model.User{}).
			Select("user_id").
			Where("mobile = ?", req.Mobile)

		// 医生
		doctorUserSubQuery := global.Db.Model(&model.User{}).
			Select("user_id").
			Where("mobile = ?", req.Mobile)

		doctorSubQuery := global.Db.Model(&model.UserDoctor{}).
			Select("doctor_id").
			Where(gorm.Expr("user_id IN (?)", doctorUserSubQuery))

		query = query.Where("user_id IN (?)", patientSubQuery).Or("doctor_id IN (?)", doctorSubQuery)
	}

	// 排序
	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
}

// GetOrderInquiryTimeList 获取问诊订单列表-时间区间
func (r *OrderInquiryDao) GetOrderInquiryTimeList(maps interface{}, startTime, endTime string) (m []*model.OrderInquiry, err error) {
	err = global.Db.Where(maps).Where("reception_time BETWEEN ? AND ?", startTime, endTime).Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// GetOrderInquiryForAccountPageSearch 获取账户关联问诊订单列表-分页
func (r *OrderInquiryDao) GetOrderInquiryForAccountPageSearch(req requests.GetOrderInquiryForAccountPage, page, pageSize int) (m []*model.OrderInquiry, total int64, err error) {
	var totalRecords int64

	// 构建查询条件
	query := global.Db.Model(&model.OrderInquiry{})

	// 医生
	query = query.Preload("UserDoctor", func(db *gorm.DB) *gorm.DB {
		return db.Omit("open_id", "union_id", "wx_session_key")
	})

	// 医生id
	if req.DoctorId != "" {
		query = query.Where("doctor_id = ?", req.DoctorId)
	}

	// 问诊订单状态
	if req.InquiryStatus != nil {
		query = query.Where("inquiry_status = ?", req.InquiryStatus)
	}

	// 系统订单编号
	if req.InquiryNo != "" {
		query = query.Where("inquiry_no = ?", req.InquiryNo)
	}

	// 第三方支付流水号
	if req.EscrowTradeNo != "" {
		query = query.Where("escrow_trade_no = ?", req.EscrowTradeNo)
	}

	// 支付时间
	if req.PayTime != "" {
		payTime := strings.Split(req.PayTime, "&")
		if len(payTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", payTime[0])
			endTime, _ := time.Parse("2006-01-02", payTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("pay_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 接诊时间
	if req.ReceptionTime != "" {
		receptionTime := strings.Split(req.ReceptionTime, "&")
		if len(receptionTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", receptionTime[0])
			endTime, _ := time.Parse("2006-01-02", receptionTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("reception_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 订单完成时间
	if req.CompleteTime != "" {
		completeTime := strings.Split(req.CompleteTime, "&")
		if len(completeTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", completeTime[0])
			endTime, _ := time.Parse("2006-01-02", completeTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("complete_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 订单结束时间
	if req.FinishTime != "" {
		finishTime := strings.Split(req.FinishTime, "&")
		if len(finishTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", finishTime[0])
			endTime, _ := time.Parse("2006-01-02", finishTime[1])

			if startTime == endTime {
				endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
			}

			query = query.Where("finish_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 是否提现
	if req.IsWithdrawal != nil {
		query = query.Where("is_withdrawal = ?", req.IsWithdrawal)
	}

	// 订单取消时间
	if req.CancelTime != "" {
		cancelTime := strings.Split(req.CancelTime, "&")
		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("cancel_time BETWEEN ? AND ?", startTime, endTime)
		}
	}

	// 创建时间
	if req.CreatedAt != "" {
		createdAt := strings.Split(req.CreatedAt, "&")
		if len(createdAt) == 2 {
			startTime, _ := time.Parse("2006-01-02", createdAt[0])
			endTime, _ := time.Parse("2006-01-02", createdAt[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.PatientName != "" {
		query = query.Where("patient_name LIKE ?", "%"+req.PatientName+"%")
	}

	// 手机号-医生/患者
	if req.Mobile != "" {
		// 患者
		subQuery := global.Db.Model(&model.User{}).
			Select("user_id").
			Where("mobile = ?", req.Mobile)

		query = query.Where("user_id IN (?)", subQuery)
	}

	// 排序
	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
}

// GetOrderInquiryForAccountExportListSearch 医生账户-关联订单
func (r *OrderInquiryDao) GetOrderInquiryForAccountExportListSearch(req requests.OrderInquiryForAccountExportList) (m []*model.OrderInquiry, err error) {
	// 构建查询条件
	query := global.Db.Model(&model.OrderInquiry{})

	// 医生
	query = query.Preload("UserDoctor", func(db *gorm.DB) *gorm.DB {
		return db.Omit("open_id", "union_id", "wx_session_key")
	})

	// 用户
	query = query.Preload("User")

	// 医生id
	if req.DoctorId != "" {
		query = query.Where("doctor_id = ?", req.DoctorId)
	}

	// 当前搜索数据
	if req.Type == 1 {
		// 问诊订单状态
		if req.InquiryStatus != nil {
			query = query.Where("inquiry_status = ?", req.InquiryStatus)
		}

		// 系统订单编号
		if req.InquiryNo != "" {
			query = query.Where("inquiry_no = ?", req.InquiryNo)
		}

		// 第三方支付流水号
		if req.EscrowTradeNo != "" {
			query = query.Where("escrow_trade_no = ?", req.EscrowTradeNo)
		}

		// 支付时间
		if req.PayTime != "" {
			payTime := strings.Split(req.PayTime, "&")
			if len(payTime) == 2 {
				startTime, _ := time.Parse("2006-01-02", payTime[0])
				endTime, _ := time.Parse("2006-01-02", payTime[1])

				if startTime == endTime {
					endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
				}

				query = query.Where("pay_time BETWEEN ? AND ?", startTime, endTime)
			}
		}

		// 接诊时间
		if req.ReceptionTime != "" {
			receptionTime := strings.Split(req.ReceptionTime, "&")
			if len(receptionTime) == 2 {
				startTime, _ := time.Parse("2006-01-02", receptionTime[0])
				endTime, _ := time.Parse("2006-01-02", receptionTime[1])

				if startTime == endTime {
					endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
				}

				query = query.Where("reception_time BETWEEN ? AND ?", startTime, endTime)
			}
		}

		// 订单完成时间
		if req.CompleteTime != "" {
			completeTime := strings.Split(req.CompleteTime, "&")
			if len(completeTime) == 2 {
				startTime, _ := time.Parse("2006-01-02", completeTime[0])
				endTime, _ := time.Parse("2006-01-02", completeTime[1])

				if startTime == endTime {
					endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
				}

				query = query.Where("complete_time BETWEEN ? AND ?", startTime, endTime)
			}
		}

		// 订单结束时间
		if req.FinishTime != "" {
			finishTime := strings.Split(req.FinishTime, "&")
			if len(finishTime) == 2 {
				startTime, _ := time.Parse("2006-01-02", finishTime[0])
				endTime, _ := time.Parse("2006-01-02", finishTime[1])

				if startTime == endTime {
					endTime = endTime.Add(23*time.Hour + 59*time.Minute + 59*time.Second)
				}

				query = query.Where("finish_time BETWEEN ? AND ?", startTime, endTime)
			}
		}

		// 是否提现
		if req.IsWithdrawal != nil {
			query = query.Where("is_withdrawal = ?", req.IsWithdrawal)
		}

		// 订单取消时间
		if req.CancelTime != "" {
			cancelTime := strings.Split(req.CancelTime, "&")
			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("cancel_time BETWEEN ? AND ?", startTime, endTime)
			}
		}

		// 创建时间
		if req.CreatedAt != "" {
			createdAt := strings.Split(req.CreatedAt, "&")
			if len(createdAt) == 2 {
				startTime, _ := time.Parse("2006-01-02", createdAt[0])
				endTime, _ := time.Parse("2006-01-02", createdAt[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.PatientName != "" {
			query = query.Where("patient_name LIKE ?", "%"+req.PatientName+"%")
		}

		// 手机号-医生/患者
		if req.Mobile != "" {
			// 患者
			subQuery := global.Db.Model(&model.User{}).
				Select("user_id").
				Where("mobile = ?", req.Mobile)

			query = query.Where("user_id IN (?)", subQuery)
		}
	}

	// 当前选中数据
	if req.Type == 2 {
		if req.Id == "" {
			return nil, errors.New("未提供需导出数据编号")
		}

		id := strings.Split(req.Id, ",")
		query = query.Where("order_inquiry_id IN (?)", id)
	}

	// 排序
	query = query.Order("created_at desc")

	err = query.Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}