feat(v2): crypto 预留表冷启动兜底——注入 loader 从 pending attempts 重建 reservation(兜重启丢内存)
This commit is contained in:
@@ -49,15 +49,29 @@ type Provider struct {
|
||||
baseURL string
|
||||
http *http.Client
|
||||
now func() time.Time
|
||||
loader ReservationLoader // 冷启动预留重建源(装配期注入,nil=不重建)
|
||||
|
||||
mu sync.Mutex
|
||||
reserved map[string]time.Time // "<address>/<amount>" → 预留到期(链上匹配维度,对齐 Query 的 to==addr)(canonical AmountRecentlyUsed 的进程内等价)
|
||||
}
|
||||
|
||||
// PendingReservation 冷启动重建一笔预留所需的最小信息(中性结构,crypto 不依赖 store)。
|
||||
type PendingReservation struct {
|
||||
AccountID string // 收款账户(用于解析地址,链上匹配维度)
|
||||
AmountMinor int64 // attempt 冻结的 base 金额(不含尾数)
|
||||
ProviderRef string // "CRYPTO-<OutTradeNo>-<tail>",用于恢复尾数
|
||||
ReservedAt time.Time // 建单时间(= attempt.CreatedAt),冷却窗自此算
|
||||
}
|
||||
|
||||
// ReservationLoader 返回当前仍活跃(pending)的 crypto 预留。装配期由 main 用 OrderStore 实现。
|
||||
type ReservationLoader func(ctx context.Context) ([]PendingReservation, error)
|
||||
|
||||
type Option func(*Provider)
|
||||
|
||||
func WithBaseURL(u string) Option { return func(p *Provider) { p.baseURL = u } }
|
||||
func WithHTTPClient(c *http.Client) Option { return func(p *Provider) { p.http = c } }
|
||||
func WithBaseURL(u string) Option { return func(p *Provider) { p.baseURL = u } }
|
||||
func WithHTTPClient(c *http.Client) Option { return func(p *Provider) { p.http = c } }
|
||||
func WithReservationLoader(l ReservationLoader) Option { return func(p *Provider) { p.loader = l } }
|
||||
func WithNow(f func() time.Time) Option { return func(p *Provider) { p.now = f } }
|
||||
|
||||
func New(accts *accounts.Registry, opts ...Option) *Provider {
|
||||
p := &Provider{
|
||||
@@ -151,6 +165,40 @@ func tailFromRef(ref string) (int64, error) {
|
||||
return strconv.ParseInt(ref[i+1:], 10, 64)
|
||||
}
|
||||
|
||||
// Warm 冷启动兜底:把仍在冷却窗内的活跃预留灌回内存表,兜住重启丢 map 导致的金额复用误配。
|
||||
// 幂等:只加不覆盖更早到期时间;冷却已过的跳过。装配期在起服务前调一次即可。
|
||||
func (p *Provider) Warm(ctx context.Context) error {
|
||||
if p.loader == nil {
|
||||
return nil
|
||||
}
|
||||
items, err := p.loader(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
now := p.now()
|
||||
p.mu.Lock()
|
||||
defer p.mu.Unlock()
|
||||
for _, it := range items {
|
||||
addr, err := p.address(it.AccountID) // 地址是链上匹配维度真相源
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
tail, err := tailFromRef(it.ProviderRef)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
until := it.ReservedAt.Add(amountCooldown)
|
||||
if !until.After(now) {
|
||||
continue // 冷却已过,金额可安全复用,无需恢复
|
||||
}
|
||||
key := addr + "/" + strconv.FormatInt(it.AmountMinor+tail, 10)
|
||||
if cur, ok := p.reserved[key]; !ok || until.After(cur) {
|
||||
p.reserved[key] = until
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *Provider) Create(_ context.Context, req provider.CreateRequest) (*provider.Session, error) {
|
||||
if req.Currency != "USDT" {
|
||||
return nil, fmt.Errorf("crypto: 仅支持 USDT, got %s", req.Currency)
|
||||
|
||||
Reference in New Issue
Block a user