go - 从 golang 中的 tcp 连接读取

Question

在像 bytes_read, err := conn.Read(tmp) 这样的语句中，我希望尝试读取 x 秒，如果没有读取开始，我希望代码继续检查一些连接并再次循环并尝试读。我可以使用 select-case 并生成两个 goroutine，一个尝试读取，另一个尝试超时。但是在这里，如果首先发生超时，代码将继续，检查条件并再次生成例程以尝试从连接中读取，同时先前的读取例程仍然存在。我希望之前的例程在超时发生时结束。

有什么建议可以继续吗？

Answer 1

希望对您有所帮助。 ^_^

for {
    // set SetReadDeadline
    err := conn.SetReadDeadline(time.Now().Add(5 * time.Second))
    if err != nil {
        log.Println("SetReadDeadline failed:", err)
        // do something else, for example create new conn
        return
    }

    recvBuf := make([]byte, 1024)

    n, err = conn.Read(recvBuf[:]) // recv data
    if err != nil {
        if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
            log.Println("read timeout:", err)
            // time out
        } else {
            log.Println("read error:", err)
            // some error else, do something else, for example create new conn
        }
    }
}