threading库设置最大线程

发布时间 2024-01-12 19:53:25作者: 夜久听山雨

问题描述

在使用python的threading库时,希望如何设置最大线程数N_max

  • 从而确保当python开启的线程数超过N_max时,将暂停程序;

  • 直到前面的线程结束后,再开始新的线程

解决方案

import threading
import time

# 定义工作函数

def worker(semaphore):
    with semaphore:
        print(f"线程 {threading.current_thread().name} 开始执行")
        time.sleep(2)  # 模拟耗时操作
        print(f"线程 {threading.current_thread().name} 结束执行")

# 设置最大线程数
N_max = 3
semaphore = threading.Semaphore(N_max)

# 需要运行的线程数
num_threads = 10

# 创建并启动线程
for i in range(num_threads):
    t = threading.Thread(target=worker, name=f"Thread-{i}", args=(semaphore,))
    t.start()

  • 在这个示例中,定义了一个worker函数,用于模拟线程的工作。

  • 创建了一个Semaphore对象,并设置最大线程数为3。

  • 接着,我们创建了10个线程并启动它们。

    • 在这种情况下,最多只有3个线程可以同时运行。

    • 当一个线程结束并释放信号量时,其他等待的线程才能开始运行。

结果