Python线程池回调函数：主线程还是工作线程？
Python线程池中的回调函数执行位置取决于Future对象的状态。具体而言：
-
Future已完成: 如果在调用add_done_callback时，Future对象已完成其任务，回调函数将在主线程执行。 -
Future未完成: 如果Future对象在add_done_callback调用时仍在执行，回调函数将在工作线程中执行，一旦Future对象完成任务。
代码示例及分析：
以下代码演示了回调函数在不同场景下的执行位置：
立即学习“Python免费学习笔记（深入）”；
<code class="python">import threading
import time
from concurrent.futures import ThreadPoolExecutor
def task():
time.sleep(1)
return "Task Completed"
def callback(future):
print(f"Callback executed in thread: {threading.current_thread().name}")
print(f"Future result: {future.result()}")
with ThreadPoolExecutor(max_workers=5) as executor:
# 场景1：任务完成前调用回调
futures1 = [executor.submit(task) for _ in range(5)]
for future in futures1:
future.add_done_callback(callback)
# 场景2：任务完成后调用回调
futures2 = [executor.submit(task) for _ in range(5)]
time.sleep(2) # 确保任务完成
for future in futures2:
future.add_done_callback(callback)</code>
运行结果将显示部分回调函数在主线程执行，部分在工作线程执行，这取决于任务完成的时机。 场景1中，由于任务执行时间较短，部分回调函数可能在主线程执行。场景2中，任务已完成，回调函数一定在工作线程执行。
最佳实践：
为了避免阻塞主线程，建议在回调函数中避免执行耗时操作。 回调函数主要用于处理异步任务的结果，例如记录日志或更新UI。 如果需要进行耗时处理，应该在单独的线程或进程中执行。
