跨机器场景下Go语言net/http无法检测客户端退出问题

问题分析 #

本地运行时，客户端断开连接会立刻触发TCP RST包，服务器的Write操作直接返回"broken pipe"错误；但跨机器或通过SSH隧道时，中间网络层（比如隧道缓冲、路由器连接保持机制）会延迟TCP连接的断开信号，导致服务器无法及时感知客户端已退出，持续发送数据造成资源浪费。

解决方案 #

1. 监听请求Context的取消信号 #

每个HTTP请求的r.Context()会在客户端断开连接时被取消，这是Go HTTP服务原生支持的最可靠检测方式。在循环中加入对Context Done通道的监听：

修改test函数的循环部分：

func test(w http.ResponseWriter, r *http.Request) {
    f, f_err := w.(http.Flusher)
    if !f_err {
        fmt.Printf("%s\n", "flush error")
        return
    }
    w.Header().Set("Cache-Control", "no-cache")
    w.Header().Set("Content-Type", "text/event-stream")
    w.WriteHeader(http.StatusOK)
    
    for timeout := 0; timeout < 1000; timeout++ {
        // 检查客户端是否断开连接
        select {
        case <-r.Context().Done():
            fmt.Printf("客户端已断开: %v\n", r.Context().Err())
            return
        default:
        }
        
        writ, err := w.Write([]byte("data: {\"result\": \"success\"}\n\n"))
        if err != nil {
            fmt.Printf("写入失败: %v\n", err)
            return
        }
        f.Flush()
        fmt.Printf("%d %d %s\n", timeout, writ, err)
        time.Sleep(time.Duration(1) * time.Second)
    }
}

2. 配置HTTP Server的超时与TCP KeepAlive #

通过设置服务器超时参数和TCP KeepAlive，让系统主动检测无效连接，加速断开感知：

修改main函数中的Server配置：

import (
    "fmt"
    "net"
    "net/http"
    "time"
)

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("/test", test)
    
    // 创建自定义Listener并设置TCP KeepAlive
    listener, err := net.Listen("tcp", ":3001")
    if err != nil {
        panic(err)
    }
    if tcpListener, ok := listener.(*net.TCPListener); ok {
        tcpListener.SetKeepAlive(true)
        // 每10秒发送一次KeepAlive探测包
        tcpListener.SetKeepAlivePeriod(10 * time.Second)
    }
    
    srv := &http.Server{
        Addr:         ":3001",
        Handler:      mux,
        // 闲置连接超时30秒
        IdleTimeout:  30 * time.Second,
        // 读操作超时30秒
        ReadTimeout:  30 * time.Second,
        // 写操作超时30秒
        WriteTimeout: 30 * time.Second,
    }
    err = srv.Serve(listener)
    if err != nil && err != http.ErrServerClosed {
        panic(err)
    }
}

3. 结合Flush后的连接状态检查（可选） #

部分场景下，Flush操作会尝试将数据推送到网络，能更早发现连接异常。可以尝试获取底层连接做非阻塞检测：

f.Flush()
// 尝试获取底层连接并检查状态
if hijacker, ok := w.(http.Hijacker); ok {
    conn, _, err := hijacker.Hijack()
    if err != nil {
        fmt.Printf("获取连接失败: %v\n", err)
        return
    }
    // 非阻塞读取检测连接是否存活
    buf := make([]byte, 1)
    conn.SetReadDeadline(time.Now())
    _, err = conn.Read(buf)
    if err != nil && !os.IsTimeout(err) {
        fmt.Printf("连接已断开: %v\n", err)
        conn.Close()
        return
    }
}

方案优先级 #

优先使用监听Context取消信号的方法，这是最直接可靠的方式；再配合TCP KeepAlive和超时参数，能覆盖绝大多数跨网络场景的连接检测需求。

内容的提问来源于stack exchange，提问作者Atom