b612
1f2e74acca
- 完善 stream/bulk DataID 分配、预留和双向命名空间，修复并发打开及 dedicated/shared 回退时的 ID 冲突 - 将收发、回复、恢复任务和 sidecar 绑定原始会话与物理连接，防止重连后的旧消息误操作新连接 - 加强 close/reset 身份校验及实例移除检查，修复 dedicated attach 失败、通道引用和资源回收竞态 - 收紧批量发送器停止准入，确保在途入队完成后统一清理请求、缓冲区和等待者 - 修复 record 满队列死锁、取消时序号消耗及关闭竞态，确保关闭有界并返回真实错误 - 增加协商式 record 逻辑半关闭，保留反向 ACK；通过 reset 传递 RecordFailure，避免背压掩盖原始失败原因 - 补齐帧长度、批次数量、序号溢出和未确认窗口校验，提前拒绝超限数据并按字节预算拆批 - 为入站分发增加全局及单连接的条数、字节预算和阻塞背压，关闭时唤醒等待者，消除正常断连日志噪音 - 完善 bulk 窗口释放失败处理与传输诊断，补充并发、重连、背压、协议边界及真实 TCP 回归覆盖
65 lines
2.2 KiB
Go
65 lines
2.2 KiB
Go
package notify
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"io"
|
|
"testing"
|
|
)
|
|
|
|
func TestStreamRuntimeAdoptFailureFinalizesCandidate(t *testing.T) {
|
|
runtime := newStreamRuntime("cstr")
|
|
scope := clientFileScope()
|
|
existing := newStreamHandle(context.Background(), runtime, scope, StreamOpenRequest{
|
|
StreamID: "duplicate",
|
|
DataID: 1,
|
|
}, 0, nil, nil, 0, nil, nil, nil, runtime.configSnapshot())
|
|
if err := runtime.register(scope, existing); err != nil {
|
|
t.Fatalf("register existing stream: %v", err)
|
|
}
|
|
defer existing.markReset(io.ErrClosedPipe)
|
|
|
|
candidate := newStreamHandle(context.Background(), runtime, scope, StreamOpenRequest{
|
|
StreamID: "duplicate",
|
|
DataID: 2,
|
|
}, 0, nil, nil, 0, nil, nil, nil, runtime.configSnapshot())
|
|
if err := runtime.adopt(scope, candidate); !errors.Is(err, errStreamAlreadyExists) {
|
|
t.Fatalf("adopt error = %v, want %v", err, errStreamAlreadyExists)
|
|
}
|
|
if err := candidate.resetErrSnapshot(); !errors.Is(err, errStreamAlreadyExists) {
|
|
t.Fatalf("candidate reset error = %v, want %v", err, errStreamAlreadyExists)
|
|
}
|
|
select {
|
|
case <-candidate.Context().Done():
|
|
default:
|
|
t.Fatal("failed stream adoption left candidate context active")
|
|
}
|
|
}
|
|
|
|
func TestStreamRuntimeStaleFinalizeDoesNotRemoveReplacement(t *testing.T) {
|
|
runtime := newStreamRuntime("cstr")
|
|
scope := clientFileScope()
|
|
old := newStreamHandle(context.Background(), runtime, scope, StreamOpenRequest{
|
|
StreamID: "reused",
|
|
DataID: 1,
|
|
}, 0, nil, nil, 0, nil, nil, nil, runtime.configSnapshot())
|
|
if err := runtime.register(scope, old); err != nil {
|
|
t.Fatalf("register old stream: %v", err)
|
|
}
|
|
old.markReset(errors.New("old failed"))
|
|
|
|
replacement := newStreamHandle(context.Background(), runtime, scope, StreamOpenRequest{
|
|
StreamID: "reused",
|
|
DataID: 2,
|
|
}, 0, nil, nil, 0, nil, nil, nil, runtime.configSnapshot())
|
|
if err := runtime.register(scope, replacement); err != nil {
|
|
t.Fatalf("register replacement stream: %v", err)
|
|
}
|
|
defer replacement.markReset(io.ErrClosedPipe)
|
|
|
|
old.markReset(errors.New("late duplicate reset"))
|
|
if got, ok := runtime.lookup(scope, "reused"); !ok || got != replacement {
|
|
t.Fatalf("replacement stream after stale finalize = %p/%v, want %p/true", got, ok, replacement)
|
|
}
|
|
}
|