k2safe/go-omni
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

release: prepare v0.6.0-preview

Browse Source
This commit is contained in:
k2safe
2026-09-22 01:38:42 +08:00
parent 9c9c8ab59a
commit c5e415ca52
18 changed files with 797 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
transport/native/server_test.go
View File

@@ -749,7 +749,7 @@ func TestServerConnectionChurnReusesStateSafely(t *testing.T) {
application := omni.NewWithRouter(netrouter.New())
application.GET("/health", func(ctx *omni.Context) error { return ctx.OK("ok") })
server := native.New(application)
server.MaxConnections = 16
server.MaxConnections = 32
listener, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatal(err)
@@ -759,8 +759,10 @@ func TestServerConnectionChurnReusesStateSafely(t *testing.T) {
waitForServer(t, listener.Addr().String())
defer stopServer(t, server, listener, done)
const workers = 8
const requestsPerWorker = 100
// Repeated short-lived connections exercise accept/close, connection-pool
// reuse and the MaxConnections gate under a bursty workload.
const workers = 16
const requestsPerWorker = 200
errors := make(chan error, workers)
var group sync.WaitGroup
for worker := 0; worker < workers; worker++ {

72
transport/native/slow_client_test.go Normal file
View File

@@ -0,0 +1,72 @@
package native_test
import (
"context"
"io"
"net"
"net/http"
"sync/atomic"
"testing"
"time"
"git.yuntuo.hk/k2safe/go-omni/transport/native"
)
func TestServerSlowHeadersCloseBeforeHandler(t *testing.T) {
var handlerCalls atomic.Int32
server := native.New(http.HandlerFunc(func(http.ResponseWriter, *http.Request) {
handlerCalls.Add(1)
}))
server.ReadHeaderTimeout = 40 * time.Millisecond
listener, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatal(err)
}
done := make(chan error, 1)
go func() { done <- server.Serve(context.Background(), listener) }()
defer stopServer(t, server, listener, done)
waitForServer(t, listener.Addr().String())
conn, err := net.DialTimeout("tcp", listener.Addr().String(), time.Second)
if err != nil {
t.Fatal(err)
}
defer conn.Close()
if _, err := io.WriteString(conn, "GET /health HTTP/1.1\r\nHost: local"); err != nil {
t.Fatal(err)
}
requirePeerClose(t, conn)
if handlerCalls.Load() != 0 {
t.Fatal("handler ran before slow headers completed")
}
}
func TestServerSlowChunkedBodyClosesBeforeHandler(t *testing.T) {
var handlerCalls atomic.Int32
server := native.New(http.HandlerFunc(func(http.ResponseWriter, *http.Request) {
handlerCalls.Add(1)
}))
server.ReadTimeout = 40 * time.Millisecond
listener, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatal(err)
}
done := make(chan error, 1)
go func() { done <- server.Serve(context.Background(), listener) }()
defer stopServer(t, server, listener, done)
waitForServer(t, listener.Addr().String())
conn, err := net.DialTimeout("tcp", listener.Addr().String(), time.Second)
if err != nil {
t.Fatal(err)
}
defer conn.Close()
request := "POST /body HTTP/1.1\r\nHost: local\r\nTransfer-Encoding: chunked\r\n\r\n4\r\npa"
if _, err := io.WriteString(conn, request); err != nil {
t.Fatal(err)
}
requirePeerClose(t, conn)
if handlerCalls.Load() != 0 {
t.Fatal("handler ran before slow chunked body completed")
}
}