fix(notify): waitgroup-based dispatcher shutdown, withoutcancel persist

internal/infra/notify/dispatcher.go

@@ -6,6 +6,7 @@ package notify
 import (
 	"context"
 	"log/slog"
+	"sync"
 	"time"
 
 	"github.com/google/uuid"
@@ -13,12 +14,13 @@ import (
 
 // Dispatcher 是 notify 模块对外的发送门面。Repository 由调用方注入（生产
 // 用 pgRepo，测试用 fake），Notifier 通过 Register 动态追加，now 钩子留给
-// 测试覆盖时间相关行为。
+// 测试覆盖时间相关行为。wg 用于 Shutdown 时等待所有异步投递结束。
 type Dispatcher struct {
 	repo      Repository
 	notifiers []Notifier
 	log       *slog.Logger
 	now       func() time.Time
+	wg        sync.WaitGroup
 }
 
 // NewDispatcher 用 Repository 与 Logger 构造 Dispatcher。注册 Notifier 通过
@@ -36,6 +38,10 @@ func (d *Dispatcher) Register(n Notifier) {
 // Dispatch 同步落库；落库失败返回错误，不再异步投递。落库成功后，
 // fire-and-forget 派发给所有外部 notifier，单个 notifier 的失败仅记日志，
 // 不阻塞其它通道；调用方无法感知异步投递结果。
+//
+// 落库使用 context.WithoutCancel + 5s 硬超时，避免请求 ctx 取消导致通知
+// 漏写；外部投递使用独立 background ctx + 30s 硬超时，lifecycle 由 Shutdown
+// 通过 wg 管控。
 func (d *Dispatcher) Dispatch(ctx context.Context, msg Message) error {
 	if msg.ID == uuid.Nil {
 		msg.ID = uuid.New()
@@ -43,14 +49,18 @@ func (d *Dispatcher) Dispatch(ctx context.Context, msg Message) error {
 	if msg.CreatedAt.IsZero() {
 		msg.CreatedAt = d.now()
 	}
-	if err := d.repo.Insert(ctx, msg); err != nil {
+	persistCtx, cancel := context.WithTimeout(context.WithoutCancel(ctx), 5*time.Second)
+	defer cancel()
+	if err := d.repo.Insert(persistCtx, msg); err != nil {
 		return err
 	}
 	for _, n := range d.notifiers {
+		d.wg.Add(1)
 		go func(n Notifier) {
-			// 异步投递使用独立 ctx，避免请求 ctx 取消导致投递中断；
-			// 真实 notifier 应在内部加超时限制。
-			if err := n.Send(context.Background(), msg); err != nil {
+			defer d.wg.Done()
+			sendCtx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+			defer cancel()
+			if err := n.Send(sendCtx, msg); err != nil {
 				d.log.Error("notify channel failed",
 					"channel", n.Name(),
 					"topic", msg.Topic,
@@ -61,3 +71,20 @@ func (d *Dispatcher) Dispatch(ctx context.Context, msg Message) error {
 	}
 	return nil
 }
+
+// Shutdown 阻塞等待所有异步投递完成。调用方应保证此后不再有新的 Dispatch
+// 调用（通常上游 HTTP server 已经 Shutdown）。ctx 用作硬截止：超时会返回
+// ctx.Err()，但泄漏的 goroutine 会继续跑到 Send 的 30s timeout 自然结束。
+func (d *Dispatcher) Shutdown(ctx context.Context) error {
+	done := make(chan struct{})
+	go func() {
+		d.wg.Wait()
+		close(done)
+	}()
+	select {
+	case <-done:
+		return nil
+	case <-ctx.Done():
+		return ctx.Err()
+	}
+}