
在 Go 语言中使用 net 包开发 TCP 服务器时,一个常见的需求是检测客户端连接是否已经关闭。仅仅依赖尝试读取或写入数据并检查 err 是否为 nil 并不总是可靠的。下面介绍一种更有效的方法来检测 TCP 连接是否已关闭。
以下代码片段展示了如何使用 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)
}
}代码解释:
注意事项:
通过结合使用 SetReadDeadline 和 Read 方法,并正确处理返回的错误类型,可以可靠地检测 TCP 连接是否已关闭。这种方法可以有效地避免因连接断开而导致的程序异常,并提高程序的健壮性。请记住,至少要读取一个字节的数据,并根据实际情况设置合适的读取超时时间。
以上就是如何在 Go 的 net 包中检测 TCP 连接是否已关闭的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号