wangjia
69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
package store
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
|
|
"github.com/wangjia/pay/internal/model"
|
|
)
|
|
|
|
// WebhookDeliveryRow is the row type returned by WebhookStore (alias to the model).
|
|
type WebhookDeliveryRow = model.WebhookDelivery
|
|
|
|
type WebhookStore struct{ db *gorm.DB }
|
|
|
|
func NewWebhookStore(db *gorm.DB) *WebhookStore { return &WebhookStore{db: db} }
|
|
|
|
// EnqueueDelivery inserts an outbox row; a duplicate (out_trade_no,event_type)
|
|
// is a no-op (idempotent enqueue) via ON CONFLICT DO NOTHING.
|
|
func (s *WebhookStore) EnqueueDelivery(outTradeNo, bizSystem, eventType, payload string) error {
|
|
row := model.WebhookDelivery{
|
|
OutTradeNo: outTradeNo, BizSystem: bizSystem, EventType: eventType, Payload: payload,
|
|
}
|
|
err := s.db.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "out_trade_no"}, {Name: "event_type"}},
|
|
DoNothing: true,
|
|
}).Create(&row).Error
|
|
if err != nil {
|
|
return fmt.Errorf("store.EnqueueDelivery: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *WebhookStore) ListUndelivered(limit int) ([]WebhookDeliveryRow, error) {
|
|
if limit <= 0 || limit > 200 {
|
|
limit = 50
|
|
}
|
|
var out []WebhookDeliveryRow
|
|
if err := s.db.Where("delivered = ?", false).Order("id ASC").Limit(limit).Find(&out).Error; err != nil {
|
|
return nil, fmt.Errorf("store.ListUndelivered: %w", err)
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func (s *WebhookStore) MarkDelivered(id uint64) error {
|
|
if err := s.db.Model(&model.WebhookDelivery{}).Where("id = ?", id).
|
|
Update("delivered", true).Error; err != nil {
|
|
return fmt.Errorf("store.MarkDelivered: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// MarkFailed increments attempts and records the last error, leaving the row
|
|
// undelivered for the next retry sweep.
|
|
func (s *WebhookStore) MarkFailed(id uint64, errMsg string) error {
|
|
if len(errMsg) > 255 {
|
|
errMsg = errMsg[:255]
|
|
}
|
|
if err := s.db.Model(&model.WebhookDelivery{}).Where("id = ?", id).
|
|
Updates(map[string]any{
|
|
"attempts": gorm.Expr("attempts + 1"),
|
|
"last_error": errMsg,
|
|
}).Error; err != nil {
|
|
return fmt.Errorf("store.MarkFailed: %w", err)
|
|
}
|
|
return nil
|
|
}
|