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

package dao

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

type DoctorWithdrawalDao struct {
}

// GetDoctorWithdrawalById 获取医生提现数据-医生提现id
func (r *DoctorWithdrawalDao) GetDoctorWithdrawalById(withdrawalId int64) (m *model.DoctorWithdrawal, err error) {
	err = global.Db.First(&m, withdrawalId).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// DeleteDoctorWithdrawal 删除医生提现
func (r *DoctorWithdrawalDao) DeleteDoctorWithdrawal(tx *gorm.DB, maps interface{}) error {
	err := tx.Where(maps).Delete(&model.DoctorWithdrawal{}).Error
	if err != nil {
		return err
	}
	return nil
}

// EditDoctorWithdrawal 修改医生提现
func (r *DoctorWithdrawalDao) EditDoctorWithdrawal(tx *gorm.DB, maps interface{}, data interface{}) error {
	err := tx.Model(&model.DoctorWithdrawal{}).Where(maps).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// EditDoctorWithdrawalById 修改医生提现-医生提现id
func (r *DoctorWithdrawalDao) EditDoctorWithdrawalById(tx *gorm.DB, withdrawalId int64, data interface{}) error {
	err := tx.Model(&model.DoctorWithdrawal{}).Where("withdrawal_id = ?", withdrawalId).Updates(data).Error
	if err != nil {
		return err
	}
	return nil
}

// GetDoctorWithdrawalList 获取医生提现列表
func (r *DoctorWithdrawalDao) GetDoctorWithdrawalList(maps interface{}) (m []*model.DoctorWithdrawal, err error) {
	err = global.Db.Where(maps).Find(&m).Error
	if err != nil {
		return nil, err
	}
	return m, nil
}

// AddDoctorWithdrawal 新增医生提现
func (r *DoctorWithdrawalDao) AddDoctorWithdrawal(tx *gorm.DB, model *model.DoctorWithdrawal) (*model.DoctorWithdrawal, error) {
	if err := tx.Create(model).Error; err != nil {
		return nil, err
	}
	return model, nil
}

// GetDoctorWithdrawalPageSearch 获取医生提现列表-分页
func (r *DoctorWithdrawalDao) GetDoctorWithdrawalPageSearch(req requests.GetDoctorWithdrawalPage, page, pageSize int) (m []*model.DoctorWithdrawal, total int64, err error) {
	var totalRecords int64

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

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

	// 提现关联银行卡
	query = query.Preload("DoctorWithdrawalBank")

	// 基础银行卡
	query = query.Preload("DoctorWithdrawalBank.BasicBank")

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

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

	// 手机号-医生
	if 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("doctor_id IN (?)", doctorSubQuery)
	}

	// 审核状态
	if req.ExamineStatus != nil {
		query = query.Where("examine_status = ?", req.ExamineStatus)
	}

	// 打款状态
	if req.PaymentStatus != nil {
		query = query.Where("payment_status = ?", req.PaymentStatus)
	}

	// 审核日期
	if req.ExamineTime != "" {
		examineTime := strings.Split(req.ExamineTime, "&")
		if len(examineTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", examineTime[0])
			endTime, _ := time.Parse("2006-01-02", examineTime[1])

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

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

	// 财务打款时间
	if req.PaymentTime != "" {
		paymentTime := strings.Split(req.PaymentTime, "&")
		if len(paymentTime) == 2 {
			startTime, _ := time.Parse("2006-01-02", paymentTime[0])
			endTime, _ := time.Parse("2006-01-02", paymentTime[1])

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

			query = query.Where("payment_time 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
}

// GetDoctorWithdrawalExportListSearch 获取医生提现列表-导出
func (r *DoctorWithdrawalDao) GetDoctorWithdrawalExportListSearch(req requests.DoctorWithdrawalExportList) (m []*model.DoctorWithdrawal, err error) {
	// 构建查询条件
	query := global.Db.Model(&model.DoctorWithdrawal{})

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

	// 用户-医生
	query = query.Preload("UserDoctor.User", func(db *gorm.DB) *gorm.DB {
		return db.Omit("user_password", "salt")
	})

	// 医生详情
	query = query.Preload("UserDoctor.UserDoctorInfo")

	// 提现关联银行卡
	query = query.Preload("DoctorWithdrawalBank")

	// 基础银行卡
	query = query.Preload("DoctorWithdrawalBank.BasicBank")

	// 当前搜索数据
	if req.Type == 1 {
		// 医生姓名
		if req.UserName != "" {
			subQuery := global.Db.Model(&model.UserDoctor{}).
				Select("doctor_id").
				Where("user_name LIKE ?", "%"+req.UserName+"%")

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

		// 手机号-医生
		if 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("doctor_id IN (?)", doctorSubQuery)
		}

		// 审核状态
		if req.ExamineStatus != nil {
			query = query.Where("examine_status = ?", req.ExamineStatus)
		}

		// 打款状态
		if req.PaymentStatus != nil {
			query = query.Where("payment_status = ?", req.PaymentStatus)
		}

		// 审核日期
		if req.ExamineTime != "" {
			examineTime := strings.Split(req.ExamineTime, "&")
			if len(examineTime) == 2 {
				startTime, _ := time.Parse("2006-01-02", examineTime[0])
				endTime, _ := time.Parse("2006-01-02", examineTime[1])

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

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

		// 财务打款时间
		if req.PaymentTime != "" {
			paymentTime := strings.Split(req.PaymentTime, "&")
			if len(paymentTime) == 2 {
				startTime, _ := time.Parse("2006-01-02", paymentTime[0])
				endTime, _ := time.Parse("2006-01-02", paymentTime[1])

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

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

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

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

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

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