本文介绍如何在 Go 中基于 /proc/[pid]/smaps 文件实时获取任意运行中子进程的内存占用（PSS），精度优于 10MB，无需等待进程结束。
本文介绍如何在 go 中基于 `/proc/[pid]/smaps` 文件实时获取任意运行中子进程的内存占用（pss），精度优于 10mb，无需等待进程结束。
在 Go 中使用 os/exec 启动外部命令后，若需持续、低开销地监控其内存消耗（例如实现内存熔断、资源预警或性能分析），标准库并不提供 cmd.Memory() 这类接口。但借助 Linux 内核暴露的 /proc 文件系统，我们可高效、安全地读取进程的 Proportional Set Size（PSS） —— 这是衡量进程实际内存贡献的权威指标（已按共享内存比例折算），单位为 KB，误差通常远低于 10 MB，完全满足生产级实时监控需求。
以下是一个健壮、可复用的内存采集函数：
package main
import (
"bufio"
"bytes"
"fmt"
"os"
"strconv"
"strings"
"syscall"
"time"
)
// calculateMemory returns the PSS (Proportional Set Size) of the process with given PID, in KB.
// Returns error if /proc/[pid]/smaps cannot be read (e.g., permission denied or process exited).
func calculateMemory(pid int) (uint64, error) {
path := fmt.Sprintf("/proc/%d/smaps", pid)
f, err := os.Open(path)
if err != nil {
return 0, fmt.Errorf("failed to open %s: %w", path, err)
}
defer f.Close()
var totalPss uint64
scanner := bufio.NewScanner(f)
for scanner.Scan() {
line := bytes.TrimSpace(scanner.Bytes())
if bytes.HasPrefix(line, []byte("Pss:")) {
// Format: "Pss: 1234 kB"
fields := bytes.Fields(line)
if len(fields) < 2 {
continue
}
numStr := strings.TrimSuffix(string(fields[1]), "kB")
numStr = strings.TrimSpace(numStr)
if n, err := strconv.ParseUint(numStr, 10, 64); err == nil {
totalPss += n
}
}
}
if err := scanner.Err(); err != nil {
return 0, fmt.Errorf("error reading %s: %w", path, err)
}
return totalPss, nil
}✅ 使用示例：启动命令并周期性检查内存
func main() {
cmd := exec.Command("stress", "--vm", "1", "--vm-bytes", "200M", "--timeout", "30s")
if err := cmd.Start(); err != nil {
panic(err)
}
defer cmd.Process.Kill() // 确保清理
ticker := time.NewTicker(2 * time.Second)
defer ticker.Stop()
for {
select {
case <-ticker.C:
memKB, err := calculateMemory(cmd.Process.Pid)
if err != nil {
fmt.Printf("Warning: failed to get memory for PID %d: %v\n", cmd.Process.Pid, err)
continue
}
memMB := memKB / 1024
fmt.Printf("PID %d memory usage: %d MB (PSS)\n", cmd.Process.Pid, memMB)
if memMB > 50 {
fmt.Println("⚠️ Oh my god, this process is hungry for memory!")
// 可在此触发告警、限流或主动终止
_ = cmd.Process.Signal(syscall.SIGTERM)
return
}
case <-time.After(35 * time.Second): // 超时保护
return
}
}
}⚠️ 关键注意事项：
- 权限要求：调用进程需有权限读取 /proc/[pid]/smaps。普通用户默认可读自身启动的进程；若需监控其他用户进程，需 CAP_SYS_PTRACE 或 root 权限。
- 进程生命周期：/proc/[pid]/smaps 在进程退出后立即失效，务必在 cmd.Process.Pid 有效期内调用，并妥善处理 os.IsNotExist 错误。
- PSS vs RSS：优先使用 Pss:（非 Rss:），因 PSS 公平分摊共享内存（如动态库、mmap 区域），更真实反映单个进程的内存“责任”，避免重复计算。
- 性能与开销：每次读取约 10–100 KB（取决于进程映射段数量），解析为 O(n) 时间复杂度，毫秒级完成，适合每秒数次的监控频率。
- 跨平台限制：此方案仅适用于 Linux。macOS 和 Windows 需依赖 psutil 绑定或平台专用 API（如 Windows 的 GetProcessMemoryInfo）。
? 进阶建议：
- 将 calculateMemory 封装为 ProcessMonitor 结构体，支持自动重试、采样率控制和内存增长速率计算；
- 结合 cgroup v2（如 /sys/fs/cgroup/.../memory.current）实现容器化环境下的统一监控；
- 使用 pprof + runtime.ReadMemStats 辅助对比 Go 自身堆内存，区分 native 与 Go runtime 开销。
通过该方法，你可在不引入 C 依赖、不执行 shell 命令（如 ps 或 pmap）的前提下，以原生 Go 实现高精度、低延迟的进程内存实时观测——这是构建可观测性基础设施的关键一环。
