TCP accept and Go 并发模型

TCP accept and Go concurrency model

正在查看 net.TCPListener。考虑到 Go 的并发范式,人们会期望这个系统功能被实现为一个通道,这样你就可以从一个 Listen() 函数中得到一个 chan *net.Conn,或者类似的东西。

但似乎 Accept() 是方法,它只是阻塞,就像系统接受一样。除了残废,因为:

所以我正在做类似的事情:

    acceptChannel = make(chan *Connection)
    go func() {
      for {
       rw, err := listener.Accept()
       if err != nil { ... handle error ... close(acceptChannel) ... return }
       s.acceptChannel <-&Connection{tcpConn: rw, .... }
      }
    }()

这样我就可以在 select 中使用多个服务器套接字,或者将 Accept() 上的等待与其他通道多路复用。我错过了什么吗?我是 Go 的新手,所以我可能会忽略一些事情——但是 Go 真的没有用自己的并发范例实现自己的阻塞系统功能吗?我真的需要为我想听的每个套接字(可能成百上千个)单独的 goroutine 吗?这是应该使用的正确习语,还是有更好的方法?

你的代码很好。您甚至可以更进一步替换:

s.acceptChannel <-&Connection{tcpConn: rw, .... }

与:

go handleConnection(&Connection{tcpConn: rw, .... })

如评论中所述,例程不是系统线程,它们是由 Go runtime 管理的轻量级线程。当您为每个连接创建一个例程时,您可以轻松地使用更容易实现的阻塞操作。 Go 运行时然后 selecting 为您编写例程,因此您正在寻找的行为只是在其他地方,埋在语言中。你看不到它,但它无处不在。

现在,如果您需要更复杂的东西,并且根据我们的谈话,实现类似于 select 的东西并带有超时,您将完全按照您的建议进行操作:将所有新连接推送到通道并用定时器复用它。这似乎是 Go 中的方法。

请注意,如果其中一个 接受器失败,则您无法关闭接受通道,因为另一个接受器在写入时会出现恐慌。

我的(更完整的)示例:

newConns := make(chan net.Conn)

// For every listener spawn the following routine
go func(l net.Listener) {
    for {
        c, err := l.Accept()
        if err != nil {
            // handle error (and then for example indicate acceptor is down)
            newConns <- nil
            return
        }
        newConns <- c
    }
}(listener)

for {
    select {
    case c := <-newConns:
        // new connection or nil if acceptor is down, in which case we should
        // do something (respawn, stop when everyone is down or just explode)
    case <-time.After(time.Minute):
        // timeout branch, no connection for a minute
    }
}