feat(notify): 重构通信内核并补齐 stream/bulk/record/transfer 能力
- 引入 LogicalConn/TransportConn 分层,ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
This commit is contained in:
@@ -0,0 +1,55 @@
|
||||
package notify
|
||||
|
||||
import (
|
||||
"b612.me/stario"
|
||||
"context"
|
||||
"fmt"
|
||||
"math"
|
||||
"net"
|
||||
"sync/atomic"
|
||||
"testing"
|
||||
)
|
||||
|
||||
var testAcceptedPeerSeq atomic.Uint64
|
||||
|
||||
func newRunningPeerAttachServerForTest(t *testing.T, configure func(*ServerCommon)) *ServerCommon {
|
||||
t.Helper()
|
||||
server := NewServer().(*ServerCommon)
|
||||
if configure != nil {
|
||||
configure(server)
|
||||
}
|
||||
stopCtx, stopFn := context.WithCancel(context.Background())
|
||||
queue := stario.NewQueueCtx(stopCtx, 8, math.MaxUint32)
|
||||
server.setServerSessionRuntime(&serverSessionRuntime{
|
||||
stopCtx: stopCtx,
|
||||
stopFn: stopFn,
|
||||
queue: queue,
|
||||
})
|
||||
server.markSessionStarted()
|
||||
|
||||
transportCtx, transportStop := context.WithCancel(context.Background())
|
||||
go server.loadMessageLoop(nil, transportCtx, queue, nil, nil)
|
||||
|
||||
t.Cleanup(func() {
|
||||
transportStop()
|
||||
stopFn()
|
||||
})
|
||||
return server
|
||||
}
|
||||
|
||||
func bootstrapPeerAttachLogicalForTest(t *testing.T, server *ServerCommon, conn net.Conn) *LogicalConn {
|
||||
t.Helper()
|
||||
if server == nil {
|
||||
t.Fatal("server is nil")
|
||||
}
|
||||
id := fmt.Sprintf("accepted-%d", testAcceptedPeerSeq.Add(1))
|
||||
logical := server.bootstrapAcceptedLogical(id, conn.RemoteAddr(), conn)
|
||||
if logical == nil {
|
||||
t.Fatal("bootstrapAcceptedLogical returned nil")
|
||||
}
|
||||
return logical
|
||||
}
|
||||
|
||||
func bootstrapPeerAttachConnForTest(t *testing.T, server *ServerCommon, conn net.Conn) *ClientConn {
|
||||
return clientConnFromLogical(bootstrapPeerAttachLogicalForTest(t, server, conn))
|
||||
}
|
||||
Reference in New Issue
Block a user