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

问题描述

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

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()

这篇关于将超时参数添加到 python 的 Queue.join()的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持跟版网！