如何处理存在依赖关系的两个任务？当前实现是否合理优雅？

我们有任务A和任务B，B在执行过程中需要A的结果，通过轮询等待该结果。若A因故障退出，B存在两种退出情况：

• 在A退出前因其他原因退出；
• 在A退出后因A或其他原因退出。

对于第二种情况，B可能远晚于A退出，但理想状态下我们希望在得知A退出后立即终止B。我编写了如下代码：

import subprocess
import psutil

B_proc = subprocess.Popen(["do_B"])

try:
    subprocess.run(["do_A"], check=True)
except subprocess.CalledProcessError:
    # We do not detect if B_proc exits or not.
    parent = psutil.Process(B_proc.pid)
    for child in parent.children(recursive=True):
        child.kill()
    parent.kill()
    raise Exception("A exits")

B_proc.communicate()

if B_proc.poll() != 0:
    raise Exception("B exits")

当A退出时，我们不检测B的状态，直接终止它而不调用communicate()/poll()。请问这种处理方式是否正确且优雅？目前该代码运行正常。

正确性分析 #

你的代码能实现核心需求，但存在几个潜在风险：

1. 误杀无关进程：如果B在A异常退出前已经自行终止，B_proc.pid可能被系统分配给其他进程，此时用psutil操作该PID会误杀无关进程，这是严重的安全隐患。
2. 资源泄漏：未调用communicate()会导致B的stdout/stderr缓冲区满时，进程阻塞无法正常退出；即使被kill，缓冲区资源也可能无法正确回收。
3. 异常信息丢失：raise Exception("A exits")覆盖了原CalledProcessError的错误详情，不利于排查A的具体故障原因。

优雅性优化建议 #

代码逻辑清晰，但可以从安全性、资源回收、可读性三个方向优化：

1. 先校验B的存活状态：在终止B前调用B_proc.poll()，返回None说明进程仍在运行，再执行kill操作，避免误操作。
2. 强制回收进程资源：无论B是否存活，调用B_proc.communicate()回收缓冲区资源，防止泄漏；如果担心阻塞，可以加timeout参数。
3. 保留原异常上下文：直接raise原异常，或者用raise Exception("A exits") from e的方式关联原异常，保留错误链。
4. 简化子进程终止逻辑：如果B不会创建子进程，无需依赖psutil，直接调用B_proc.kill()即可；如果需要处理子进程组，可以在启动B时加start_new_session=True，然后发送信号给整个进程组。

优化后的示例代码 #

import subprocess
import psutil

B_proc = subprocess.Popen(["do_B"])

try:
    subprocess.run(["do_A"], check=True)
except subprocess.CalledProcessError as e:
    # 先检查B是否仍在运行
    if B_proc.poll() is None:
        try:
            parent = psutil.Process(B_proc.pid)
            # 先尝试优雅终止，失败再强制kill
            for child in parent.children(recursive=True):
                child.terminate()
            parent.terminate()
            parent.wait(timeout=3)
        except (psutil.NoSuchProcess, psutil.TimeoutExpired):
            if B_proc.poll() is None:
                parent.kill()
        finally:
            # 回收缓冲区资源
            B_proc.communicate(timeout=2)
    # 保留原异常信息
    raise Exception("A执行失败") from e

# 正常流程等待B完成并校验结果
B_proc.communicate()
if B_proc.returncode != 0:
    raise Exception(f"B执行失败，退出码：{B_proc.returncode}")

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