Go 怎样用 ConnState 统计 HTTP 连接状态

Go 可以在 http.Server.ConnState 中接收连接状态回调，并用“连接对象→最近状态”的表统计当前分布。关键不是每遇到 StateActive 就加一，而是先扣除同一连接的旧状态，再计入新状态；关闭和接管时删除记录。

官方地址：https://pkg.go.dev/net/http#ConnState

我会先问：现在看到的是连接，还是请求？

面对“请求结束了，活跃数却没有归零”的现象，我不会先怀疑服务端泄漏，而会先确认统计口径。这里用一个假设排障场景讲取舍，不是某次线上事故复盘，也没有本机压测数字。

我的第一选择是把指标分成两类：当前状态数使用 gauge 思路，状态回调累计次数使用 counter 思路。二者都不直接等于请求总量。HTTP/2 可在一个连接上承载多个并发请求，ConnState 只报告连接整体的活跃边界，不会为每个请求单独通知。

先把状态桶的含义写清楚

我特别保留接管计数，因为接管后不会再等到 StateClosed 回调。若只在关闭时清理，统计表就可能长期持有已脱离 HTTP 服务管理的连接。表的含义是“仍由本统计器跟踪的 HTTP 连接”，不是进程的全部 socket。

用一把锁维护旧状态与新状态

下面的统计器只保存 New、Active、Idle，避免保存无限增长的终态对象。以 net.Conn 对象作键，不用 RemoteAddr 字符串代替连接身份。不同连接的回调与快照读取都可能并发，因此更新操作必须作为整体加锁。

package connstats

import (
    "net"
    "net/http"
    "sync"
)

type Snapshot struct {
    New, Active, Idle int
    ClosedTotal, HijackedTotal uint64
}

type Stats struct {
    mu sync.Mutex
    states map[net.Conn]http.ConnState
    closed, hijacked uint64
}

// Observe 应在 Server 启动前注册；回调里只做短小的内存更新。
func (s *Stats) Observe(c net.Conn, state http.ConnState) {
    s.mu.Lock()
    defer s.mu.Unlock()
    if s.states == nil {
        s.states = make(map[net.Conn]http.ConnState)
    }
    switch state {
    case http.StateNew, http.StateActive, http.StateIdle:
        // 同一连接覆盖旧状态，不把重复回调算成第二条连接。
        s.states[c] = state
    case http.StateClosed:
        delete(s.states, c)
        s.closed++
    case http.StateHijacked:
        // 接管也是终态，不能等一个不会到来的关闭回调。
        delete(s.states, c)
        s.hijacked++
    }
}

// Snapshot 在同一锁内扫描，避免分别读取产生跨时刻组合。
func (s *Stats) Snapshot() Snapshot {
    s.mu.Lock()
    defer s.mu.Unlock()
    out := Snapshot{ClosedTotal: s.closed, HijackedTotal: s.hijacked}
    for _, state := range s.states {
        switch state {
        case http.StateNew:
            out.New++
        case http.StateActive:
            out.Active++
        case http.StateIdle:
            out.Idle++
        }
    }
    return out
}

这里不是手工维护三组加减法，而是覆盖最近状态、快照时重新分桶；效果上避免了旧桶未扣除的问题。代价是快照扫描与连接数成正比。我更愿意先选择容易解释的版本，只有采样开销确实成为问题，再考虑增量计数。

累计终态数依赖标准服务器正常发送终态回调的约定；不要把同一个终态事件手工重复调用当作新连接。这个统计器应专属于一个从启动起就挂载回调的 Server，不能中途挂载后声称覆盖了先前全部连接。

注册回调时，不在锁内输出日志

package connstats

import "net/http"

// NewServer 只负责组装；返回后由调用方启动并管理关闭。
func NewServer(handler http.Handler, stats *Stats) *http.Server {
    if stats == nil {
        // 保证回调有独立统计对象，避免空指针。
        stats = &Stats{}
    }
    return &http.Server{
        Addr: "127.0.0.1:8080",
        Handler: handler,
        ConnState: stats.Observe,
    }
}

真正接入项目时，我会先创建并保留 Stats 指针，再传入工厂函数，便于读取快照。已有 ConnState 时要明确组合旧回调，而不是悄悄覆盖。慢日志、网络推送和格式化大对象不放进 Observe 的锁内；本例也不负责完整指标导出系统。

怎样区分实现错误与口径误读

我会先给同一个连接身份依次输入 New、Active、Idle，分别取快照；这是供读者进行的合成检查，预期每次只有一个当前桶为一。再输入 Closed，应只增加关闭累计数，当前三桶为零；另外用独立连接检查 Hijacked 的清理行为。本文没有把这些预期写成已执行结果。

随后再观察真实 HTTP/1 的连接复用，以及真实 HTTP/2 的多请求共享连接。不要期望每个请求都增加 ClosedTotal，也不要用 Active 变化次数重建请求数。如果快照暴露在同一 HTTP 服务上，采集请求本身也可能占用一个 Active 连接，这不是统计器凭空多算。

若 Idle 偏多，我会先看连接复用和客户端池；若 New 长期偏多，再检查请求是否迟迟没进入处理阶段。仅凭一个桶不能直接判断根因，更不能把 HTTP 连接表当成操作系统连接总表。

我的取舍与常见问题

为什么没有保存每个已关闭连接？

累计值已经满足数量观察。保存全部终态键会让内存随着历史连接量增加，反而制造新的排障负担。

为什么不使用多个独立原子变量？

原子加减不能自动解决旧状态归属和一致快照。小型统计器使用一把锁更容易维护；快照频率与连接规模上去后，才值得另做权衡。

能用 ConnState 统计处理中请求吗？

不能用它替代逐请求计数，特别是 HTTP/2。我的建议是：连接分布交给 ConnState，请求生命周期交给 Handler 层，两个口径分别命名。

完整状态含义见官方标准库说明：https://pkg.go.dev/net/http#ConnState。这套简化实现适合观察连接分布，不承诺完成整套监控或故障归因。