将超时参数添加到 python 的 Queue.join()

2022-01-21 00:00:00 python queue timeout

问题描述

我希望能够加入()队列类，但如果调用尚未返回，则在一段时间后超时.最好的方法是什么?是否可以通过子类化队列使用元类来做到这一点?

I want to be able to join() the Queue class but timeouting after some time if the call hasn't returned yet. What is the best way to do it? Is it possible to do it by subclassing queueusing metaclass?

解决方案

子类化 Queue 可能是最好的方法.像这样的东西应该可以工作(未经测试):

Subclassing Queue is probably the best way. Something like this should work (untested):

def join_with_timeout(self, timeout):
    self.all_tasks_done.acquire()
    try:
        endtime = time() + timeout
        while self.unfinished_tasks:
            remaining = endtime - time()
            if remaining <= 0.0:
                raise NotFinished
            self.all_tasks_done.wait(remaining)
    finally:
        self.all_tasks_done.release()

相关文章