在 Go 语言中使用 net 包开发 TCP 服务器时，一个常见的需求是检测客户端连接是否已经关闭。仅仅依赖尝试读取或写入数据并检查 err 是否为 nil 并不总是可靠的。下面介绍一种更有效的方法来检测 TCP 连接是否已关闭。
使用 SetReadDeadline 和 Read 检测连接状态
以下代码片段展示了如何使用 net.Conn 的 SetReadDeadline 和 Read 方法来检测连接是否已关闭。
package main
import (
"fmt"
"io"
"log"
"net"
"time"
)
func handleConnection(c net.Conn, id string, logger *log.Logger) {
defer c.Close()
one := make([]byte, 1)
c.SetReadDeadline(time.Now()) // 设置立即超时
if _, err := c.Read(one); err == io.EOF {
logger.Printf("DEBUG: %s detected closed LAN connection", id)
return
} else if err != nil {
// 检测是否是超时错误
if neterr, ok := err.(net.Error); ok && neterr.Timeout() {
// 连接正常，重置读取超时时间
c.SetReadDeadline(time.Now().Add(10 * time.Millisecond))
// 这里可以继续处理连接
fmt.Println("Timeout occurred, connection is likely still open")
return
} else {
logger.Printf("ERROR: Error reading from connection: %v", err)
return
}
} else {
// 读取到数据，重置读取超时时间
var zero time.Time
c.SetReadDeadline(zero) // 或者设置为一个合理的超时时间
fmt.Printf("Received data: %v\n", one)
// 这里可以继续处理接收到的数据
}
}
func main() {
listener, err := net.Listen("tcp", ":8080")
if err != nil {
log.Fatal(err)
}
defer listener.Close()
logger := log.Default()
id := "client1"
for {
conn, err := listener.Accept()
if err != nil {
log.Println(err)
continue
}
go handleConnection(conn, id, logger)
}
}代码解释：
- one := make([]byte, 1): 创建一个长度为 1 的字节切片，用于读取数据。
- c.SetReadDeadline(time.Now()): 设置读取超时时间为当前时间，这意味着 Read 方法会立即超时。
- c.Read(one): 尝试从连接中读取至少一个字节。
- err == io.EOF: 如果 Read 返回 io.EOF 错误，则表示连接已关闭。
- neterr, ok := err.(net.Error); ok && neterr.Timeout(): 如果 Read 返回一个 net.Error 类型的错误，并且 Timeout() 方法返回 true，则表示发生了超时。这通常意味着连接仍然存在，只是暂时没有数据可读。
- *`c.SetReadDeadline(time.Now().Add(10 time.Millisecond))`**: 在发生超时后，重新设置一个较短的读取超时时间，以便在连接空闲时快速检测到连接断开。
- c.SetReadDeadline(zero): 如果成功读取到数据，则清除之前的超时设置，或者设置为一个合理的超时时间，以便正常读取后续数据。
注意事项：
- 至少读取一个字节: Go 1.7 及以上版本中，零字节读取会立即返回，不会返回错误。因此，必须至少读取一个字节才能触发 io.EOF 错误。
- 错误处理: 除了 io.EOF 和超时错误，还需要处理其他可能的错误，例如网络错误或协议错误。
总结
通过结合使用 SetReadDeadline 和 Read 方法，并正确处理返回的错误类型，可以可靠地检测 TCP 连接是否已关闭。这种方法可以有效地避免因连接断开而导致的程序异常，并提高程序的健壮性。请记住，至少要读取一个字节的数据，并根据实际情况设置合适的读取超时时间。
