Python 中使用 fork 多进程时实现统一进度条的正确方法

夜墨酱_8647

夜墨酱_8647

2026-04-04

641人浏览

原创

Python 中使用 fork 多进程时实现统一进度条的正确方法

在基于 os.fork() 的多进程场景中,子进程无法直接更新父进程的 Rich 进度条(因内存隔离),需通过进程间通信或重构为单进程+异步/线程模型来实现可靠进度反馈。

在基于 `os.fork()` 的多进程场景中,子进程无法直接更新父进程的 rich 进度条(因内存隔离),需通过进程间通信或重构为单进程+异步/线程模型来实现可靠进度反馈。

使用 os.fork() 时,父子进程拥有独立的内存空间,rich.Progress 实例(包括其内部状态、任务计数器和渲染器)不会跨进程共享。因此,在子进程中调用 progress.advance() 完全无效——它仅修改子进程本地副本,对父进程的进度条无任何影响。原问题中尝试在 else 分支(子进程)中调用 progress.advance(),本质上是无效操作。

❌ 错误模式:在 fork 子进程中更新父进程进度条

from rich.progress import Progress
import os

with Progress() as progress:
    task = progress.add_task("Processing", total=len(args_list))
    for balanced_batch in even_batches_it:
        pid = os.fork()
        if pid > 0:  # 父进程
            continue  # 等待或管理子进程
        else:  # 子进程 —— 此处的 advance 不会影响父进程进度条!
            for my_tuple in balanced_batch:
                do_something(my_tuple)
            # ⚠️ 下行无效:progress 是子进程中的新对象(或未定义),task 也非同一实例
            progress.advance(task, advance=len(balanced_batch))  # 错误!
            os._exit(0)  # 子进程必须显式退出

✅ 推荐方案:避免 fork + 进度条耦合,改用更可控的并发模型

方案 1:使用 concurrent.futures.ProcessPoolExecutor(推荐)

它天然支持主线程收集结果,并可安全更新进度条:

from rich.progress import Progress, TaskID
from concurrent.futures import ProcessPoolExecutor, as_completed
import time

def worker(batch: list) -> int:
    """子任务函数:执行实际工作,返回处理项数"""
    for item in batch:
        do_something(item)
        time.sleep(0.01)  # 模拟耗时操作
    return len(batch)

# 主流程:单进程驱动进度条,多进程执行任务
with Progress() as progress:
    task: TaskID = progress.add_task("Processing batches...", total=len(even_batches_it))

    with ProcessPoolExecutor(max_workers=os.cpu_count()) as executor:
        # 提交所有任务
        future_to_batch = {
            executor.submit(worker, batch): batch 
            for batch in even_batches_it
        }

        # 按完成顺序更新进度
        for future in as_completed(future_to_batch):
            try:
                count = future.result()  # 获取子进程返回的处理数量
                progress.update(task, advance=1)  # 每完成一个 batch +1
            except Exception as exc:
                print(f'Batch generated an exception: {exc}')

方案 2:若必须用 fork,则通过 IPC 同步进度(如 multiprocessing.Manager)

from rich.progress import Progress
from multiprocessing import Manager, Process
import os

def child_worker(batch, counter, lock):
    for item in batch:
        do_something(item)
    with lock:
        counter.value += 1  # 原子递增已完成 batch 数

if __name__ == "__main__":
    with Manager() as manager:
        counter = manager.Value('i', 0)
        lock = manager.Lock()
        processes = []

        with Progress() as progress:
            task = progress.add_task("Fork-based processing...", total=len(even_batches_it))

            for batch in even_batches_it:
                p = Process(target=child_worker, args=(batch, counter, lock))
                p.start()
                processes.append(p)

            # 主进程轮询更新进度条
            while any(p.is_alive() for p in processes):
                time.sleep(0.1)
                with lock:
                    progress.update(task, completed=counter.value)

            # 等待全部结束
            for p in processes:
                p.join()

⚠️ 关于答案中 exec() 动态注入方案的说明

原答案提供的 run(func) 方法虽具创意,但存在严重缺陷:

Biopython
Biopython

全面的分子生物学工具集。适用于序列操作、文件解析(FASTA/GenBank/PDB)、系统发育分析,以及通过 Bio.Entrez 以编程方式访问 NCBI/

下载
  • 使用 exec() 执行动态生成代码,破坏可读性、调试性与安全性;
  • 强制按行计数更新进度,语义失真(如 import 行耗时长,循环体行执行多次,但进度条仅按行数而非实际工作量推进);
  • 完全规避了多进程需求,退化为单进程串行执行,丧失并行优势;
  • inspect.getsource() 在闭包、内联函数、Jupyter 环境中常失效。

因此,该方案不适用于真实生产场景,仅可作为教学反例参考。

总结建议

  • ✅ 优先选用 concurrent.futures.ProcessPoolExecutor 或 multiprocessing.Pool,它们提供清晰的主从通信契约;
  • ✅ 进度更新逻辑必须在主(控制)进程内执行,子进程只负责计算并返回结果;
  • ❌ 避免在 fork 子进程中尝试操作父进程 UI 对象;
  • ❌ 拒绝 exec()、eval() 等动态代码执行方式用于生产级进度控制;
  • ? 若需精细控制(如实时吞吐量统计),可结合 multiprocessing.Queue 由子进程推送进度事件。

正确的并发进度设计,核心在于分离关注点:计算逻辑交给子进程,状态聚合与可视化保留在主进程——这才是健壮、可维护、可扩展的工程实践。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1911

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

5404

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1869

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

32877

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3867

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3887

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1283

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

616

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

3003

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程