Go语言客户端侧TCP连接优雅关闭：确保全数据读写响应

基于TCP半关闭（CloseWrite）的无应用层信号方案 #

Go的net.TCPConn提供的CloseWrite()可实现TCP半关闭，替代应用层关闭信号完成优雅关闭，核心是利用TCP协议的FIN包作为服务器停止写入的信号，流程如下：

客户端实现逻辑 #

1. 触发关闭流程：当需要发起关闭时，调用conn.CloseWrite()。
   • 此操作会向服务器发送FIN包，告知服务器：客户端将不再向连接写入数据，但仍会接收服务器已在传输或尚未发送的剩余数据。
2. 读取所有剩余数据：持续从连接读取数据，直到读操作返回io.EOF（服务器收到FIN后，会在发送完所有待发数据后关闭自身写方向，客户端此时会收到EOF）。
3. 注意事项：由于CloseWrite()已关闭客户端写方向，若需要在读取完成后向服务器发送响应，必须将响应发送放在CloseWrite()之前。如果业务逻辑要求读完数据再发响应，这种方式不适用，需结合应用层信号。

服务器实现逻辑 #

1. 监听读操作状态：在循环读取客户端数据时，若收到io.EOF（对应客户端的CloseWrite()），立即停止向客户端写入新数据。
2. 发送剩余待发数据：停止写入新数据后，将已准备好的待发数据全部发送给客户端，随后调用conn.CloseWrite()，向客户端发送FIN包。
3. 等待客户端响应（若需要）：如果服务器需要接收客户端的最终响应，需在调用CloseWrite()之前完成读取，因为CloseWrite()不影响读方向。
4. 最终关闭连接：待所有数据收发完成后，调用conn.Close()关闭整个连接。

满足“读后响应”需求的混合方案 #

如果必须实现“读取完服务器所有数据后再发送响应”，可以采用以下流程：

1. 客户端先向服务器发送一个应用层的“停止发送”信号（如"STOP_SEND"），告知服务器停止写入新数据。
2. 客户端持续读取服务器的剩余数据，直到收到服务器发送的“数据发送完毕”信号（如"SEND_COMPLETE"）。
3. 客户端发送最终响应给服务器。
4. 客户端调用conn.CloseWrite()，服务器收到FIN后读取完响应，再调用conn.Close()完成优雅关闭。

关键代码示例 #

客户端核心代码 #

func clientHandler(conn net.Conn) {
    defer conn.Close()

    // 业务逻辑：正常读写...

    // 若需要先发送响应，在此处写入响应
    // _, err := conn.Write([]byte("FINAL_RESPONSE"))
    // if err != nil { /* 错误处理 */ }

    // 关闭写方向，通知服务器停止发送新数据
    if tcpConn, ok := conn.(*net.TCPConn); ok {
        err := tcpConn.CloseWrite()
        if err != nil { /* 错误处理 */ }
    }

    // 读取服务器剩余数据
    buf := make([]byte, 1024)
    for {
        n, err := conn.Read(buf)
        if err == io.EOF {
            break
        }
        if err != nil { /* 错误处理 */ }
        // 处理读取到的数据
        processData(buf[:n])
    }
}

服务器核心代码 #

func serverHandler(conn net.Conn) {
    defer conn.Close()

    buf := make([]byte, 1024)
    for {
        n, err := conn.Read(buf)
        if err == io.EOF {
            // 收到客户端FIN，停止发送新数据
            break
        }
        if err != nil { /* 错误处理 */ }
        // 处理客户端请求，发送响应
        _, err = conn.Write([]byte("SERVER_RESPONSE"))
        if err != nil { /* 错误处理 */ }
    }

    // 发送剩余待发数据（如果有）
    sendRemainingData(conn)

    // 关闭写方向
    if tcpConn, ok := conn.(*net.TCPConn); ok {
        tcpConn.CloseWrite()
    }
}

注意事项 #

• 必须将net.Conn类型断言为*net.TCPConn才能调用CloseWrite()和CloseRead()，因为基础的net.Conn接口不包含这些方法。
• CloseRead()会直接关闭客户端的读方向，若服务器继续写数据会触发RST包，导致连接异常终止，不建议用于需要无数据丢失的场景。
• 无论哪种方案，都需要确保双方的读写循环正确处理io.EOF和其他错误，避免出现阻塞或数据丢失。

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