将超时参数添加到python的Queue.join() [英] Add timeout argument to python's Queue.join()

问题描述

我希望能够加入Queue类，但是如果尚未返回调用，则在一段时间后超时. 最好的方法是什么?是否可以通过使用metaclass继承queue \来做到这一点?

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 queue\using 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()的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！