diff --git a/server/internal/push/hms.go b/server/internal/push/hms.go index a408bd2..3b62843 100644 --- a/server/internal/push/hms.go +++ b/server/internal/push/hms.go @@ -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() diff --git a/server/internal/push/push_test.go b/server/internal/push/push_test.go index a425fe5..275d014 100644 --- a/server/internal/push/push_test.go +++ b/server/internal/push/push_test.go @@ -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