feat(v2): LimitAware 真实用量源——DBUsageSource 快照 + 对账 job 周期聚合当日已收(替 NopUsage)

2026-07-10 17:32:48 +08:00
parent e04cd73983
commit 4ac4e6ca9f
4 changed files with 142 additions and 0 deletions
@@ -94,3 +94,32 @@ func TestExpireStaleOrders(t *testing.T) {
 		t.Fatalf("重跑应 0, got %d", n2)
 	}
 }
+
+func TestSumPaidAttemptMinorByAccountSince(t *testing.T) {
+	db := model.OpenTestDB(t)
+	s := store.NewOrderStore(db)
+	now := time.Date(2026, 7, 10, 12, 0, 0, 0, time.UTC)
+	dayStart := time.Date(2026, 7, 10, 0, 0, 0, 0, time.UTC)
+
+	mk := func(no, acct string, minor int64, st model.AttemptStatus, paidAgo time.Duration) {
+		paid := now.Add(-paidAgo)
+		a := &model.Attempt{OutTradeNo: no, Channel: "alipay", AccountID: acct, ProviderRef: "R-" + no,
+			AmountMinor: minor, Currency: "CNY", Status: st, PaidAt: &paid}
+		if err := s.CreateAttempt(a); err != nil {
+			t.Fatalf("attempt %s: %v", no, err)
+		}
+	}
+	mk("A", "acct-1", 10000, model.AttemptPaid, 1*time.Hour)      // 今日,计入
+	mk("B", "acct-1", 5000, model.AttemptPaid, 2*time.Hour)       // 今日,计入 → acct-1=15000
+	mk("C", "acct-2", 7000, model.AttemptPaid, 30*time.Minute)    // 今日 acct-2=7000
+	mk("D", "acct-1", 9999, model.AttemptPending, 10*time.Minute) // 未付,不计
+	mk("E", "acct-1", 8888, model.AttemptPaid, 20*time.Hour)      // 昨天(paid_at < dayStart),不计
+
+	got, err := s.SumPaidAttemptMinorByAccountSince(dayStart)
+	if err != nil {
+		t.Fatalf("sum: %v", err)
+	}
+	if got["acct-1"] != 15000 || got["acct-2"] != 7000 {
+		t.Fatalf("聚合 = %+v, want acct-1=15000 acct-2=7000", got)
+	}
+}