我想从 Queue.Queue 或 TCP 套接字中读取消息,以先到者为准。 不借助 2 个线程如何实现? 平台是 Windows 上的 CPython 2.7.5
最佳答案
有一个非常好的技巧可以做到这一点 here适用于您的问题。
import queue
import socket
import os
class PollableQueue(queue.Queue):
def __init__(self):
super().__init__()
# Create a pair of connected sockets
if os.name == 'posix':
self._putsocket, self._getsocket = socket.socketpair()
else:
# Compatibility on non-POSIX systems
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(('127.0.0.1', 0))
server.listen(1)
self._putsocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self._putsocket.connect(server.getsockname())
self._getsocket, _ = server.accept()
server.close()
def fileno(self):
return self._getsocket.fileno()
def put(self, item):
super().put(item)
self._putsocket.send(b'x')
def get(self):
self._getsocket.recv(1)
return super().get()
关于Python - 如何同时等待队列和套接字,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/17495877/