Python多进程：找到问题解后如何终止子进程并返回结果

针对你的多进程改造需求，这里提供一套可靠的实现方案，核心是用共享终止信号和结果队列来协调主进程与子进程：

核心思路 #

• 用multiprocessing.Queue做跨进程任务队列，另一个队列专门传递找到的解
• 用multiprocessing.Event作为全局终止信号：只要有一个子进程找到解，就触发该信号，所有子进程检测到后立即停止工作
• 主进程等待结果队列返回解，之后触发终止信号，确保所有子进程优雅退出

完整代码示例 #

import multiprocessing
from multiprocessing import Queue, Event

# ---------------------- 替换为你的业务逻辑 ----------------------
def is_solution(task):
    # 模拟判断当前任务是否是解
    return task == "target_solution"

def condition_met(task):
    # 模拟判断是否需要生成新任务
    return len(task) < 15

def generate_next_task(task):
    # 模拟生成下一个任务
    return task + "_next"
# -------------------------------------------------------------

def worker(task_queue, result_queue, stop_event):
    while not stop_event.is_set():
        try:
            # 带超时获取任务，避免子进程在空队列上无限阻塞
            current_task = task_queue.get(timeout=1)
        except multiprocessing.queues.Empty:
            # 任务队列已空，直接退出
            break

        # 再次检查终止信号，避免刚拿到任务就收到停止指令
        if stop_event.is_set():
            # 可选：把未处理的任务放回队列（如果需要后续复用）
            # task_queue.put(current_task)
            break

        # 检查是否找到解
        if is_solution(current_task):
            result_queue.put(current_task)
            # 触发终止信号，通知所有子进程停止
            stop_event.set()
            break

        # 满足条件则生成新任务加入队列
        if condition_met(current_task):
            next_task = generate_next_task(current_task)
            task_queue.put(next_task)

if __name__ == "__main__":
    # 初始化跨进程共享资源
    task_queue = Queue()
    result_queue = Queue()
    stop_event = Event()

    # 放入初始任务
    task_queue.put("initial_task")

    # 启动子进程（根据CPU核心数调整数量）
    worker_count = 4
    processes = []
    for _ in range(worker_count):
        p = multiprocessing.Process(target=worker, args=(task_queue, result_queue, stop_event))
        p.start()
        processes.append(p)

    # 主进程阻塞等待结果
    solution = result_queue.get()
    print(f"找到问题解：{solution}")

    # 触发终止信号，让所有子进程优雅退出
    stop_event.set()

    # 等待所有子进程结束
    for p in processes:
        p.join()

关键细节说明 #

1. 跨进程队列选择：必须用multiprocessing.Queue，不能用标准库的queue.Queue——后者不支持跨进程共享，会导致子进程无法访问队列内容。
2. 终止信号的作用：Event是轻量级的跨进程同步原语，所有子进程循环时都会优先检查它的状态，一旦触发就能快速停止，避免无效计算。
3. 任务队列的超时获取：用get(timeout=1)避免子进程在任务队列空了之后一直阻塞，确保即使没有解也能正常退出。
4. 优雅退出 vs 强制终止：优先用Event让子进程自行退出，比直接调用p.terminate()更安全，能避免资源泄漏（比如未关闭的文件、未释放的锁）。

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