从golang中的tcp连接读取

时间:2016-01-20 05:54:58

标签: go architecture network-programming server distributed

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

有关如何进行的任何建议吗?

1 个答案:

答案 0 :(得分:6)

希望这可以帮到你。 ^ _ ^

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
        }
    }
}