feat(notify): 重构通信内核并补齐 stream/bulk/record/transfer 能力

- 引入 LogicalConn/TransportConn 分层，ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
2026-04-15 15:24:36 +08:00
parent d14d13c393
commit 09d972c7b7
216 changed files with 51374 additions and 1715 deletions
@@ -0,0 +1,193 @@
+package notify
+
+import "testing"
+
+func TestFileAckPoolPreparedWaitConsumesEarlyAck(t *testing.T) {
+	pool := newFileAckPool()
+	wait := pool.prepare("client:a", "file-1", "chunk", 64)
+
+	ok := pool.deliver("client:a", FileEvent{
+		Packet: FilePacket{
+			FileID: "file-1",
+			Stage:  "chunk",
+			Offset: 64,
+		},
+	})
+	if !ok {
+		t.Fatalf("deliver should match prepared waiter")
+	}
+
+	if err := pool.waitPrepared(wait, defaultFileAckTimeout); err != nil {
+		t.Fatalf("waitPrepared failed: %v", err)
+	}
+}
+
+func TestFileAckPoolPreparedWaitReturnsAckError(t *testing.T) {
+	pool := newFileAckPool()
+	wait := pool.prepare("client:a", "file-2", "meta", 0)
+
+	ok := pool.deliver("client:a", FileEvent{
+		Packet: FilePacket{
+			FileID: "file-2",
+			Stage:  "meta",
+			Offset: 0,
+			Error:  "checksum mismatch",
+		},
+	})
+	if !ok {
+		t.Fatalf("deliver should match prepared waiter")
+	}
+
+	err := pool.waitPrepared(wait, defaultFileAckTimeout)
+	if err == nil {
+		t.Fatal("waitPrepared should return ack error")
+	}
+	if got, want := err.Error(), "checksum mismatch"; got != want {
+		t.Fatalf("unexpected error: %v", err)
+	}
+}
+
+func TestFileAckPoolCancelRemovesPreparedWaiter(t *testing.T) {
+	pool := newFileAckPool()
+	wait := pool.prepare("client:a", "file-3", "end", 0)
+	wait.cancel()
+
+	ok := pool.deliver("client:a", FileEvent{
+		Packet: FilePacket{
+			FileID: "file-3",
+			Stage:  "end",
+			Offset: 0,
+		},
+	})
+	if ok {
+		t.Fatal("deliver should not match canceled waiter")
+	}
+}
+
+func TestFileAckPoolScopeIsolation(t *testing.T) {
+	pool := newFileAckPool()
+	waitA := pool.prepare("server:client-a", "file-4", "chunk", 128)
+	waitB := pool.prepare("server:client-b", "file-4", "chunk", 128)
+
+	ok := pool.deliver("server:client-a", FileEvent{
+		Packet: FilePacket{
+			FileID: "file-4",
+			Stage:  "chunk",
+			Offset: 128,
+		},
+	})
+	if !ok {
+		t.Fatal("deliver should match scopeA waiter")
+	}
+
+	if err := pool.waitPrepared(waitA, defaultFileAckTimeout); err != nil {
+		t.Fatalf("waitPrepared scopeA failed: %v", err)
+	}
+
+	ok = pool.deliver("server:client-a", FileEvent{
+		Packet: FilePacket{
+			FileID: "file-4",
+			Stage:  "chunk",
+			Offset: 128,
+		},
+	})
+	if ok {
+		t.Fatal("scopeA ack should not consume scopeB waiter")
+	}
+
+	ok = pool.deliver("server:client-b", FileEvent{
+		Packet: FilePacket{
+			FileID: "file-4",
+			Stage:  "chunk",
+			Offset: 128,
+		},
+	})
+	if !ok {
+		t.Fatal("deliver should match scopeB waiter")
+	}
+
+	if err := pool.waitPrepared(waitB, defaultFileAckTimeout); err != nil {
+		t.Fatalf("waitPrepared scopeB failed: %v", err)
+	}
+}
+
+func TestFileAckPoolCloseAllCancelsPreparedWaiters(t *testing.T) {
+	pool := newFileAckPool()
+	wait := pool.prepare("client:a", "file-5", "chunk", 256)
+
+	pool.closeAll()
+
+	err := pool.waitPrepared(wait, defaultFileAckTimeout)
+	if err == nil {
+		t.Fatal("waitPrepared should return cancel error after closeAll")
+	}
+	if got, want := err.Error(), "file ack canceled"; got != want {
+		t.Fatalf("unexpected error after closeAll: got %q want %q", got, want)
+	}
+}
+
+func TestFileAckPoolCloseScopeCancelsMatchingWaitersOnly(t *testing.T) {
+	pool := newFileAckPool()
+	waitA := pool.prepare("server:client-a", "file-6", "chunk", 256)
+	waitB := pool.prepare("server:client-b", "file-6", "chunk", 256)
+
+	pool.closeScope("server:client-a")
+
+	err := pool.waitPrepared(waitA, defaultFileAckTimeout)
+	if err == nil {
+		t.Fatal("scopeA waiter should be canceled")
+	}
+	if got, want := err.Error(), "file ack canceled"; got != want {
+		t.Fatalf("unexpected scopeA error: got %q want %q", got, want)
+	}
+
+	ok := pool.deliver("server:client-b", FileEvent{
+		Packet: FilePacket{
+			FileID: "file-6",
+			Stage:  "chunk",
+			Offset: 256,
+		},
+	})
+	if !ok {
+		t.Fatal("scopeB waiter should remain deliverable")
+	}
+
+	if err := pool.waitPrepared(waitB, defaultFileAckTimeout); err != nil {
+		t.Fatalf("waitPrepared scopeB failed: %v", err)
+	}
+}
+
+func TestServerRemoveClientClosesScopedFileAckWaiters(t *testing.T) {
+	server := NewServer().(*ServerCommon)
+	clientA := &ClientConn{ClientID: "client-a"}
+	clientB := &ClientConn{ClientID: "client-b"}
+	pool := server.getFileAckPool()
+
+	waitA := pool.prepare(serverFileScope(clientA), "file-7", "end", 0)
+	waitB := pool.prepare(serverFileScope(clientB), "file-7", "end", 0)
+
+	server.removeClient(clientA)
+
+	err := pool.waitPrepared(waitA, defaultFileAckTimeout)
+	if err == nil {
+		t.Fatal("clientA waiter should be canceled when client is removed")
+	}
+	if got, want := err.Error(), "file ack canceled"; got != want {
+		t.Fatalf("unexpected clientA error: got %q want %q", got, want)
+	}
+
+	ok := pool.deliver(serverFileScope(clientB), FileEvent{
+		Packet: FilePacket{
+			FileID: "file-7",
+			Stage:  "end",
+			Offset: 0,
+		},
+	})
+	if !ok {
+		t.Fatal("clientB waiter should remain deliverable")
+	}
+
+	if err := pool.waitPrepared(waitB, defaultFileAckTimeout); err != nil {
+		t.Fatalf("waitPrepared clientB failed: %v", err)
+	}
+}