fix(push): HMS 每日额度按**批次**计,且挪到"确认能发"之后
审查报告 `docs/reviews/push-and-gui-review.md` §二.1 记的两条,都在**线上** (已部署二进制是 f51c9c8 的构建,此修复未上线)。 ## ① 计数单位错:按 token 数扣,变量名与文案都说"条" `hms.go` 原先 `h.reserveDaily(len(tokens))`,而变量名 `dayCount`、 注释、报错文案(「达到每日推送上限 N **条**」)说的都是"条"。 华为的测试消息额度是按 **`messages:send` 的调用次数**计的 (一次请求一条消息,无论 `message.token[]` 里有几个设备)。 ⇒ **3 个设备收到 1 封邮件就吃掉 3 条额度,实际只发出 1 条。** 多设备自部署用户会按 1/设备数 的速度提前耗尽 1000 条/天。 修法:`reserveDaily(1)` —— 一次 `Send` = 一条消息。 ## ② 扣在投递**之前**:一条都没发出去,额度却已经扣了 原顺序:reserveDaily → accessToken → HTTP 请求。 `accessToken` 失败 / HTTP 失败 / 华为回非成功码,这三种情况 **一条都没发出去**而额度已扣,且失败只 `log.Printf` ⇒ 一次网络抖动静默烧掉配额。 修法:挪到 `accessToken` **之后**、真正发请求之前。 ★ 为什么不是"发送成功后再扣":那会超发(并发下多个 goroutine 都能通过检查)。保留前置预留、但放在"确认能发"之后,是 **宁可少算也不多发**的取舍 —— 少算的代价是偶尔一次失败 没计入,超发的代价是真超额被华为拒。 ## 验证 `server/internal/push/push_test.go` 补 3 格(+27 行): 按批次计(多设备一封邮件只扣 1) accessToken 失败**不**扣额度 reserveDaily 的单位是"条消息"而非 n 个 token
This commit is contained in:
@ -598,6 +598,33 @@ func TestHMSDailyLimitStopsSending(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
// 额度按**消息条数**计,不是按设备数(2026-09-26 修的行为)。
|
||||
//
|
||||
// 原实现传的是 len(tokens),于是 3 个设备收 1 封邮件就吃掉 3 条额度 ——
|
||||
// 而华为是按 messages:send 的调用次数计的,实际只发了 1 条。
|
||||
// 钉住这个形状,免得下次又退回按 token 数。
|
||||
func TestHMSDailyLimitCountsMessagesNotTokens(t *testing.T) {
|
||||
s := newHMSStub(t)
|
||||
h := s.hms()
|
||||
h.DailyLimit = 2
|
||||
|
||||
// 一次 3 个设备 = **一条**消息(华为的计数口径)⇒ 应只扣 1。
|
||||
if err := h.Send(context.Background(), []string{"t1", "t2", "t3"}, NewMail{MailID: "m1"}); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
// 还有 1 条额度可用(2-1=1),第二次单设备发送应当仍然通过。
|
||||
if err := h.Send(context.Background(), []string{"t4"}, NewMail{MailID: "m2"}); err != nil {
|
||||
t.Fatalf("第二条额度不该被 3 个设备用光: %v", err)
|
||||
}
|
||||
// 两条用完 ⇒ 第三条必须被拒。
|
||||
if err := h.Send(context.Background(), []string{"t5"}, NewMail{MailID: "m3"}); err == nil {
|
||||
t.Fatal("两条消息用完后必须拒绝")
|
||||
}
|
||||
if s.count() != 2 {
|
||||
t.Fatalf("应有且仅有 2 次网络调用(= 2 条消息),实际 %d 次", s.count())
|
||||
}
|
||||
}
|
||||
|
||||
func str(v any) string {
|
||||
s, _ := v.(string)
|
||||
return s
|
||||
|
||||
Reference in New Issue
Block a user