如何检测 socket 连接是否已关闭（io.Copy 场景下的优雅断连处理）-Golang-PHP中文网

如何检测 socket 连接是否已关闭（io.Copy 场景下的优雅断连处理）

本文介绍在使用 io.Copy 将标准流与网络连接双向桥接时，如何准确检测 socket 断开、避免 goroutine 泄漏，并实现连接终止后的程序优雅退出。核心在于捕获 io.Copy 返回的错误，尤其是 io.EOF 及其他底层网络错误。

本文介绍在使用 `io.copy` 将标准流与网络连接双向桥接时，如何准确检测 socket 断开、避免 goroutine 泄漏，并实现连接终止后的程序优雅退出。核心在于捕获 `io.copy` 返回的错误，尤其是 `io.eof` 及其他底层网络错误。

在基于 io.Copy 构建的 stdin/stdout/stderr 与 socket 的双向代理程序中，常见误区是认为 io.Copy 会“静默阻塞直到对端关闭”，而忽略了它其实明确返回错误——这正是判断连接状态的关键信号。

io.Copy(dst, src) 在读取源（如 conn）时，一旦连接被对端关闭或发生网络异常，会立即返回 (0, error)。典型情况包括：

对端调用 close() 或进程退出 → error == io.EOF
网络中断、超时、RST 包 → error 为 *net.OpError（如 read: connection reset by peer、use of closed network connection 等）

因此，绝不能忽略 io.Copy 的返回值。以下是改进后的健壮实现：

func runProxy(conn net.Conn, stdin io.ReadCloser, stdout, stderr io.Writer) {
    // 使用 WaitGroup 协调两个方向的复制
    var wg sync.WaitGroup

    // 从 conn → stdout/stderr（服务端输出）
    wg.Add(1)
    go func() {
        defer wg.Done()
        defer conn.Close() // 主动关闭连接，避免资源泄漏
        defer stdout.Close()
        defer stderr.Close()

        // 先复制 stdout，再复制 stderr；也可并发，但需注意顺序语义
        if _, err := io.Copy(stdout, conn); err != nil {
            if errors.Is(err, io.EOF) || errors.Is(err, net.ErrClosed) {
                log.Println("Connection closed gracefully (EOF)")
            } else {
                log.Printf("Error copying to stdout: %v", err)
            }
            return
        }
        if _, err := io.Copy(stderr, conn); err != nil {
            if !errors.Is(err, io.EOF) && !errors.Is(err, net.ErrClosed) {
                log.Printf("Error copying to stderr: %v", err)
            }
        }
    }()

    // 从 stdin → conn（客户端输入）
    wg.Add(1)
    go func() {
        defer wg.Done()
        defer stdin.Close()

        if _, err := io.Copy(conn, stdin); err != nil {
            if errors.Is(err, io.EOF) {
                log.Println("Stdin closed, terminating input stream")
            } else if errors.Is(err, net.ErrClosed) || strings.Contains(err.Error(), "broken pipe") {
                log.Printf("Connection lost during write: %v", err)
            } else {
                log.Printf("Unexpected error copying from stdin: %v", err)
            }
            return
        }
    }()

    // 等待任一方向结束（可选：也可加超时或信号控制）
    wg.Wait()
    log.Println("Proxy session ended.")
}

⚠️ 关键注意事项：

不要重复调用 conn.Close()：两个 goroutine 中都 defer conn.Close() 会导致 panic（use of closed network connection）。应由最先检测到断连的一方关闭连接，另一方通过错误感知后退出。
io.Copy 不会自动重试或恢复：它是一次性操作，返回即代表流结束。如需重连，需在外层封装重试逻辑。
区分 io.EOF 与真实错误：io.EOF 表示对端正常关闭，属预期行为；而 net.OpError、syscall.ECONNRESET 等则表明异常断连，可能需记录告警。
避免 select{} 死锁：原代码中 select{} 无 case，导致永久阻塞且无法响应任何事件。改用 sync.WaitGroup 或 context.WithCancel 控制生命周期更可控、可测试。