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

@ -189,7 +189,27 @@ func (h *HMS) Send(ctx context.Context, tokens []string, n NewMail) error {
if len(tokens) == 0 {
return nil
}
if !h.reserveDaily(len(tokens)) {
/*
* ★★ 2026-09-26 修两件事(用户「推送配额是不是按设备数的倍数在掉」):
*
* ① **计数单位错**:原来传的是 `len(tokens)`,而变量名、注释与报错文案
* (「达到每日推送上限 N **条**」)说的都是"条"。
* 华为的测试消息额度是**按 messages:send 的调用次数**(一次请求一条消息,
* 无论 `message.token[]` 里有几个设备)计的 ⇒
* **3 个设备收到 1 封邮件就吃掉 3 条额度,实际只发出 1 条**。
* 多设备自部署用户会按 1/设备数 的速度提前耗尽 1000 条/天。
* ⇒ 改按批次计:一次 `Send` = 一条消息。
*
* ② **扣在投递之前**:`accessToken` 失败 / HTTP 失败 / 华为回非成功码,
* 这些**一条都没发出去**,额度却已经扣了,而失败只 `log.Printf`
* ⇒ 一次网络抖动静默烧掉配额。
* ⇒ 挪到 `accessToken` **之后**、真正发请求之前。
*
* ★ 为什么不是"发送成功后再扣":那会超发(并发下多个 goroutine 都能通过检查)。
* 保留前置预留、但放在"确认能发"之后,是**宁可少算也不多发**的取舍 ——
* 少算的代价是偶尔一次失败没计入,超发的代价是真超额被华为拒。
*/
if !h.reserveDaily(1) {
return fmt.Errorf("达到每日推送上限 %d 条(HMS_DAILY_LIMIT)", h.DailyLimit)
}
tok, err := h.accessToken(ctx)
@ -257,6 +277,12 @@ func (h *HMS) Send(ctx context.Context, tokens []string, n NewMail) error {
// reserveDaily 记一次每日用量。超上限时返回 false(不把额度打光:
// 打光之后的失败响应刷日志,而且真需要的那条也发不出去)。
/*
* reserveDaily 预留 n **条消息**的当日额度(不是 n 个 token)。
*
* ★ 单位是"条":华为按 `messages:send` 的调用次数计,`token[]` 里有几个设备
* 都算一条。调用点已按 1 传(见 `Send` 里的注释)。
*/
func (h *HMS) reserveDaily(n int) bool {
h.dayMu.Lock()
defer h.dayMu.Unlock()

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