代码之家 › 专栏 › 技术社区 › user3740387

从golang中的tcp连接读取

distributed server network-programming architecture go

user3740387 · 技术社区 · 8 年前

在像bytes_read,err:=conn.read(tmp)这样的语句中,我希望读取尝试x秒,如果没有读取开始,我希望代码继续前进,检查一些连接,然后再次循环并尝试读取。我可以使用select case并生成两个goroutine,一个尝试读取,另一个尝试超时。但是在这里,如果超时首先发生,代码将继续运行,检查条件并再次生成一个例程,以便在前一个读取例程仍然有效的情况下尝试从连接读取。我希望在超时发生时前一个例程终止。

对我如何进行有什么建议吗?

1 回复 | 直到 8 年前

Justlike 8 年前

希望这能帮助你^_^

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

推荐文章