如何使用多线程并行执行函数,并限制线程数?
使用多线程处理列表中字典参数执行函数
给定一个包含字典参数的列表,我们需要使用多线程来执行一个带有这些参数的函数。目标是根据指定的线程数(例如 2 个),并行地执行同一函数的多个实例。
我们可以使用 threadpoolexecutor 线程池来实现这个目标。它允许我们创建具有指定线程数限制的线程池,这些线程可以同时执行任务。
以下是实现代码:
from concurrent.futures import ThreadPoolExecutor import threading def dosome(ip, password, user_name): tname = threading.current_thread().getName() time.sleep(1) print(f'{tname} {ip}') tpe = ThreadPoolExecutor(max_workers=2) my_list = [ {'ip': '192.168.1.2', 'password': '123456', 'user_name': '654321'}, {'ip': '192.168.1.3', 'password': '123456', 'user_name': '654321'}, {'ip': '192.168.1.4', 'password': '123456', 'user_name': '654321'}, {'ip': '192.168.1.5', 'password': '123456', 'user_name': '654321'}, {'ip': '192.168.1.6', 'password': '123456', 'user_name': '654321'} ] for m in my_list: tpe.submit(dosome, **m)
在上面的代码中,我们使用 threadpoolexecutor 创建了一个具有 2 个工作线程的线程池。然后,我们遍历 my_list 中的每个字典参数,并将它们作为关键字参数传递给 dosome 函数。
submit 方法将 dosome 函数作为第一个参数,后跟要作为关键字参数传递给它的字典参数。这将触发线程池使用一个可用的线程来执行 dosome 函数。当一个线程完成时,另一个线程将从线程池中选择一个新的任务来执行。
这种方法将确保 dosome 函数的多个实例同时运行,并且线程数限制在指定的 2 个内。