如何判断 TCP 连接是由服务端还是客户端主动关闭?

在 go 中,可通过检查 `io.readfull`(或普通 `read`)返回的错误类型与值,结合本地连接状态,准确区分连接是被对端(客户端/服务端)关闭,还是本端主动关闭、超时中断等。核心依据是 `io.eof` 表示对端优雅关闭,而 `net.operror`(含 "use of closed network connection")表明本端已关闭连接。

在 TCP 网络编程中,区分连接关闭责任方(服务端 or 客户端)对日志记录、连接复用、异常诊断至关重要。Go 的 net.Conn 接口本身不提供“关闭来源”元信息,但可通过错误语义和上下文推断:

✅ 关键错误判别规则

错误类型 典型表现 含义 是否可明确归因
io.EOF err == io.EOF 或 errors.Is(err, io.EOF) 对端调用 Close() 或 CloseWrite(),完成 FIN 握手后读取完毕 ✅ 是(客户端或远端服务端)
*net.OpError + "use of closed network connection" errors.Is(err, net.ErrClosed) 或 strings.Contains(err.Error(), "closed network connection") 本端(当前代码所在侧)已调用 conn.Close() ✅ 是(服务端自身逻辑关闭)
net.Error 且 err.Timeout() 为 true if ne, ok := err.(net.Error); ok && ne.Timeout() 连接因读超时(如 SetReadDeadline 触发)中断 ❌ 非主动关闭,属异常中断
⚠️ 注意:io.ReadFull 在遇到 io.EOF 且未读满指定字节数时会返回 io.ErrUnexpectedEOF,而非 io.EOF。因此若需精确捕获对端关闭信号,建议改用 conn.Read() 手动判断:
func handleRecv(conn *net.TCPConn) {
    header := make([]byte, 2)
    for {
        n, err := conn.Read(header) // 替代 io.ReadFull
        if err != nil {
            if errors.Is(err, io.EOF) {
                log.Info("connection closed gracefully by remote peer (client or server)")
                break // 对端关闭,正常退出
            } else if errors.Is(err, net.ErrClosed) || 
                     strings.Contains(err.Error(), "use of closed network connection") {
                log.Warn("connection closed locally by this server")
                break
            } else if ne, ok := err.(net.Error); ok && ne.Timeout() {
                log.Warn("read timeout occurred")
                break
            } else {
                log.Error("read error:", err)
                break
            }
        }
        if n < len(header) {
            // 处理短读(如仅读到1字节),可能需继续读取或协议校验
            log.Warn("incomplete header read, got", n, "bytes")
            continue
        }
        // 正常处理 header...
    }
}

? 补充说明与最佳实践

  • CloseRead() 的特殊性:若本端调用了 conn.(*net.TCPConn).CloseRead()(半关闭读端),后续 Read 仍会返回 io.EOF —— 此时无法单靠 io.EOF 区分对端关闭还是本端半关闭。务必通过代码逻辑追踪 CloseRead() 调用点
  • 避免依赖错误字符串:优先使用 errors.Is(err, io.EOF) 和 errors.Is(err, net.ErrClosed) 进行判断,而非字符串匹配,以保证兼容性和健壮性。
  • 连接生命周期管理:在服务端,建议将连接关闭操作集中封装(如 closeConnection(conn, reason string)),并记录关闭方与原因,便于审计。

综上,Go 中判断关闭方的核心逻辑是:io.EOF ≈ 对端关闭;net.ErrClosed ≈ 本端关闭;其余错误需按类型细分处理。结合清晰的连接管理策略,即可实现精准的关闭溯源。