Python 的 futures
包允许我们使用 ThreadPoolExecutor
和 ProcessPoolExecutor
并行执行任务。
但是,为了调试,有时用虚拟并行临时替换真正的并行很有用,虚拟并行在主线程中以串行方式执行任务,而不会产生任何线程或进程。
有没有DummyExecutor
的实现?
最佳答案
应该这样做:
from concurrent.futures import Future, Executor
from threading import Lock
class DummyExecutor(Executor):
def __init__(self):
self._shutdown = False
self._shutdownLock = Lock()
def submit(self, fn, *args, **kwargs):
with self._shutdownLock:
if self._shutdown:
raise RuntimeError('cannot schedule new futures after shutdown')
f = Future()
try:
result = fn(*args, **kwargs)
except BaseException as e:
f.set_exception(e)
else:
f.set_result(result)
return f
def shutdown(self, wait=True):
with self._shutdownLock:
self._shutdown = True
if __name__ == '__main__':
def fnc(err):
if err:
raise Exception("test")
else:
return "ok"
ex = DummyExecutor()
print(ex.submit(fnc, True))
print(ex.submit(fnc, False))
ex.shutdown()
ex.submit(fnc, True) # raises exception
在这种情况下可能不需要锁定,但拥有它也无妨。
关于python - `DummyExecutor` 用于 Python 's ` future `,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/10434593/