Python,子线程上的函数超时,不使用signal和thread.join

时间:2013-02-01 12:34:53

标签: python python-multithreading

我想在一个在子线程中调用的函数上添加超时。 我不能使用信号,因为信号应该在主线程上。 我不能使用thread.join(time_out),因为该函数有时可以在几秒钟内执行,在这种情况下,线程将始终等待time_out

还有其他方法吗?

来源:

1 个答案:

答案 0 :(得分:0)

您可以使用鲜为人知的 ctyps hack 来引发针对特定线程的 TimeoutError。我使用这种方法制作了一个 non_blocking 超时脚本,并刚刚在 GitHub 上发布:https://github.com/levimluke/PyTimeoutAfter

实施起来非常简单,但技术上很复杂。

    def raise_caller(self):
        ret = ctypes.pythonapi.PyThreadState_SetAsyncExc(ctypes.c_long(self._caller_thread._ident), ctypes.py_object(self._exception))
        if ret == 0:
            raise ValueError("Invalid thread ID")
        elif ret > 1:
            ctypes.pythonapi.PyThreadState_SetAsyncExc(self._caller_thread._ident, NULL)
            raise SystemError("PyThreadState_SetAsyncExc failed")

https://github.com/levimluke/PyTimeoutAfter/blob/main/timeout_after.py#L47

我使用一个类对象并将调用线程保存到,这样我就可以从定时子线程内在父类中引发异常。