Linux中如何准确获取一个进程及其子进程的总内存使用情况？

2024-03-20 16:42:41

如何在 Linux 中获取通过 shell fork 众多子进程的进程的总内存？

问题

在 Linux 系统中，当一个进程通过 shell fork 产生多个子进程时，计算父进程及其所有子进程的总内存使用情况至关重要。传统方法存在局限性，可能会导致计算错误。

准确的方法

要准确计算总内存使用情况，需遵循以下步骤：

1. 获取进程 ID (PID)

ps aux | grep httpd | grep -v grep | awk '{print $2}'

2. 访问 procfs

/proc/<PID>/smaps

3. 解析 smaps 文件

TotalPss: (?P<TotalPss>\d+) kB

4. 累加子进程内存使用情况

重复步骤 1-3，累加每个子进程的 TotalPss 值，得到总内存使用情况。

示例代码 (Python)

def get_process_memory_usage(pid):
    child_pids = subprocess.check_output(["pgrep", "-P", str(pid)]).decode("utf-8").split()
    total_pss = 0
    for child_pid in child_pids:
        with open(f"/proc/{child_pid}/smaps", "r") as f:
            match = re.search(r"TotalPss: (?P<TotalPss>\d+) kB", f.read())
            if match:
                total_pss += int(match.group("TotalPss"))
    return total_pss

pid = int(subprocess.check_output(["pgrep", "-f", "httpd"]).decode("utf-8"))
total_pss = get_process_memory_usage(pid)
print(f"httpd 及其子进程的总内存使用情况：{total_pss} kB")