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:
2026-09-28 08:26:41 +08:00
parent e2472287f0
commit 044a664cc3
2 changed files with 54 additions and 1 deletions

View File

@ -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