draft

backend/internal/repository/reconciliation.go

@@ -0,0 +1,62 @@
+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
+}