Python TCP Sockets：如何知道特定连接是否已发送信息

Question

我有一个多线程 Python 3 应用程序，它在线程 #1 上接受 TCP 套接字通信。线程 #2 将检查所有当前连接是否有任何要接收的内容，然后采取相应措施。

所以，目前我有一个名为all_connections 的列表，它是一个接受的套接字连接对象的列表。

使用for connection in all_connections: 我可以遍历所有连接对象。我知道我使用conn.recv(256) 来检查是否有任何东西可以在这个套接字上接收。这会阻塞循环，直到有东西要接收？我事先设置了conn.setblocking(1)，尽管我不确定这是否是解决它的最佳方法：

下面是一些示例代码：

线程 1

self.all_connections = [] # init a list to hold connection objs
while 1:
    try:
        conn, address = self.socket.accept()
        conn.setblocking(1) # non blocking
    except Exception as e:
        continue
    self.all_connections.append(conn) # Save the connection object

线程 2

while True:
    for connection in self.all_connections:
        received = connection.recv(256)
return

所以，我只对实际发送了一些东西的连接感兴趣，因为我很可能会向他们发送一些东西。

我知道我可以使用select.select 来检查套接字上是否有要接收的内容，但这无助于我参考具体的连接。

Answer 1

是的，read() 将阻止；这是默认行为。调用 socket.setblocking(1) 实际上会启用阻塞，这与您想要的相反。 setblocking(False) 将设置非阻塞模式。非阻塞套接字上的 I/O 要求您使用异常处理。

更好的方法是使用select()，并且您已经朝着正确的方向前进。实际上，您确实知道哪个套接字发送了数据，因为select() 返回一个可用于读取、写入或具有错误状态的套接字列表。您将您感兴趣的套接字列表传递给select()，它会返回可用于 I/O 的套接字。这是函数签名：

select(...)
    select(rlist, wlist, xlist[, timeout]) -> (rlist, wlist, xlist)

所以线程 2 中的代码如下所示：

from select import select

while True:
      rlist, wlist, xlist = select(self.all_connections, [], [])
      for connection in rlist:
          received = connection.recv(256)

上面的代码只检查所有连接列表中的可读套接字，并从准备好的那些中读取数据。读取不会阻塞。