在 golang 中检测 tcp 超时方法:使用 setreaddeadline 和 setwritedeadline 设置读/写超时。执行读/写操作时,超时过后会返回 net.error,其中 timeout 方法返回 true,表示已超时。
如何判断 Golang 中的 TCP 超时
在 Golang 中,可以通过设置读/写超时来判断 TCP 连接是否超时。
设置超时
使用 net.Conn 接口的 SetReadDeadline 和 SetWriteDeadline 方法设置超时:
conn.SetReadDeadline(time.Now().Add(timeout))
conn.SetWriteDeadline(time.Now().Add(timeout))
其中 timeout 是超时时间,单位为纳秒。
检查超时
执行读/写操作时,如果超时已过则会返回 net.Error 错误,其中 Timeout 方法返回 true:
_, err := conn.Read(buf)
if err != nil && err.(net.Error).Timeout() {
// 超时
}
代码示例
以下代码示例展示了如何在 TCP 连接上设置并检查超时:
package main
import (
"log"
"net"
"time"
)
func main() {
// 创建 TCP 连接
conn, err := net.Dial("tcp", ":80")
if err != nil {
log.Fatal(err)
}
defer conn.Close()
// 设置超时
conn.SetReadDeadline(time.Now().Add(5 * time.Second))
conn.SetWriteDeadline(time.Now().Add(5 * time.Second))
// 发送数据
_, err = conn.Write([]byte("Hello, world!"))
if err != nil {
if err.(net.Error).Timeout() {
log.Println("发送数据超时")
} else {
log.Fatal(err)
}
}
// 接收数据
buf := make([]byte, 1024)
_, err = conn.Read(buf)
if err != nil {
if err.(net.Error).Timeout() {
log.Println("接收数据超时")
} else {
log.Fatal(err)
}
}
}