draft

backend/internal/repository/trade_order.go

@@ -0,0 +1,80 @@
+package repository
+
+import (
+	"context"
+	"errors"
+	"time"
+
+	"gorm.io/gorm"
+	"pay-bridge/internal/model"
+)
+
+// TradeOrderRepository 交易订单数据访问
+type TradeOrderRepository struct {
+	db *gorm.DB
+}
+
+func NewTradeOrderRepository(db *gorm.DB) *TradeOrderRepository {
+	return &TradeOrderRepository{db: db}
+}
+
+// Create 创建订单
+func (r *TradeOrderRepository) Create(ctx context.Context, order *model.TradeOrder) error {
+	return r.db.WithContext(ctx).Create(order).Error
+}
+
+// GetByTradeNo 按 trade_no 查询
+func (r *TradeOrderRepository) GetByTradeNo(ctx context.Context, tradeNo string) (*model.TradeOrder, error) {
+	var order model.TradeOrder
+	err := r.db.WithContext(ctx).Where("trade_no = ?", tradeNo).First(&order).Error
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, nil
+	}
+	return &order, err
+}
+
+// GetByMerchantOrderNo 按 app_id + merchant_order_no 查询
+func (r *TradeOrderRepository) GetByMerchantOrderNo(ctx context.Context, appID, merchantOrderNo string) (*model.TradeOrder, error) {
+	var order model.TradeOrder
+	err := r.db.WithContext(ctx).Where("app_id = ? AND merchant_order_no = ?", appID, merchantOrderNo).First(&order).Error
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, nil
+	}
+	return &order, err
+}
+
+// GetByChannelTradeNo 按渠道交易号查询
+func (r *TradeOrderRepository) GetByChannelTradeNo(ctx context.Context, channelTradeNo string) (*model.TradeOrder, error) {
+	var order model.TradeOrder
+	err := r.db.WithContext(ctx).Where("channel_trade_no = ?", channelTradeNo).First(&order).Error
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, nil
+	}
+	return &order, err
+}
+
+// UpdateStatus 乐观锁更新状态（只允许从 fromStatus 流转到 toStatus）
+// 返回 bool 表示是否更新成功（false = 已被其他 goroutine 更新）
+func (r *TradeOrderRepository) UpdateStatus(ctx context.Context, tradeNo string, fromStatus, toStatus model.TradeStatus, updates map[string]any) (bool, error) {
+	if updates == nil {
+		updates = make(map[string]any)
+	}
+	updates["status"] = toStatus
+
+	result := r.db.WithContext(ctx).Model(&model.TradeOrder{}).
+		Where("trade_no = ? AND status = ?", tradeNo, fromStatus).
+		Updates(updates)
+	if result.Error != nil {
+		return false, result.Error
+	}
+	return result.RowsAffected > 0, nil
+}
+
+// ListPayingExpired 查询已过期的 PAYING 订单（用于定时关单补偿）
+func (r *TradeOrderRepository) ListPayingExpired(ctx context.Context, before time.Time, limit int) ([]*model.TradeOrder, error) {
+	var orders []*model.TradeOrder
+	err := r.db.WithContext(ctx).
+		Where("status = ? AND expire_time < ?", model.TradeStatusPaying, before).
+		Limit(limit).Find(&orders).Error
+	return orders, err
+}