python ThreadPoolExecutor用法详解

ThreadPoolExecutor 是 Python 的 concurrent.futures 模块下的一个线程池，用于执行一批具有相似特征的线程，从而实现多线程执行任务的效率优化。以下是 ThreadPoolExecutor 的使用方法详解。

创建 ThreadPoolExecutor 实例

首先需要导入 concurrent.futures 模块，然后使用 ThreadPoolExecutor 类来创建一个线程池。ThreadPoolExecutor 的构造函数有两个参数，一个是线程池大小 max_workers，另一个是线程的上下文，默认值为 None，表示使用默认上下文。

from concurrent.futures import ThreadPoolExecutor

executor = ThreadPoolExecutor(max_workers=4)

使用 submit 提交任务

使用 ThreadPoolExecutor 实例的 submit 方法可以向线程池提交任务。submit 方法接收一个 Python callable 对象作为参数，该对象将在线程池中执行。

def print_hello():
    print('Hello')

# 提交一个执行简单打印信息的任务
executor.submit(print_hello)

使用 map 同时提交多个任务

使用 ThreadPoolExecutor 实例的 map 方法可以同时提交多个任务。map 方法接收一个 Python callable 对象和一个可迭代对象作为参数，将可迭代对象的每个元素作为输入，将结果以生成器的形式返回。

def print_hello(name):
    print('Hello', name)

# 提交多个执行简单打印信息的任务
names = ['Alice', 'Bob', 'Charlie', 'Dave']
executor.map(print_hello, names)

使用 as_completed 实时获取任务结果

使用 concurrent.futures.as_completed 方法可以实时获取任务的执行结果。as_completed 方法接收一个迭代器作为参数，返回一个生成器对象，可以通过迭代获取任务的结果。

from concurrent.futures import as_completed

def get_hello(name):
    return 'Hello ' + name

# 提交多个执行简单打印信息的任务
names = ['Alice', 'Bob', 'Charlie', 'Dave']
tasks = [executor.submit(get_hello, name) for name in names]

# 获取任务结果
for task in as_completed(tasks):
    print(task.result())

使用 shutdown 方法关闭线程池

使用 ThreadPoolExecutor 实例的 shutdown 方法可以关闭线程池。shutdown 方法将等待所有任务完成后关闭线程池。

executor.shutdown()

以上就是 ThreadPoolExecutor 的使用方法详解。通过使用 ThreadPoolExecutor 可以很方便地实现 Python 多线程编程，提高程序的执行效率。