230 lines
8.5 KiB
Go
230 lines
8.5 KiB
Go
package services
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"time"
|
||
|
||
"quyun/v2/app/errorx"
|
||
"quyun/v2/database/models"
|
||
"quyun/v2/pkg/consts"
|
||
|
||
"github.com/sirupsen/logrus"
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/clause"
|
||
)
|
||
|
||
// LedgerApplyResult 表示一次账本写入（含幂等命中）的结果，包含账本记录与租户用户余额快照。
|
||
type LedgerApplyResult struct {
|
||
// Ledger 为本次创建的账本记录（若幂等命中则返回已有记录）。
|
||
Ledger *models.TenantLedger
|
||
// TenantUser 为写入后余额状态（若幂等命中则返回当前快照）。
|
||
TenantUser *models.TenantUser
|
||
}
|
||
|
||
// ledger 提供租户余额账本能力（冻结/解冻/扣减/退款/充值），支持幂等与行锁保证一致性。
|
||
//
|
||
// @provider
|
||
type ledger struct {
|
||
db *gorm.DB
|
||
}
|
||
|
||
// Freeze 将可用余额转入冻结余额，并写入账本记录。
|
||
func (s *ledger) Freeze(ctx context.Context, tenantID, userID, orderID, amount int64, idempotencyKey, remark string, now time.Time) (*LedgerApplyResult, error) {
|
||
return s.apply(ctx, s.db, tenantID, userID, orderID, consts.TenantLedgerTypeFreeze, amount, -amount, amount, idempotencyKey, remark, now)
|
||
}
|
||
|
||
// Unfreeze 将冻结余额转回可用余额，并写入账本记录。
|
||
func (s *ledger) Unfreeze(ctx context.Context, tenantID, userID, orderID, amount int64, idempotencyKey, remark string, now time.Time) (*LedgerApplyResult, error) {
|
||
return s.apply(ctx, s.db, tenantID, userID, orderID, consts.TenantLedgerTypeUnfreeze, amount, amount, -amount, idempotencyKey, remark, now)
|
||
}
|
||
|
||
// FreezeTx 为 Freeze 的事务版本（由外层事务控制提交/回滚）。
|
||
func (s *ledger) FreezeTx(ctx context.Context, tx *gorm.DB, tenantID, userID, orderID, amount int64, idempotencyKey, remark string, now time.Time) (*LedgerApplyResult, error) {
|
||
return s.apply(ctx, tx, tenantID, userID, orderID, consts.TenantLedgerTypeFreeze, amount, -amount, amount, idempotencyKey, remark, now)
|
||
}
|
||
|
||
// UnfreezeTx 为 Unfreeze 的事务版本（由外层事务控制提交/回滚）。
|
||
func (s *ledger) UnfreezeTx(ctx context.Context, tx *gorm.DB, tenantID, userID, orderID, amount int64, idempotencyKey, remark string, now time.Time) (*LedgerApplyResult, error) {
|
||
return s.apply(ctx, tx, tenantID, userID, orderID, consts.TenantLedgerTypeUnfreeze, amount, amount, -amount, idempotencyKey, remark, now)
|
||
}
|
||
|
||
// DebitPurchaseTx 将冻结资金转为实际扣款（减少冻结余额），并写入账本记录。
|
||
func (s *ledger) DebitPurchaseTx(ctx context.Context, tx *gorm.DB, tenantID, userID, orderID, amount int64, idempotencyKey, remark string, now time.Time) (*LedgerApplyResult, error) {
|
||
return s.apply(ctx, tx, tenantID, userID, orderID, consts.TenantLedgerTypeDebitPurchase, amount, 0, -amount, idempotencyKey, remark, now)
|
||
}
|
||
|
||
// CreditRefundTx 将退款金额退回到可用余额，并写入账本记录。
|
||
func (s *ledger) CreditRefundTx(ctx context.Context, tx *gorm.DB, tenantID, userID, orderID, amount int64, idempotencyKey, remark string, now time.Time) (*LedgerApplyResult, error) {
|
||
return s.apply(ctx, tx, tenantID, userID, orderID, consts.TenantLedgerTypeCreditRefund, amount, amount, 0, idempotencyKey, remark, now)
|
||
}
|
||
|
||
// CreditTopupTx 将充值金额记入可用余额，并写入账本记录。
|
||
func (s *ledger) CreditTopupTx(ctx context.Context, tx *gorm.DB, tenantID, userID, orderID, amount int64, idempotencyKey, remark string, now time.Time) (*LedgerApplyResult, error) {
|
||
return s.apply(ctx, tx, tenantID, userID, orderID, consts.TenantLedgerTypeCreditTopup, amount, amount, 0, idempotencyKey, remark, now)
|
||
}
|
||
|
||
func (s *ledger) apply(
|
||
ctx context.Context,
|
||
tx *gorm.DB,
|
||
tenantID, userID, orderID int64,
|
||
ledgerType consts.TenantLedgerType,
|
||
amount, deltaBalance, deltaFrozen int64,
|
||
idempotencyKey, remark string,
|
||
now time.Time,
|
||
) (*LedgerApplyResult, error) {
|
||
// 关键前置校验：金额必须为正；时间允许由调用方注入，便于测试与一致性落库。
|
||
if amount <= 0 {
|
||
return nil, errorx.ErrInvalidParameter.WithMsg("amount must be > 0")
|
||
}
|
||
if now.IsZero() {
|
||
now = time.Now()
|
||
}
|
||
|
||
logrus.WithFields(logrus.Fields{
|
||
"tenant_id": tenantID,
|
||
"user_id": userID,
|
||
"order_id": orderID,
|
||
"type": ledgerType,
|
||
"amount": amount,
|
||
"idempotency_key": idempotencyKey,
|
||
"delta_balance": deltaBalance,
|
||
"delta_frozen": deltaFrozen,
|
||
"remark_non_empty": remark != "",
|
||
}).Info("services.ledger.apply")
|
||
|
||
var out LedgerApplyResult
|
||
|
||
err := tx.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
||
// 幂等快速路径：在进入行锁之前先查一次，减少锁竞争（命中则直接返回）。
|
||
if idempotencyKey != "" {
|
||
var existing models.TenantLedger
|
||
if err := tx.
|
||
Where("tenant_id = ? AND user_id = ? AND idempotency_key = ?", tenantID, userID, idempotencyKey).
|
||
First(&existing).Error; err == nil {
|
||
var current models.TenantUser
|
||
if err := tx.Where("tenant_id = ? AND user_id = ?", tenantID, userID).First(¤t).Error; err != nil {
|
||
return err
|
||
}
|
||
out.Ledger = &existing
|
||
out.TenantUser = ¤t
|
||
return nil
|
||
} else if !errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return err
|
||
}
|
||
}
|
||
|
||
// 使用行锁锁住 tenant_users，确保同一租户下同一用户余额更新的串行一致性。
|
||
var tu models.TenantUser
|
||
if err := tx.
|
||
Clauses(clause.Locking{Strength: "UPDATE"}).
|
||
Where("tenant_id = ? AND user_id = ?", tenantID, userID).
|
||
First(&tu).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return errorx.ErrRecordNotFound.WithMsg("tenant user not found")
|
||
}
|
||
return err
|
||
}
|
||
|
||
// 二次幂等校验：防止并发下在获取锁前后插入账本导致的重复写入。
|
||
if idempotencyKey != "" {
|
||
var existing models.TenantLedger
|
||
if err := tx.
|
||
Where("tenant_id = ? AND user_id = ? AND idempotency_key = ?", tenantID, userID, idempotencyKey).
|
||
First(&existing).Error; err == nil {
|
||
out.Ledger = &existing
|
||
out.TenantUser = &tu
|
||
return nil
|
||
} else if !errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return err
|
||
}
|
||
}
|
||
|
||
balanceBefore := tu.Balance
|
||
frozenBefore := tu.BalanceFrozen
|
||
balanceAfter := balanceBefore + deltaBalance
|
||
frozenAfter := frozenBefore + deltaFrozen
|
||
|
||
// 关键不变量：余额/冻结余额不能为负，避免透支或超额解冻。
|
||
if balanceAfter < 0 {
|
||
return errorx.ErrPreconditionFailed.WithMsg("余额不足")
|
||
}
|
||
if frozenAfter < 0 {
|
||
return errorx.ErrPreconditionFailed.WithMsg("冻结余额不足")
|
||
}
|
||
|
||
// 先更新余额，再写账本：任何一步失败都回滚，保证“余额变更”和“账本记录”一致。
|
||
if err := tx.Model(&models.TenantUser{}).
|
||
Where("id = ?", tu.ID).
|
||
Updates(map[string]any{
|
||
"balance": balanceAfter,
|
||
"balance_frozen": frozenAfter,
|
||
"updated_at": now,
|
||
}).Error; err != nil {
|
||
return err
|
||
}
|
||
|
||
// 写入账本：记录变更前后快照，便于对账与审计；幂等键用于去重。
|
||
ledger := &models.TenantLedger{
|
||
TenantID: tenantID,
|
||
UserID: userID,
|
||
OrderID: orderID,
|
||
Type: ledgerType,
|
||
Amount: amount,
|
||
BalanceBefore: balanceBefore,
|
||
BalanceAfter: balanceAfter,
|
||
FrozenBefore: frozenBefore,
|
||
FrozenAfter: frozenAfter,
|
||
IdempotencyKey: idempotencyKey,
|
||
Remark: remark,
|
||
CreatedAt: now,
|
||
UpdatedAt: now,
|
||
}
|
||
if err := tx.Create(ledger).Error; err != nil {
|
||
// 并发下可能出现“先写成功后再重试”的情况：尝试按幂等键回读，保持接口幂等。
|
||
if idempotencyKey != "" {
|
||
var existing models.TenantLedger
|
||
if e2 := tx.
|
||
Where("tenant_id = ? AND user_id = ? AND idempotency_key = ?", tenantID, userID, idempotencyKey).
|
||
First(&existing).Error; e2 == nil {
|
||
out.Ledger = &existing
|
||
out.TenantUser = &tu
|
||
return nil
|
||
}
|
||
}
|
||
return err
|
||
}
|
||
|
||
tu.Balance = balanceAfter
|
||
tu.BalanceFrozen = frozenAfter
|
||
tu.UpdatedAt = now
|
||
|
||
out.Ledger = ledger
|
||
out.TenantUser = &tu
|
||
return nil
|
||
})
|
||
if err != nil {
|
||
logrus.WithFields(logrus.Fields{
|
||
"tenant_id": tenantID,
|
||
"user_id": userID,
|
||
"order_id": orderID,
|
||
"type": ledgerType,
|
||
"idempotency_key": idempotencyKey,
|
||
}).WithError(err).Warn("services.ledger.apply.failed")
|
||
return nil, err
|
||
}
|
||
|
||
logrus.WithFields(logrus.Fields{
|
||
"tenant_id": tenantID,
|
||
"user_id": userID,
|
||
"order_id": orderID,
|
||
"type": ledgerType,
|
||
"ledger_id": out.Ledger.ID,
|
||
"idempotency_key": idempotencyKey,
|
||
"balance_after": out.TenantUser.Balance,
|
||
"frozen_after": out.TenantUser.BalanceFrozen,
|
||
}).Info("services.ledger.apply.ok")
|
||
|
||
return &out, nil
|
||
}
|