python的asyncio

时间:2015-07-12 17:42:07

标签: python python-3.x asynchronous queue python-asyncio

根据documentation,有几种队列实现。我感兴趣的是JoinableQueue和PriorityQueue,因为我想拥有一个具有优先级的可连接队列。

似乎我只能获得较低版本中的一项功能。即在3.5中,我可以使用Queue(可连接)和PriorityQueue进行区分,但在3.5以下的python中有JoinableQueue和PriorityQueue(参见下面的示例)。

是否可以将它们组合起来以获得在3.4中获得可连接的PriorityQueue的一般方法?

try:
    # Python 3.4.
    from asyncio import JoinableQueue as Queue # joinable
    from asyncio import PriorityQueue          # I assume this one is not joinable
except ImportError:
    # Python 3.5.
    from asyncio import Queue                  # standard joinable
    from asyncio import PriorityQueue          # I assume this is the one I want

另一种方法是以某种方式影响队列?

1 个答案:

答案 0 :(得分:5)

由于实现了JoinableQueuePriorityQueue的方式,只要先列出JoinablePriorityQueue,就可以通过多重继承对两者进行子类化来获得JoinableQueue。 / p>

这样做的原因是PriorityQueue非常简单:

class PriorityQueue(Queue):
    """A subclass of Queue; retrieves entries in priority order (lowest first).

    Entries are typically tuples of the form: (priority number, data).
    """

    def _init(self, maxsize):
        self._queue = []

    def _put(self, item, heappush=heapq.heappush):
        heappush(self._queue, item)

    def _get(self, heappop=heapq.heappop):
        return heappop(self._queue)

虽然JoinableQueue更加复杂,但它和PriorityQueue实现的唯一方法是_put,而且至关重要的是JoinableQUeue调用super()._put(..) {1}}实施,这意味着它将与put正确合作。

这是一个证明它有效的例子:

PriorityQueue

输出:

from asyncio import PriorityQueue, JoinableQueue
import asyncio
import random

class JoinablePriorityQueue(JoinableQueue, PriorityQueue):
    pass


@asyncio.coroutine
def consume(q):
    while True:
        a = yield from q.get()
        print("got a {}".format(a))
        if a[1] is None:
            q.task_done()
            return
        asyncio.sleep(1)
        q.task_done()

@asyncio.coroutine
def produce(q):
    for i in range(10):
        yield from q.put((random.randint(0,10), i))
    yield from q.put((100, None)) # Will be last
    asyncio.async(consume(q))
    print("waiting...")
    yield from q.join()
    print("waited")

loop = asyncio.get_event_loop()
q = JoinablePriorityQueue()
loop.run_until_complete(produce(q))