Python并发——concurrent.futures梳理
参考官方文档: concurrent.futures — 启动并行任务
Executor对象
class concurrent.funtures.Executor
该抽象类是 ThreadPoolExecutor
和 ProcessPoolExecutor
的父类,提供异步执行调用方法。要通过它的子类调用,而不是直接调用。
submit方法
submit(fn, /, *args, **kwargs)
提交一个可执行对象,fn 将按照如下方式被调用:fn(*args. **kwargs)
,并返回一个 Future
对象(下面会详细说)表示执行的结果,
例:
with ThreadPoolExecutor(max_workers=1) as executor:future = executor.submit(pow, 323, 1235)print(future.result())
map方法
map(func, *iterables, timeout=None, chunksize=1)
-
类似于
map(func, *iterables)
函数,但是以下两点不同:- iterables 是立即执行而不是延迟执行的;
- func 是异步执行的,对 func 的多个调用可以并发执行。
-
如果从原始调用到
Executor.map()
经过 timeout 秒后,__next__()
已被调用且返回的结果还不可用,那么已返回的迭代器将触发concurrent.futures.TimeoutError
。 timeout 可以是整数或浮点数。如果 timeout 没有指定或为 None ,则没有超时限制。 -
使用
ProcessPoolExecutor
时,这个方法会将 iterables 分割任务块并作为独立的任务并提交到执行池中。这些块的大概数量可以由 chunksize 指定正整数设置。 对很长的迭代器来说,使用大的 chunksize 值比默认值 1 能显著地提高性能。 chunksize 对ThreadPoolExecutor
没有效果。 -
如果 func 调用引发一个异常,当从迭代器中取回它的值时这个异常将被引发。
-
iterables 是一个可迭代对象,其中每个元素是传递给 func 的参数列表,与
submit
方法直接传参不同的是,如果需要传递给 func 函数多个列表,通常需要再加一个中间层来进行列表参数的解析。python 线程池map()方法传递多参数list
shutdown方法
shutdown(wait=True, *, cancel_futures=False)
当待执行的 future
对象完成执行后向执行者发送信号,它就会释放正在使用的任何资源。 在关闭后调用 Executor.submit()
和 Executor.map()
将会引发 RuntimeError
。
-
如果 wait 为 True 则此方法只有在所有待执行的
future
对象完成执行且释放已分配的资源后才会返回; -
如果 wait 为 False,方法立即返回,所有待执行的
future
对象完成执行后会释放已分配的资源。 不管 wait 的值是什么,整个 Python 程序将等到所有待执行的future
对象完成执行后才退出; -
如果 cancel_futures 为 True,此方法将取消所有执行器还未开始运行的挂起的
future
。 任何已完成或正在运行的future
将不会被取消,无论 cancel_futures 的值是什么; -
如果 cancel_futures 和 wait 均为
True
,则执行器已开始运行的所有 Future 将在此方法返回之前完成。 其余的 Future 会被取消。
如果使用 with 语句,你就可以避免显式调用这个方法,它将会停止 Executor
(就相当于 Executor.shutdown()
调用时 wait 设为 True 一样等待):
import shutil
with ThreadPoolExecutor(max_workers=4) as e:e.submit(shutil.copy, 'src1.txt', 'dest1.txt')e.submit(shutil.copy, 'src2.txt', 'dest2.txt')e.submit(shutil.copy, 'src3.txt', 'dest3.txt')e.submit(shutil.copy, 'src4.txt', 'dest4.txt')
ThreadPoolExecutor
ThreadPoolExecutor
是 Executor
的子类,它使用线程池来异步执行调用。
class concurrent.futures.ThreadPoolExecutor(max_workers=None, thread_name_prefix='', initializer=None, initargs=())
-
使用最多 max_workers 个线程的线程池来异步执行调用。
-
(Python3.7之后)initializer 是在每个 worker 线程开始处调用的一个可选可调用对象。 initargs 是传递给 initializer 的元组参数。任何向池提交更多工作的尝试, initializer 都将引发一个异常,当前所有等待的工作都会引发一个
BrokenThreadPool
。 -
(Python3.5之后)如果 max_workers 为 None 或没有指定,将默认为机器处理器的个数,假如
ThreadPoolExecutor
则重于I/O操作而不是CPU运算,那么可以乘以 5 ,同时工作线程的数量可以比ProcessPoolExecutor
的数量高。 -
(Python3.6之后)thread_name_prefix 参数允许用户控制由线程池创建的
threading.Thread
工作线程名称以方便调试。 -
(Python3.8之后)max_workers 的默认值已改为
min(32, os.cpu_count() + 4)
。 这个默认值会保留至少 5 个工作线程用于 I/O 密集型任务。 对于那些释放了 GIL 的 CPU 密集型任务,它最多会使用 32 个 CPU 核心。这样能够避免在多核机器上不知不觉地使用大量资源。现在 ThreadPoolExecutor 在启动 max_workers 个工作线程之前也会重用空闲的工作线程。 -
当回调已关联了一个
Future
然后再等待另一个Future
的结果时就会发产死锁情况。例如:# 例一 import time def wait_on_b():time.sleep(5)print(b.result()) # b will never complete because it is waiting on a.return 5def wait_on_a():time.sleep(5)print(a.result()) # a will never complete because it is waiting on b.return 6executor = ThreadPoolExecutor(max_workers=2) a = executor.submit(wait_on_b) b = executor.submit(wait_on_a)# 例二 def wait_on_future():f = executor.submit(pow, 5, 2)# This will never complete because there is only one worker thread and# it is executing this function.print(f.result())executor = ThreadPoolExecutor(max_workers=1) executor.submit(wait_on_future)
ThreadPoolExecutor
实例:
import concurrent.futures
import urllib.requestURLS = ['http://www.foxnews.com/','http://www.cnn.com/','http://europe.wsj.com/','http://www.bbc.co.uk/','http://some-made-up-domain.com/']# Retrieve a single page and report the URL and contents
def load_url(url, timeout):with urllib.request.urlopen(url, timeout=timeout) as conn:return conn.read()# We can use a with statement to ensure threads are cleaned up promptly
with concurrent.futures.ThreadPoolExecutor(max_workers=5) as executor:# Start the load operations and mark each future with its URLfuture_to_url = {executor.submit(load_url, url, 60): url for url in URLS}for future in concurrent.futures.as_completed(future_to_url):url = future_to_url[future]try:data = future.result()except Exception as exc:print('%r generated an exception: %s' % (url, exc))else:print('%r page is %d bytes' % (url, len(data)))
ProcessPoolExecutor
ProcessPoolExecutor
类是 Executor
的子类,它使用进程池来异步地执行调用。
class concurrent.futures.ProcessPoolExecutor(max_workers=None, mp_context=None, initializer=None, initargs=())
-
ProcessPoolExecutor
会使用multiprocessing
模块,这允许它绕过 GIL,但也意味着只可以处理和返回可封存的对象。 -
__main__
模块必须可以被工作者子进程导入。这意味着ProcessPoolExecutor
不可以工作在交互式解释器中。 -
从可调用对象中调用
Executor
或Future
的方法提交给ProcessPoolExecutor
会导致死锁。这与ThreadPoolExecutor
类似。 -
(Python3.7之后)异步地执行调用的
Executor
子类使用最多具有 max_workers 个进程的进程池。 如果 max_workers 为 None 或未给出,它将默认为机器的处理器个数。 如果 max_workers 小于等于 0,则将引发ValueError
。 在 Windows 上,max_workers 必须小于等于 61,否则将引发ValueError
。 如果 max_workers 为 None,则所选择的默认值最多为 61,即使存在更多的处理器。 mp_context 可以是一个多进程上下文或是 None。 它将被用来启动工作进程。 如果 mp_context 为 None 或未给出,则将使用默认的多进程上下文。 -
(Python3.7之后)initializer 是一个可选的可调用对象,它会在每个工作进程启动时被调用;initargs 是传给 initializer 的参数元组。 如果 initializer 引发了异常,则所有当前在等待的任务以及任何向进程池提交更多任务的尝试都将引发
BrokenProcessPool
。 -
(Python3.3之后)如果其中一个工作进程被突然终止,
BrokenProcessPool
就会马上触发。 可预计的行为没有定义,但执行器上的操作或它的future
对象会被冻结或死锁。
可以看到,大部分参数含义与 ThreadPoolExecutor
是一致的。
ProcessPoolExecutor
实例:
import concurrent.futures
import mathPRIMES = [112272535095293,112582705942171,112272535095293,115280095190773,115797848077099,1099726899285419]def is_prime(n):if n < 2:return Falseif n == 2:return Trueif n % 2 == 0:return Falsesqrt_n = int(math.floor(math.sqrt(n)))for i in range(3, sqrt_n + 1, 2):if n % i == 0:return Falsereturn Truedef main():with concurrent.futures.ProcessPoolExecutor() as executor:for number, prime in zip(PRIMES, executor.map(is_prime, PRIMES)):print('%d is prime: %s' % (number, prime))if __name__ == '__main__':main()
Future对象
Future
类将可调用对象封装为异步执行。Future
实例由 Executor.submit()
创建。将可调用对象封装为异步执行。Future
实例由 Executor.submit()
创建,除非测试,不应手动直接创建。
class concurrent.futures.Future
-
cancel()
尝试取消调用。 如果调用正在执行或已结束运行不能被取消则该方法将返回 False,否则调用会被取消并且该方法将返回 True。
-
cancelled()
如果调用成功取消返回 True。
-
running()
如果调用正在执行而且不能被取消那么返回 True 。
-
done()
如果调用已被取消或正常结束那么返回 True。
-
result(timeout=None)
返回调用返回的值。如果调用还没完成那么这个方法将等待 timeout 秒。如果在 timeout 秒内没有执行完成,
concurrent.futures.TimeoutError
将会被触发。timeout 可以是整数或浮点数。如果 timeout 没有指定或为 None,那么等待时间就没有限制。如果futrue
在完成前被取消则CancelledError
将被触发。如果调用引发了一个异常,这个方法也会引发同样的异常。 -
exception(timeout=None)
返回由调用引发的异常。如果调用还没完成那么这个方法将等待 timeout 秒。如果在 timeout 秒内没有执行完成,
concurrent.futures.TimeoutError
将会被触发。timeout 可以是整数或浮点数。如果 timeout 没有指定或为 None,那么等待时间就没有限制。如果futrue
在完成前被取消则CancelledError
将被触发。如果调用正常完成那么返回 None。 -
add_done_callback(fn)
附加可调用 fn 到
future
对象。当future
对象被取消或完成运行时,将会调用 fn,而这个future
对象将作为它唯一的参数。加入的可调用对象总被属于添加它们的进程中的线程按加入的顺序调用。如果可调用对象引发一个Exception
子类,它会被记录下来并被忽略掉。如果可调用对象引发一个BaseException
子类,这个行为没有定义。如果future
对象已经完成或已取消,fn 会被立即调用。
下面这些 Future
方法用于单元测试和 Executor
实现,一般不会用到,就不详细介绍了。
-
set_running_or_notify_cancel()
-
set_result(result)
-
set_exception(exception)
concurrent.futures模块的函数
-
wait
concurrent.futures.wait(fs, timeout=None, return_when=ALL_COMPLETED)
等待 fs 所指定的 Future 实例(可能有不同的 Exectutor 实例创建)完成。fs 若给出重复的 futures 将被移除,只返回一次。
-
返回值是一个 set 组成的命名二元组,其中第一个 set 叫做
done
,包含完成(包括 finished 和 cancelled)的 futures,第二个 set 叫做not_done
,包含未完成(包括 pending 和 running)的 futures。 -
timeout 可以用来控制返回前最大的等待秒数。 timeout 可以为 int 或 float 类型。 如果 timeout 未指定或为
None
,则不限制等待时间。 -
return_when 指定此函数应在何时返回。它必须为以下常数之一:
常量 描述 FIRST_COMPLETED
函数将在任意可等待对象结束或取消时返回。 FIRST_EXCEPTION
函数将在任意可等待对象因引发异常而结束时返回。当没有引发任何异常时它就相当于 ALL_COMPLETED
。ALL_COMPLETED
函数将在所有可等待对象结束或取消时返回。 -
-
as_complete
concurrent.futures.as_completed(fs, timeout=None)
返回一个包含 fs 所指定的
Future
实例(可能由不同的Executor
实例创建)的迭代器,这些实例会在完成时生成 future 对象(包括正常结束或被取消的 future 对象)。 任何由 fs 所指定的重复 future 对象将只被返回一次。 任何在as_completed()
被调用之前完成的 future 对象将优先被生成。 如果__next__()
被调用并且在对as_completed()
的原始调用 timeout 秒之后结果仍不可用,则返回的迭代器将引发concurrent.futures.TimeoutError
。 timeout 可以为整数或浮点数。 如果 timeout 未指定或为 None,则不限制等待时间。
Exception 类
-
concurrent.futures.CancelledError
future 对象被取消时会触发。
-
concurrent.futures.TimeoutError
future 对象执行超出给定的超时数值时引发。
-
concurrent.futures.BrokenExecutor
当执行器被某些原因中断而且不能用来提交或执行新任务时就会被引发派生于
RuntimeError
的异常类。3.7 新版功能. -
concurrent.futures.InvalidStateError
当某个操作在一个当前状态所不允许的 future 上执行时将被引发。3.8 新版功能.
-
concurrent.futures.thread.BrokenThreadPool
当
ThreadPoolExecutor
中的其中一个工作者初始化失败时会引发派生于BrokenExecutor
的异常类。3.7 新版功能. -
concurrent.futures.process.BrokenProcessPool
当 ThreadPoolExecutor
中的其中一个 worker 不完整终止时(比如,被外部杀死)会引发派生于 BrokenExecutor
( 原名 RuntimeError
) 的异常类。3.3 新版功能.