如何用Python自动重启参数一致的子进程？现有实现是否最优？

原实现的几个问题 #

• 遍历列表时修改列表的风险：在for n, p in enumerate(children)循环里执行children.pop(n)，会导致后续元素的索引错位，可能漏掉对某些进程的存活检查。
• pid复用的潜在问题：用进程pid作为args字典的键，操作系统可能会复用已终止进程的pid，导致参数映射错误。
• 数据结构冗余：分开维护children列表和args字典，增加了代码的维护成本，容易出现数据不一致的情况。

更符合Python风格的实现方式 #

我们可以将进程实例和它的启动参数绑定在一起，用单一的数据结构管理，同时避免遍历过程中修改列表的问题。以下是两种优化方案：

方案1：用字典封装进程与参数 #

import multiprocessing
import time
import random

def printNumber(number):
    print("starting :", number)
    while random.randint(0, 5) > 0:
        print(number)
        time.sleep(2)

if __name__ == '__main__':
    # 用列表存储字典，每个字典包含进程实例和启动参数
    processes = []
    for processNumber in range(10, 15):
        p = multiprocessing.Process(target=printNumber, args=(processNumber,))
        p.start()
        processes.append({"proc": p, "args": (processNumber,)})

    while True:
        time.sleep(1)
        # 先收集需要重启的进程信息，避免遍历中修改列表
        to_restart = []
        for idx, proc_info in enumerate(processes):
            if not proc_info["proc"].is_alive():
                to_restart.append((idx, proc_info["args"]))
        
        # 倒序批量处理重启，避免pop操作影响剩余元素的索引
        for idx, args in reversed(to_restart):
            # 移除已终止的进程
            dead_proc = processes.pop(idx)["proc"]
            print(f"重启进程，原参数：{args[0]}")
            
            # 启动新进程
            new_proc = multiprocessing.Process(target=printNumber, args=args)
            new_proc.start()
            processes.append({"proc": new_proc, "args": args})

方案2：用dataclass封装（Python 3.7+） #

如果追求更清晰的结构，可以用dataclasses模块定义进程信息，让代码可读性更强：

import multiprocessing
import time
import random
from dataclasses import dataclass

@dataclass
class ProcessInfo:
    proc: multiprocessing.Process
    args: tuple

def printNumber(number):
    print("starting :", number)
    while random.randint(0, 5) > 0:
        print(number)
        time.sleep(2)

if __name__ == '__main__':
    processes = []
    for processNumber in range(10, 15):
        p = multiprocessing.Process(target=printNumber, args=(processNumber,))
        p.start()
        processes.append(ProcessInfo(proc=p, args=(processNumber,)))

    while True:
        time.sleep(1)
        to_restart = []
        for idx, proc_info in enumerate(processes):
            if not proc_info.proc.is_alive():
                to_restart.append((idx, proc_info.args))
        
        for idx, args in reversed(to_restart):
            dead_proc = processes.pop(idx).proc
            print(f"重启进程，原参数：{args[0]}")
            
            new_proc = multiprocessing.Process(target=printNumber, args=args)
            new_proc.start()
            processes.append(ProcessInfo(proc=new_proc, args=args))

优化点说明 #

• 避免遍历中修改列表：先收集所有需要重启的进程索引和参数，再倒序处理（倒序pop不会影响未处理元素的索引），解决了原代码中索引错位的问题。
• 绑定进程与参数：用单一结构存储进程和参数，避免了原代码中两个数据结构不一致的风险。
• 避免pid复用问题：不再依赖pid映射参数，直接从封装的结构中获取参数，逻辑更可靠。

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