中断 Select 以添加另一个要在 Python 中监视的套接字

Question

我正在 Windows XP 应用程序中使用 TCP 实现点对点 IPC。

我在 Python 2.6.6 中使用 select 和 socket 模块。

我有三个 TCP 线程，一个读取线程通常在 select() 上阻塞，一个写入线程通常等待一个事件（该事件表明有东西要写入 TCP），还有一个监听线程接受连接的线程。

如果我启动一个新连接或关闭当前连接，那么我需要中断读取选择并重新启动它，以便它也侦听新接受的套接字。

在 winsock 下，我可以调用 WSACancelBlockingCall 这将优雅地中断选择。

所以我的问题是：是否有可能以 pythonic 方式完成所有这一切，而无需求助于使用 poll() ？

非常感谢

--DM

Answer 1

您可以尝试向您的集合中添加一个额外的文件描述符，用作信号机制。然后，您可以向该描述符写入一个虚拟值以导致 select 退出。例如：

my_pipe = os.pipe()
...
while True:
    ready_fds = select.select(my_read_fds + [my_pipe[0]],
                              my_write_fds, my_except_fds, timeout)
    if my_pipe[0] in ready_fds[0]:
        # Add another fd to my_read_fds, etc.
        os.read(my_pipe[0], 1)
    ...

# To interrupt the current select call and add a new fd, write to the pipe:
os.write(my_pipe[1], 'x')

Answer 2

我无法添加评论，因此我将其添加为答案。

不要使用WSACancelBlockingCall。您应该使用 Adam Rosenfield 的 的方法回答。只需制作一对连接到另一个的虚拟套接字，而不是虚拟文件描述符。在您的 select() 调用中包含该套接字之一。当您需要中断呼叫时 - 只需将一个字节写入第二个虚拟套接字即可。

哦，别忘了当 select() 返回时从第一个套接字读回该字节。