在线程中拥有Queue的双端队列

时间:2019-06-30 00:33:35

标签: python python-3.x multithreading queue deque

我需要一个结构,其中我可以将pop()和append()移到右侧(就像双端队列一样),同时让该结构阻塞并等待它是否为空(就像Queue)。我可以直接使用Queue,但是我还需要deque的出色功能,即如果结构已满,则可以删除项目而不会阻塞。

from collections import deque
d = deque(maxlen=2)
d.append(1)
d.append(2)
d.append(3) # d should be [2,3] (it is the case)
d.pop()
d.pop()
d.pop() # should wait (not the case)

将deque(使其等待)或Queue(添加popLeft函数)的子类更好吗?

2 个答案:

答案 0 :(得分:2)

如何创建自己的队列,同时兼顾两者?

import queue as Queue
from collections import deque

class QueuePro:
  def __init__(self, maxlenDeque): 
    self.deque = deque(maxlen=maxlenDeque)
    self.queue = Queue.Queue()

  def append(self, elem):
    self.deque.append(elem)
    self.queue.put(elem)

  def pop(self):
    if(not self.deque):
      self.queue.get()
    else:
      self.deque.pop()
      self.queue.get()


q2 = QueuePro(2)
q2.append(1)
q2.append(2)

q2.pop()
q2.pop()
q2.pop()
#waiting

答案 1 :(得分:1)

不确定哪种方法更好,但这是通过threading.Event

添加 pop on wait 的想法
from collections import deque
from threading import Event

class MyDeque(deque):
    def __init__(self, max_length):
        super().__init__(maxlen=max_length)
        self.not_empty = Event()
        self.not_empty.set()

    def append(self, elem):
        super().append(elem)
        self.not_empty.set()

    def pop(self):
        self.not_empty.wait()  # Wait until not empty, or next append call
        if not (len(q) - 1):
            self.not_empty.clear()
        return super().pop()

q = MyDeque(2)
q.append(1)
q.append(2)
q.append(3)
q.pop()
q.pop()
q.pop()  # Waits