如何在concurrent.futures.ThreadPoolExecutor中获取当前工作线程ID

以下两种方案均可实现需求，根据是否需要固定1-5范围的顺序ID选择即可：

方案1：自定义1-5顺序Worker ID（推荐，匹配输出要求） #

利用ThreadPoolExecutor提供的initializer初始化参数+线程本地存储threading.local()，给每个工作线程预分配唯一的顺序ID，全程不需要修改任务提交的传参逻辑：

实现步骤 #

1. 定义线程本地存储变量，用于每个线程存储自己专属的Worker ID
2. 编写线程初始化函数，工作线程启动时自动领取一个唯一ID
3. 创建线程池时传入初始化函数和预生成的ID列表
4. 任务内部直接读取线程本地存储的ID即可使用

完整示例代码 #

import concurrent.futures
import threading

# 线程本地存储：每个线程只会读写自己的变量，不会出现线程安全问题
thread_local = threading.local()

def worker_init(available_ids):
    # 工作线程初始化时领取一个专属ID，5个线程刚好拿完1-5的ID
    thread_local.worker_id = available_ids.pop()

def worker_task(command, site):
    # 直接获取当前工作线程的ID
    worker_id = thread_local.worker_id
    # 你的业务逻辑
    print(f'worker {worker_id} is processing {site}')
    # 实现要求的标识输出
    print(f'worker {worker_id} has finished')
    # 返回业务数据
    return f"{site} processed"

if __name__ == "__main__":
    # 你的业务参数
    URLS = ["https://example.com", "https://google.com", "https://github.com"]
    command_ = "your business command"
    # 预生成1-5的Worker ID列表
    worker_ids = list(range(1, 6))

    with concurrent.futures.ThreadPoolExecutor(
        max_workers=5,
        # 传入线程初始化函数
        initializer=worker_init,
        # 初始化函数的入参，注意加逗号变成元组
        initargs=(worker_ids,)
    ) as executor:
        future_to_url = {executor.submit(worker_task, command_, site_): site_ for site_ in URLS}
        for future in concurrent.futures.as_completed(future_to_url):
            url = future_to_url[future]
            try:
                data = future.result()
                # 你的后续业务逻辑
            except Exception as e:
                print(f"site {url} run error: {str(e)}")

方案2：直接获取系统原生线程ID（无额外配置） #

如果不需要固定的1-5顺序编号，只要能区分不同工作线程，无需修改线程池配置，直接在任务内调用threading.get_ident()获取系统分配的线程唯一标识即可：

import threading

def worker_task(command, site):
    # 直接获取当前线程的唯一标识，不同线程值不同
    worker_id = threading.get_ident()
    print(f'worker {worker_id} has finished')
    # 其余业务逻辑

该方案的优点是无需额外配置，缺点是线程ID是系统生成的长整数，不是1-5的顺序编号。

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