98 lines
3.6 KiB
Python
98 lines
3.6 KiB
Python
"""fin_account 仓储：按客户 ID 取资金账户 + 余额加减（原子 UPDATE）。
|
||
|
||
注：本表主键为 customer_id（非 id），故不复用 BaseRepository.delete/count 中的 id 约定。
|
||
余额增减用原子 UPDATE（balance = balance ± delta）防并发丢更新，不依赖乐观锁重试。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
from decimal import Decimal
|
||
|
||
from sqlalchemy import select, update
|
||
|
||
from model.fin_account import FinAccount
|
||
from repositories.base import BaseRepository
|
||
|
||
|
||
class FinAccountRepo(BaseRepository):
|
||
model = FinAccount
|
||
|
||
async def get_by_customer_id(self, customer_id: int) -> FinAccount | None:
|
||
return await self.db.scalar(
|
||
select(FinAccount).where(FinAccount.customer_id == customer_id)
|
||
)
|
||
|
||
async def create(self, customer_id: int, balance: Decimal) -> FinAccount:
|
||
"""开资金户（充值时账户不存在则自动开户入账）。"""
|
||
account = FinAccount(customer_id=customer_id, balance=balance)
|
||
self.db.add(account)
|
||
await self.db.commit()
|
||
await self.db.refresh(account)
|
||
return account
|
||
|
||
async def add_balance(self, customer_id: int, delta: Decimal) -> FinAccount:
|
||
"""入账：balance += delta，原子自增后回读最新余额。"""
|
||
await self.db.execute(
|
||
update(FinAccount)
|
||
.where(FinAccount.customer_id == customer_id)
|
||
.values(
|
||
balance=FinAccount.balance + delta,
|
||
version=FinAccount.version + 1,
|
||
)
|
||
)
|
||
await self.db.commit()
|
||
return await self.get_by_customer_id(customer_id)
|
||
|
||
async def subtract_balance(
|
||
self, customer_id: int, delta: Decimal
|
||
) -> FinAccount | None:
|
||
"""出账：balance -= delta，可用余额（balance - frozen_amount）不足时返回 None。"""
|
||
result = await self.db.execute(
|
||
update(FinAccount)
|
||
.where(
|
||
FinAccount.customer_id == customer_id,
|
||
FinAccount.balance - FinAccount.frozen_amount >= delta,
|
||
)
|
||
.values(
|
||
balance=FinAccount.balance - delta,
|
||
version=FinAccount.version + 1,
|
||
)
|
||
)
|
||
await self.db.commit()
|
||
if result.rowcount == 0:
|
||
return None
|
||
return await self.get_by_customer_id(customer_id)
|
||
|
||
async def deduct_balance(self, customer_id: int, delta: Decimal) -> bool:
|
||
"""申购事务内扣款：balance -= delta（可用余额不足则不动）。
|
||
|
||
不 commit，由 service 层事务统一提交，保证「扣款 + 加仓」原子性。
|
||
可用余额（balance - frozen_amount）不足时返回 False。
|
||
"""
|
||
result = await self.db.execute(
|
||
update(FinAccount)
|
||
.where(
|
||
FinAccount.customer_id == customer_id,
|
||
FinAccount.balance - FinAccount.frozen_amount >= delta,
|
||
)
|
||
.values(
|
||
balance=FinAccount.balance - delta,
|
||
version=FinAccount.version + 1,
|
||
)
|
||
)
|
||
return result.rowcount > 0
|
||
|
||
async def credit_balance(self, customer_id: int, delta: Decimal) -> bool:
|
||
"""赎回事务内入账：balance += delta。
|
||
|
||
不 commit，由 service 层事务统一提交，保证「减仓 + 入账」原子性。
|
||
"""
|
||
result = await self.db.execute(
|
||
update(FinAccount)
|
||
.where(FinAccount.customer_id == customer_id)
|
||
.values(
|
||
balance=FinAccount.balance + delta,
|
||
version=FinAccount.version + 1,
|
||
)
|
||
)
|
||
return result.rowcount > 0
|