63 lines
2.1 KiB
Go
63 lines
2.1 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"gorm.io/gorm"
|
|
"pay-bridge/internal/model"
|
|
)
|
|
|
|
// ReconciliationRepository 对账数据访问
|
|
type ReconciliationRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewReconciliationRepository(db *gorm.DB) *ReconciliationRepository {
|
|
return &ReconciliationRepository{db: db}
|
|
}
|
|
|
|
// CreateReport 创建对账报告
|
|
func (r *ReconciliationRepository) CreateReport(ctx context.Context, report *model.ReconciliationReport) error {
|
|
return r.db.WithContext(ctx).Create(report).Error
|
|
}
|
|
|
|
// GetReport 查询对账报告
|
|
func (r *ReconciliationRepository) GetReport(ctx context.Context, appID, billDate, channelCode string) (*model.ReconciliationReport, error) {
|
|
var report model.ReconciliationReport
|
|
err := r.db.WithContext(ctx).
|
|
Where("app_id = ? AND bill_date = ? AND channel_code = ?", appID, billDate, channelCode).
|
|
First(&report).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, nil
|
|
}
|
|
return &report, err
|
|
}
|
|
|
|
// UpdateReport 更新对账报告
|
|
func (r *ReconciliationRepository) UpdateReport(ctx context.Context, id uint64, updates map[string]any) error {
|
|
return r.db.WithContext(ctx).Model(&model.ReconciliationReport{}).Where("id = ?", id).Updates(updates).Error
|
|
}
|
|
|
|
// CreateException 创建对账异常记录
|
|
func (r *ReconciliationRepository) CreateException(ctx context.Context, ex *model.ReconciliationException) error {
|
|
return r.db.WithContext(ctx).Create(ex).Error
|
|
}
|
|
|
|
// ListExceptions 查询报告下的异常明细
|
|
func (r *ReconciliationRepository) ListExceptions(ctx context.Context, reportID uint64) ([]*model.ReconciliationException, error) {
|
|
var exs []*model.ReconciliationException
|
|
err := r.db.WithContext(ctx).Where("report_id = ?", reportID).Find(&exs).Error
|
|
return exs, err
|
|
}
|
|
|
|
// ListPaidOrdersByDate 查询指定日期的已支付订单（用于对账）
|
|
func (r *ReconciliationRepository) ListPaidOrdersByDate(ctx context.Context, appID, date string) ([]*model.TradeOrder, error) {
|
|
var orders []*model.TradeOrder
|
|
err := r.db.WithContext(ctx).
|
|
Where("app_id = ? AND status = ? AND DATE(pay_time) = ?",
|
|
appID, model.TradeStatusPaid, date).
|
|
Find(&orders).Error
|
|
return orders, err
|
|
}
|