2. pprof入门

性能调优的第一步是“可观测”。在 Go 生态中，pprof 是官方提供的运行时性能分析工具，它能够以极低的侵入性采集 CPU、内存、阻塞、互斥锁、goroutine 等关键指标，帮助开发者快速定位性能瓶颈。本节将系统讲解 pprof 的定义、核心工作原理以及最常用的三种采样方式。

pprof 是什么

pprof 是 Go 标准库 runtime/pprof 和 net/http/pprof 提供的性能分析工具集。它通过周期性采样或事件触发的方式，收集程序运行时的堆栈信息，并生成可视化的分析报告。pprof 支持两种使用模式：

• 内置集成：在代码中导入 net/http/pprof，通过 HTTP 端点实时获取分析数据。
• 离线采集：使用 runtime/pprof 在特定代码段前后手动启动和停止采样，生成 profile 文件。

pprof 的核心价值在于：将程序“看不见”的执行行为转化为可量化的火焰图、调用图、堆栈统计，让开发者能精确回答“CPU 花在哪里”“内存被谁占用”“goroutine 卡在何处”等问题。

pprof 的工作原理

pprof 的采样机制基于 Go 运行时内置的信号处理和事件钩子。其核心流程如下：

1. 注册采样器：Go 运行时在初始化时注册 CPU、内存、阻塞等采样器，每个采样器绑定一个信号处理器或事件回调。
2. 周期性触发：CPU 采样由操作系统定时器（SIGPROF 信号）驱动，内存和阻塞采样由运行时事件（如 malloc、channel 操作）触发。
3. 记录堆栈：每次触发时，采样器暂停当前 goroutine，读取其调用栈（PC 寄存器链），并将堆栈哈希后存入全局缓冲区。
4. 数据聚合：采样结束后，pprof 工具对堆栈样本进行聚合统计，生成按调用路径分组的计数或时间分布。
5. 可视化输出：通过 go tool pprof 或 pprof 交互式界面，将聚合数据渲染为文本、SVG 火焰图、调用图等格式。

下图展示了 pprof 从采样到可视化的完整工作流：

pprof 的三种采样方式

pprof 支持多种采样类型，其中 CPU 采样、内存采样 和 阻塞采样 是最基础、最常用的三种。下表从采样目标、触发方式、数据含义和典型场景四个维度进行对比：

1. CPU 采样

CPU 采样是 pprof 最常用的模式。Go 运行时注册一个 SIGPROF 信号处理器，操作系统以固定频率（默认每秒 100 次）向进程发送该信号。每次信号到达时，采样器中断当前 goroutine 的执行，记录其调用栈。采样结束后，pprof 统计每个函数在采样点中出现的次数，近似反映 CPU 时间消耗分布。

启动 CPU 采样的典型代码：

import (
    "os"
    "runtime/pprof"
)

func main() {
    f, _ := os.Create("cpu.pprof")
    pprof.StartCPUProfile(f)
    defer pprof.StopCPUProfile()

    // 业务逻辑代码
    doWork()
}

2. 内存采样

内存采样关注的是 堆内存分配。Go 运行时在每次内存分配（malloc）时，以一定概率（默认每 512 KB 分配采样一次）记录当前堆栈。采样结果包含分配次数、分配字节数、存活对象数等指标。内存采样不记录栈内存（局部变量），只记录堆上分配的对象。

获取内存 profile 的两种方式：

• HTTP 端点：访问 /debug/pprof/heap 获取当前堆快照。
• 代码采集：调用 pprof.WriteHeapProfile(f) 写入文件。

import (
    "os"
    "runtime/pprof"
)

func main() {
    f, _ := os.Create("mem.pprof")
    defer f.Close()

    // 执行业务逻辑
    allocateMemory()

    // 写入堆 profile
    pprof.WriteHeapProfile(f)
}

3. 阻塞采样

阻塞采样用于分析 goroutine 在 同步原语 上的等待时间。当 goroutine 因 channel 发送/接收、mutex 加锁、select 等待等操作被挂起时，运行时记录阻塞开始和结束的时间戳，以及阻塞发生的堆栈。阻塞采样默认关闭，需要显式设置采样率：

import (
    "os"
    "runtime"
    "runtime/pprof"
)

func main() {
    runtime.SetBlockProfileRate(1) // 记录所有阻塞事件

    f, _ := os.Create("block.pprof")
    defer f.Close()

    // 执行业务逻辑
    doBlockingWork()

    pprof.Lookup("block").WriteTo(f, 0)
}

阻塞采样的关键参数 SetBlockProfileRate：

下图对比了三种采样方式在 Go 运行时中的触发路径：

采样方式选择建议

在实际调优中，应根据问题表象选择合适的采样方式：

• CPU 使用率飙升 → 优先使用 CPU 采样，定位热点函数。
• 内存持续增长或 OOM → 使用内存采样，检查堆分配和存活对象。
• 接口延迟高、吞吐低 → 结合 CPU 采样和阻塞采样，判断是计算瓶颈还是同步等待瓶颈。
• goroutine 数量异常 → 使用 goroutine profile（非本节重点，后续章节展开）。