开发者

Clear all items from the queue

开发者 https://www.devze.com 2023-03-16 16:10 出处:网络
How can I clear a queue. For example I have datas in a queue, but for some reason I don\'t need the existing data, andjust want to clear the qu开发者_如何学Ceue.

How can I clear a queue. For example I have datas in a queue, but for some reason I don't need the existing data, and just want to clear the qu开发者_如何学Ceue.

Is there any way? Will this work:

oldQueue = Queue.Queue()


q = Queue.Queue()
q.queue.clear()

EDIT I omitted the issue of thread safety for clarity and brevity, but @Dan D is quite correct, the following is better.

q = Queue.Queue()
with q.mutex:
    q.queue.clear()


You just can not clear the queue, because every put also add the unfinished_tasks member. The join method depends on this value. And all_tasks_done needs to be notified also.

with q.mutex:
    q.queue.clear()
    q.all_tasks_done.notify_all()
    q.unfinished_tasks = 0

or in decent way, use get and task_done pair to safely clear the tasks.

    while not q.empty():
        try:
            q.get(block=False)
        except Empty:
            continue
        q.task_done()

or just create a new Queue and delete old one.


This seems to do it pretty well for me. I welcome comments/additions in case I missed anything important.

class Queue(queue.Queue):
  '''
  A custom queue subclass that provides a :meth:`clear` method.
  '''

  def clear(self):
    '''
    Clears all items from the queue.
    '''

    with self.mutex:
      unfinished = self.unfinished_tasks - len(self.queue)
      if unfinished <= 0:
        if unfinished < 0:
          raise ValueError('task_done() called too many times')
        self.all_tasks_done.notify_all()
      self.unfinished_tasks = unfinished
      self.queue.clear()
      self.not_full.notify_all()
0

精彩评论

暂无评论...
验证码 换一张
取 消

关注公众号