feat: expose native server stats

This commit is contained in:
2026-09-22 10:04:26 +08:00
parent ef98d1dd8d
commit 8e474a24d9
5 changed files with 94 additions and 0 deletions

View File

@@ -789,6 +789,57 @@ func TestServerConnectionLimitReleasesAfterRequest(t *testing.T) {
}
}
func TestServerStatsTrackLifecycle(t *testing.T) {
started := make(chan struct{})
release := make(chan struct{})
application := omni.NewWithRouter(netrouter.New())
application.GET("/hold", func(ctx *omni.Context) error {
close(started)
<-release
return ctx.OK("done")
})
server, listener, done := startServer(t, application)
defer stopServer(t, server, listener, done)
conn, err := net.Dial("tcp", listener.Addr().String())
if err != nil {
t.Fatal(err)
}
defer conn.Close()
waitForStats := func(check func(native.Stats) bool) {
t.Helper()
deadline := time.Now().Add(time.Second)
for time.Now().Before(deadline) {
if check(server.Stats()) {
return
}
time.Sleep(time.Millisecond)
}
t.Fatalf("stats condition not reached: %+v", server.Stats())
}
waitForStats(func(stats native.Stats) bool { return stats.ActiveConnections == 1 && stats.AcceptedConnections >= 1 })
if _, err := io.WriteString(conn, "GET /hold HTTP/1.1\r\nHost: test\r\nConnection: close\r\n\r\n"); err != nil {
t.Fatal(err)
}
select {
case <-started:
case <-time.After(time.Second):
t.Fatal("hold handler did not start")
}
waitForStats(func(stats native.Stats) bool { return stats.ActiveRequests == 1 })
close(release)
response, err := http.ReadResponse(bufio.NewReader(conn), &http.Request{Method: http.MethodGet})
if err != nil {
t.Fatal(err)
}
response.Body.Close()
if response.StatusCode != http.StatusOK {
t.Fatalf("unexpected hold response: %d", response.StatusCode)
}
waitForStats(func(stats native.Stats) bool { return stats.ActiveRequests == 0 })
waitForStats(func(stats native.Stats) bool { return stats.ActiveConnections == 0 })
}
func TestServerConnectionChurnReusesStateSafely(t *testing.T) {
application := omni.NewWithRouter(netrouter.New())
application.GET("/health", func(ctx *omni.Context) error { return ctx.OK("ok") })