如何等待任何套接字有数据？

Question

我正在实现一个同时打开多个套接字的socket客户机。任何套接字可能在不同的时间有数据，我想在任何套接字有数据且可读时执行代码。在

我不确定如何实现这一点，我正在研究select.select，但它似乎等待所有套接字都可读。在

我希望避免使用多处理来处理套接字上的数据，我希望它在从每个套接字读取数据时是串行的，但在有可用数据时读取。在

如何等待任何套接字可读？在

# psudo code

sockets = [sock1, sock2, sock3]

while True:
  if len(sockets) == 0:
    break
  for sock in sockets:
    if sock.has_data():
      do_stuff(sock)
      sockets.remove(sock)
  sleep(0.1)

Answer 1

您可以使用select.select来解决您的问题：

sockets = [sock1, sock2, sock3]
while sockets:
    rlist, _, _ = select.select(sockets, [], [])
    for sock in rlist:
        do_stuff(sock)
        sockets.remove(sock)

Answer 2

如果您使用POSIX，请查看select.poll：

import socket
import select
p = select.poll()
s1 = socket.socket()
s2 = socket.socket()
# call connect on sockets here...
p.register(s1, select.POLLIN)
p.register(s2, select.POLLIN)
p.poll()

Answer 3

如果您使用的是python3.4或更新版本，则标准库中有selectors模块。它将使用您的系统提供的“最佳”I/O多路复用实现（select、poll、kqueue…）在文档页的末尾有一个简单的echo服务器示例https://docs.python.org/3/library/selectors.html

对于旧的Python版本也有一个backport。在