对于ThreadPoolExecutor线程池中提交的任务,在执行过程中是不会主动抛出异常的,需要主动捕获这个异常。
from concurrent.futures import ThreadPoolExecutor
def job():
print('333')
print(1/0)
print('444')
# add_done_callback()
def thread_pool_callback(worker):
print("called thread pool executor callback function")
worker_exception = worker.exception()
if worker_exception:
print("Worker return exception: {}".format(worker_exception))
if __name__ == '__main__':
with ThreadPoolExecutor(max_workers=5) as executor:
thread_pool_exc = executor.submit(job)
thread_pool_exc.add_done_callback(thread_pool_callback)
# 使用result()
if __name__ == '__main__':
with ThreadPoolExecutor(max_workers=5) as executor:
thread_pool_exc = executor.submit(job).result()
print(thread_pool_exc)
print('555')
结果:
回调:
result:
以上面代码为例,比较通用的做法是使用add_done_callback方法获取到线程执行结束后的结果,从中获取exception的信息。
使用回调方法和使用result的区别:
回调方法不会导致当前主线程异常终止,使用result相当于是拿当前主线程重新执行一遍,这时候的异常会自动抛出导致主线程异常终止。
原因:
异步执行:线程池执行器是为了异步执行任务而设计的。这意味着当你提交一个任务给线程池时,它不会立即执行,而是在后台的某个线程中执行。如果异常立即在任务线程中抛出,它会中断整个线程池的执行,这可能不是你期望的行为。
任务封装:每个任务都被封装在一个 Future 对象中,该对象可以追踪任务的状态和结果。这种封装允许你异步地监视和管理多个任务,而不必等待每个任务完成。Future 对象的目的是提供一个异步的、非阻塞的接口,以便你可以有效地提交、等待和取消任务。
异常处理:将异常封装在 Future 对象中允许你更容易地捕获和处理异常。你可以使用 future.result() 方法获取任务的结果,该方法会在任务抛出异常时将异常重新引发,这允许你在需要时捕获异常并采取适当的措施。
避免线程中断:如果异常立即在任务线程中抛出,它可能导致线程中断,这会导致线程退出并影响线程池的性能。通过将异常包装在 Future 对象中,线程池可以继续正常运行其他任务,而不受影响。