使用ThreadPoolExecutor时代码无法终止问题求助

为啥程序停不下来？ #

• 你的task_submit函数是无限循环逻辑，没有设置终止开关，只要它在线程池里运行，就会占用非守护线程，直接阻止进程退出。
• 主线程从队列取任务用的是阻塞式queue.get()，既没加超时也没配终止信号，哪怕线程池执行了shutdown，主线程也会一直卡在这一步等待任务。
• ThreadPoolExecutor的工作线程默认都是非守护线程，只要有一个线程还在运行，整个进程就不会终止。

怎么改才能正常终止？ #

1. 给task_submit加终止信号 #

用threading.Event做一个可触发的停止开关，让循环能收到终止指令：

from concurrent.futures import ThreadPoolExecutor
import queue
import threading

# 全局终止信号
stop_flag = threading.Event()
task_queue = queue.Queue()

def task_submit():
    # 信号未触发时持续生成任务
    while not stop_flag.is_set():
        # 模拟业务生成任务，可根据实际场景调整
        new_task = f"task_{threading.get_ident()}"
        task_queue.put(new_task)
        # 加间隔避免循环占用过多资源
        threading.Event().wait(0.5)

def worker(task):
    # 模拟任务处理逻辑
    print(f"完成任务: {task}")

2. 让主线程的队列读取逻辑能退出 #

不要一直死等队列，给get()加超时，同时循环检查终止信号：

def main():
    executor = ThreadPoolExecutor(max_workers=2)
    # 提交任务生成线程
    executor.submit(task_submit)
    
    while not stop_flag.is_set():
        try:
            # 超时1秒，无任务时跳出等待，检查终止信号
            task = task_queue.get(timeout=1)
            executor.submit(worker, task)
        except queue.Empty:
            # 队列为空时继续循环，判断是否需要终止
            continue
    
    # 先触发终止信号，让task_submit停止循环
    stop_flag.set()
    # 关闭线程池，等待所有任务处理完成
    executor.shutdown(wait=True)

if __name__ == "__main__":
    # 捕获Ctrl+C信号触发终止
    try:
        main()
    except KeyboardInterrupt:
        stop_flag.set()

3. 可选：把持续运行的线程设为守护线程 #

如果不需要等待task_submit或未完成的任务，可直接将其设为守护线程，主线程退出时它会强制终止，但可能导致未完成任务丢失，需谨慎使用：

# 替代ThreadPoolExecutor提交方式，直接用Thread创建并设为守护线程
submit_thread = threading.Thread(target=task_submit)
submit_thread.daemon = True
submit_thread.start()

核心提醒 #

• 所有无限循环的线程必须添加明确的终止条件，不能让线程无限制“野跑”。
• 阻塞式队列操作一定要配合超时或终止信号，避免线程卡死在get()步骤。
• 调用executor.shutdown(wait=True)时，要确保所有提交的任务都能正常结束，否则会一直处于等待状态。

内容的提问来源于stack exchange，提问作者user20426821