Files
junhong_cmp_fiber/internal/infrastructure/wallet/credit_consumer.go
2026-07-24 16:07:18 +08:00

60 lines
2.8 KiB
Go

package wallet
import (
"context"
"github.com/bytedance/sonic"
"gorm.io/gorm"
walletapp "github.com/break/junhong_cmp_fiber/internal/application/wallet"
"github.com/break/junhong_cmp_fiber/internal/infrastructure/messaging/outbox"
"github.com/break/junhong_cmp_fiber/internal/model"
"github.com/break/junhong_cmp_fiber/pkg/constants"
"github.com/break/junhong_cmp_fiber/pkg/errors"
)
// CreditEventConsumer 校验已投递的代理主钱包入账事件具有对应权威资金流水。
type CreditEventConsumer struct {
db *gorm.DB
}
// NewCreditEventConsumer 创建代理主钱包入账事件消费者。
func NewCreditEventConsumer(db *gorm.DB) *CreditEventConsumer {
return &CreditEventConsumer{db: db}
}
// Consume 校验事件载荷及不可变流水,拒绝确认未知或损坏事件。
func (c *CreditEventConsumer) Consume(ctx context.Context, envelope outbox.DeliveryEnvelope) error {
if c == nil || c.db == nil {
return errors.New(errors.CodeInternalError, "代理主钱包入账事件消费者未配置")
}
if envelope.EventType != constants.OutboxEventTypeAgentMainWalletCredited || envelope.PayloadVersion != constants.AgentMainWalletCreditedPayloadVersionV1 {
return errors.New(errors.CodeInvalidParam, "代理主钱包入账事件类型或版本不受支持")
}
var event walletapp.CreditedEvent
if err := sonic.Unmarshal(envelope.Payload, &event); err != nil {
return errors.Wrap(errors.CodeInvalidParam, err, "代理主钱包入账事件载荷无法解析")
}
validRecharge := event.ReferenceType == constants.ReferenceTypeTopup && event.TransactionType == constants.AgentTransactionTypeRecharge
validAdjustment := event.ReferenceType == constants.ReferenceTypeManualAdjustment && event.TransactionType == constants.AgentTransactionTypeAdjustment
if event.EventID != envelope.EventID || event.WalletID == 0 || event.ReferenceID == 0 || event.Amount <= 0 || (!validRecharge && !validAdjustment) {
return errors.New(errors.CodeInvalidParam, "代理主钱包入账事件载荷不完整")
}
var transaction model.AgentWalletTransaction
err := c.db.WithContext(ctx).Unscoped().
Where("agent_wallet_id = ? AND reference_type = ? AND reference_id = ? AND transaction_type = ? AND status = ?",
event.WalletID, event.ReferenceType, event.ReferenceID, event.TransactionType, constants.TransactionStatusSuccess).
First(&transaction).Error
if err != nil {
if err == gorm.ErrRecordNotFound {
return errors.New(errors.CodeInternalError, "代理主钱包入账事件缺少权威资金流水")
}
return errors.Wrap(errors.CodeDatabaseError, err, "查询代理主钱包入账流水失败")
}
if transaction.Amount != event.Amount || transaction.BalanceBefore != event.BalanceBefore || transaction.BalanceAfter != event.BalanceAfter {
return errors.New(errors.CodeInternalError, "代理主钱包入账事件与权威资金流水不一致")
}
return nil
}