python threading库设置最大线程

问题描述

在使用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个线程可以同时运行。
  • 当一个线程结束并释放信号量时，其他等待的线程才能开始运行。

这里使用with semaphore语句，确保在代码块内的操作完成后，信号量会被正确释放。这样，其他等待的线程就能获取到信号量并开始执行。

结果