Python如何优雅等待指定子进程列表中首个进程结束？

嘿，那种轮询的方式确实够繁琐的，还在浪费CPU资源——咱们换个更优雅的思路，用操作系统的IO多路复用机制（select或epoll）来实现无轮询的等待，完全符合你的需求。

核心原理 #

当子进程终止时，它的标准输出/标准错误管道会被内核关闭，对应的文件描述符会触发可读事件（因为管道里会出现EOF）。我们可以通过select或epoll监听这些文件描述符，一旦有事件触发，就知道对应的子进程已经结束了。这种方式是系统主动通知，不需要我们定时轮询。

方案1：跨平台的select实现 #

select是POSIX标准的多路复用接口，在Windows、macOS、Linux上都能运行，适合跨平台场景：

import subprocess
import select

# 1. 创建需要等待的子进程列表（注意要指定stdout/stderr为PIPE）
processes = [
    subprocess.Popen(['sleep', '5'], stdout=subprocess.PIPE, stderr=subprocess.PIPE),
    subprocess.Popen(['sleep', '10'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
]

# 2. 建立文件描述符到进程的映射
fd_to_process = {p.stdout.fileno(): p for p in processes}
watch_fds = list(fd_to_process.keys())

# 3. 阻塞等待任意一个fd触发可读事件
ready_fds, _, _ = select.select(watch_fds, [], [])

# 4. 处理第一个结束的进程
for fd in ready_fds:
    target_process = fd_to_process[fd]
    return_code = target_process.wait()
    print(f"进程 {target_process.pid} 已结束，返回码: {return_code}")
    break  # 只处理第一个结束的进程

方案2：Linux专属的epoll实现 #

如果你的代码只在Linux环境运行，epoll的性能比select好得多（尤其是子进程数量较多时）：

import subprocess
import select

# 1. 创建子进程列表
processes = [
    subprocess.Popen(['sleep', '5'], stdout=subprocess.PIPE, stderr=subprocess.PIPE),
    subprocess.Popen(['sleep', '10'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
]

# 2. 初始化epoll并注册监听事件
epoll = select.epoll()
fd_to_process = {}

for proc in processes:
    fd = proc.stdout.fileno()
    # 注册可读事件（子进程结束时管道关闭会触发EPOLLIN）
    epoll.register(fd, select.EPOLLIN)
    fd_to_process[fd] = proc

# 3. 阻塞等待事件触发
events = epoll.poll()  # 无超时，直到有事件发生
epoll.close()

# 4. 处理第一个结束的进程
for fd, event in events:
    if event & select.EPOLLIN:
        target_proc = fd_to_process[fd]
        return_code = target_proc.wait()
        print(f"进程 {target_proc.pid} 已结束，返回码: {return_code}")
        break

关键注意事项 #

1. 必须指定stdout=subprocess.PIPE：如果不指定，Popen不会创建管道文件描述符，我们就没法监听进程结束的事件。即使你的子进程没有输出内容也没关系——进程终止时管道关闭会触发EOF，同样会触发可读事件。
2. 避免无关进程干扰：我们只监听自己创建的子进程的文件描述符，完全不会拿到其他进程的结果，比os.wait()更安全。
3. 性能优势：这种方式是被动通知，CPU不会在循环中空转，比你原来的轮询方案高效得多，尤其是子进程运行时间较长时。

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