Python中如何等待multiprocessing.connection.Client可用？

关于282次连接尝试的影响 #

短时间内对本地localhost端口发起282次连接尝试，不会造成网络层面的问题，但你的当前实现是无间隔的忙等循环，会持续占用CPU资源（循环一直在执行，没有休眠），这是主要弊端。如果主进程还有其他业务逻辑，高CPU占用会影响整体性能。

更优的实现方案 #

方案1：给重试添加固定延迟（最简单，无需修改process.py） #

在每次连接失败后休眠一小段时间，既减少尝试次数，又降低CPU占用。同时建议设置最大尝试次数，避免无限循环：

from multiprocessing.connection import Client
import time

max_attempts = 100  # 限制最大尝试次数，防止无限等待
wait_interval = 0.1  # 每次重试间隔0.1秒
n_attempts = 0

while n_attempts < max_attempts:
    try:
        with Client(('localhost', 6000)) as connection:
            connection.send(n_attempts)
            message = connection.recv()
            # 连接成功，退出循环
            break
    except ConnectionRefusedError:
        n_attempts += 1
        time.sleep(wait_interval)
else:
    # 超过最大尝试次数仍未连接成功，抛出异常或处理错误
    raise RuntimeError("无法连接到process.py服务，已达到最大尝试次数")

方案2：让子进程主动发送就绪信号（最可靠，需修改process.py） #

通过标准输出、进程管道等方式，让process.py在启动Listener并准备好接受连接后，主动通知主进程，避免盲目重试。

方式A：使用标准输出传递信号

主进程代码：

import threading
import subprocess

# 启动子进程并捕获标准输出
proc = subprocess.Popen(
    ['python', 'process.py'],
    stdout=subprocess.PIPE,
    stderr=subprocess.PIPE,
    text=True
)

# 等待子进程输出就绪标记
for line in proc.stdout:
    if line.strip() == "SERVER_READY":
        print("子进程已准备好接受连接")
        break

修改后的process.py：

from multiprocessing.connection import Listener
import sys

with Listener(('localhost', 6000)) as listener:
    # 输出就绪标记，通知主进程
    print("SERVER_READY")
    sys.stdout.flush()  # 强制刷新输出缓冲区，确保主进程能立即收到
    with listener.accept() as connection:
        message = connection.recv()
        connection.send(message)

方式B：使用进程间管道传递信号

主进程代码：

import threading
import subprocess
from multiprocessing import Pipe

# 创建双向管道
parent_conn, child_conn = Pipe()

def start_subprocess():
    # 将子管道的文件描述符传递给子进程（Unix系统适用，Windows需调整）
    subprocess.run(
        ['python', 'process.py', str(child_conn.fileno())],
        pass_fds=[child_conn.fileno()]
    )

threading.Thread(target=start_subprocess).start()

# 等待子进程发送的就绪信号
ready_signal = parent_conn.recv()
if ready_signal == "READY":
    print("子进程已就绪")

修改后的process.py：

from multiprocessing.connection import Listener
from multiprocessing import Pipe
import sys

# 获取主进程传递的管道文件描述符
child_conn = Pipe(int(sys.argv[1]))

with Listener(('localhost', 6000)) as listener:
    # 发送就绪信号给主进程
    child_conn.send("READY")
    with listener.accept() as connection:
        message = connection.recv()
        connection.send(message)

方案对比 #

• 方案1：无需修改子进程代码，实现成本低，但仍存在少量重试，适合快速迭代场景。
• 方案2：能精准感知服务就绪状态，无多余重试，可靠性更高，但需要修改子进程代码，适合对性能和可靠性要求较高的场景。

内容的提问来源于stack exchange，提问作者arthur.sw