Python并发场景下如何在首个任务返回True后立即跳过其余未完成IO任务

问题原因 #

原有代码存在两个核心问题导致无法达到预期效果：

1. ThreadPoolExecutor的with上下文管理器退出时会默认执行shutdown(wait=True)，必须等待所有提交的任务全部执行完成才会退出上下文块，执行后续的打印逻辑。
2. 没有监听任务的返回结果，无法第一时间捕捉到首个返回True的信号来提前终止等待。

修改后可实现需求的代码 #

import concurrent.futures
import time
import threading

# 初始化守护线程，主线程退出时子线程会直接终止，避免多余输出
def init_daemon():
    threading.current_thread().daemon = True

def one():
    time.sleep(2)
    print('First')
    return False

def two():
    time.sleep(1)
    print('Second')
    return True

def three():
    time.sleep(4)
    print('Third')
    return False

tasks = [one, two, three]
# 手动创建线程池，不使用with上下文避免自动等待所有任务
executor = concurrent.futures.ThreadPoolExecutor(max_workers=3, initializer=init_daemon)
futures = [executor.submit(task) for task in tasks]

# 按任务完成顺序获取返回值
for future in concurrent.futures.as_completed(futures):
    res = future.result()
    if res is True:
        # 找到第一个返回True的任务，直接关闭线程池不等待剩余任务
        # Python3.9+可加cancel_futures=True参数，直接取消未开始的任务
        executor.shutdown(wait=False)
        break

print('Move on..')

核心逻辑说明 #

• 放弃默认的with上下文管理，手动控制线程池的关闭逻辑，避免强制等待所有任务完成。
• 通过concurrent.futures.as_completed按任务完成的先后顺序获取返回结果，无需等待全部任务跑完就可以处理已完成任务的返回值。
• 检测到首个返回True的任务后，立即调用shutdown(wait=False)关闭线程池，主线程不会等待未完成的任务，直接执行后续逻辑。
• 给线程池配置守护线程初始化方法，主线程退出时所有未完成的子线程会直接被终止，不会产生多余的打印输出。

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