youngee_b_api/app/dao/recharge_record_dao.go

package dao

import (
	"youngee_b_api/app/entity"
	"youngee_b_api/app/vo"
)

type RechargeRecordDao struct{}

func (d RechargeRecordDao) Insert(rechargeRecord *entity.RechargeRecord) error {
	err := Db.Debug().Model(entity.RechargeRecord{}).Omit("confirm_at").Create(rechargeRecord).Error
	if err != nil {
		return err
	}
	return nil
}

// 获取指定企业id的充值金额、确认中金额
func (d RechargeRecordDao) GetRechargeAmount(enterpriseId string, status int64) (float64, error) {
	var totalAmount float64
	query := Db.Debug().Model(&entity.RechargeRecord{})
	err := query.Where("enterprise_id = ? AND status = ?", enterpriseId, status).Select("SUM(recharge_amount)").Scan(&totalAmount).Error
	if err != nil {
		return 0, err
	}
	return totalAmount, nil
}

// 获取指定企业id的充值记录
func (d RechargeRecordDao) RechargeInfoList(param *vo.RechargeParam) ([]entity.RechargeRecord, int64, error) {
	rechargeRecords := []entity.RechargeRecord{}
	var total int64
	query := Db.Debug().Model(&entity.RechargeRecord{}).Where("enterprise_id = ? AND status = ?", param.EnterpriseId, param.RechargeState)
	query.Count(&total)
	query = query.Select("recharge_id, recharge_amount, transfer_voucher_url, recharge_method, commit_at, confirm_at, refuse_at, fail_reason")
	offset := (param.Page - 1) * param.PageSize
	var err error
	if param.RechargeState == 1 {
		err = query.Order("commit_at desc").Offset(offset).Limit(param.PageSize).Find(&rechargeRecords).Error
	} else if param.RechargeState == 2 {
		err = query.Order("confirm_at desc").Offset(offset).Limit(param.PageSize).Find(&rechargeRecords).Error
	} else if param.RechargeState == 3 {
		err = query.Order("refuse_at desc").Offset(offset).Limit(param.PageSize).Find(&rechargeRecords).Error
	}
	if err != nil {
		return nil, 0, err
	}

	return rechargeRecords, total, nil
}