我想在一个在子线程中调用的函数上添加超时。
我不能使用信号,因为信号应该在主线程上。
我不能使用thread.join(time_out)
,因为该函数有时可以在几秒钟内执行,在这种情况下,线程将始终等待time_out
。
还有其他方法吗?
来源:
答案 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
我使用一个类对象并将调用线程保存到,这样我就可以从定时子线程内在父类中引发异常。