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,144 @@
|
||||
package notify
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"io"
|
||||
"net"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func BenchmarkRawTCPLocalhostThroughput(b *testing.B) {
|
||||
cases := []struct {
|
||||
name string
|
||||
payloadSize int
|
||||
}{
|
||||
{
|
||||
name: "raw_64KiB",
|
||||
payloadSize: 64 * 1024,
|
||||
},
|
||||
{
|
||||
name: "raw_256KiB",
|
||||
payloadSize: 256 * 1024,
|
||||
},
|
||||
{
|
||||
name: "raw_512KiB",
|
||||
payloadSize: 512 * 1024,
|
||||
},
|
||||
{
|
||||
name: "raw_1MiB",
|
||||
payloadSize: 1024 * 1024,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range cases {
|
||||
b.Run(tc.name, func(b *testing.B) {
|
||||
benchmarkRawTCPLocalhostThroughput(b, tc.payloadSize)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func benchmarkRawTCPLocalhostThroughput(b *testing.B, payloadSize int) {
|
||||
b.Helper()
|
||||
|
||||
listener, err := net.Listen("tcp", "127.0.0.1:0")
|
||||
if err != nil {
|
||||
b.Fatalf("net.Listen failed: %v", err)
|
||||
}
|
||||
b.Cleanup(func() {
|
||||
_ = listener.Close()
|
||||
})
|
||||
|
||||
acceptCh := make(chan net.Conn, 1)
|
||||
acceptErrCh := make(chan error, 1)
|
||||
go func() {
|
||||
conn, err := listener.Accept()
|
||||
if err != nil {
|
||||
acceptErrCh <- err
|
||||
return
|
||||
}
|
||||
acceptCh <- conn
|
||||
}()
|
||||
|
||||
clientConn, err := net.Dial("tcp", listener.Addr().String())
|
||||
if err != nil {
|
||||
b.Fatalf("net.Dial failed: %v", err)
|
||||
}
|
||||
b.Cleanup(func() {
|
||||
_ = clientConn.Close()
|
||||
})
|
||||
if tcpConn, ok := clientConn.(*net.TCPConn); ok {
|
||||
_ = tcpConn.SetNoDelay(true)
|
||||
}
|
||||
|
||||
var serverConn net.Conn
|
||||
select {
|
||||
case conn := <-acceptCh:
|
||||
serverConn = conn
|
||||
case err := <-acceptErrCh:
|
||||
b.Fatalf("Accept failed: %v", err)
|
||||
case <-time.After(5 * time.Second):
|
||||
b.Fatal("timed out waiting for accept")
|
||||
}
|
||||
b.Cleanup(func() {
|
||||
if serverConn != nil {
|
||||
_ = serverConn.Close()
|
||||
}
|
||||
})
|
||||
|
||||
drainDone := make(chan error, 1)
|
||||
go func() {
|
||||
_, err := io.Copy(io.Discard, serverConn)
|
||||
if err != nil && !errors.Is(err, io.EOF) {
|
||||
drainDone <- err
|
||||
return
|
||||
}
|
||||
drainDone <- nil
|
||||
}()
|
||||
|
||||
payload := make([]byte, payloadSize)
|
||||
for i := range payload {
|
||||
payload[i] = byte(i)
|
||||
}
|
||||
|
||||
b.ReportAllocs()
|
||||
b.SetBytes(int64(payloadSize))
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
if err := benchmarkRawTCPWriteFull(clientConn, payload); err != nil {
|
||||
b.Fatalf("raw tcp write failed at iter %d: %v", i, err)
|
||||
}
|
||||
}
|
||||
b.StopTimer()
|
||||
|
||||
if tcpConn, ok := clientConn.(*net.TCPConn); ok {
|
||||
_ = tcpConn.CloseWrite()
|
||||
} else {
|
||||
_ = clientConn.Close()
|
||||
}
|
||||
|
||||
select {
|
||||
case err := <-drainDone:
|
||||
if err != nil {
|
||||
b.Fatalf("server drain failed: %v", err)
|
||||
}
|
||||
case <-time.After(10 * time.Second):
|
||||
b.Fatal("timed out waiting for server drain")
|
||||
}
|
||||
}
|
||||
|
||||
func benchmarkRawTCPWriteFull(conn net.Conn, payload []byte) error {
|
||||
for len(payload) > 0 {
|
||||
n, err := conn.Write(payload)
|
||||
if n > 0 {
|
||||
payload = payload[n:]
|
||||
}
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if n == 0 {
|
||||
return io.ErrNoProgress
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user