pprof 标签在新 goroutine 中丢失的传递处理

pprof 标签在新 goroutine 中“丢失”，通常不是 context.Context 被清空，而是把两条传播链路混在了一起：pprof.WithLabels 只返回带标签的新 context；新 goroutine 默认继承的是创建者 goroutine 当时的 pprof 标签。要让标签稳定进入子任务，优先用 pprof.Do 包住创建边界，或在 goroutine 入口显式调用 pprof.SetGoroutineLabels。

先分清 context 标签和 goroutine 标签

可以把标签看成两个相邻但不相同的域。WithLabels 把 LabelSet 合并进一个新的 context，业务函数拿到这个 context 后可以读取标签；而 profile 关联的 goroutine 标签来自运行时当前 goroutine。直接把带标签的 context 传给 go worker(ctx)，并不会自动把 context 内的标签写入新 goroutine。

用 Do 或 SetGoroutineLabels 固定传递边界

如果子 goroutine 就在当前调用的创建边界内产生，推荐使用 pprof.Do。它会在回调期间设置当前 goroutine 的标签，回调结束后恢复原状态，因此不会把一次请求的标签永久留在复用的 goroutine 上。

package main

import (
    "context"
    "runtime/pprof"
)

func startTask(ctx context.Context, jobID string) {
    labels := pprof.Labels("job", jobID)
    // Do 临时设置当前 goroutine；回调内新建的 goroutine 继承这组标签。
    pprof.Do(ctx, labels, func(ctx context.Context) {
        go worker(ctx)
    })
}

func worker(ctx context.Context) {
    // context 仍要显式传入，业务取消与 pprof 标签是两条不同链路。
    job, ok := pprof.Label(ctx, "job")
    if !ok || job == "" {
        return // 缺少标签时避免把未归属任务写入 profile。
    }
    _ = job
}

如果 goroutine 由通用 worker、回调或第三方调度器创建，无法把创建动作放进 Do，可以在入口使用较底层的 API：

func workerEntry(ctx context.Context) {
    // 入口显式同步运行时标签，适合无法控制 goroutine 创建点的场景。
    pprof.SetGoroutineLabels(ctx)

    // ForLabels 用来确认关键键是否仍在当前 context 中。
    found := false
    pprof.ForLabels(ctx, func(key, value string) bool {
        if key == "job" && value != "" {
            found = true
        }
        return true // 继续检查其他标签，便于记录完整上下文。
    })
    if !found {
        return // 没有任务标签就不进入需要归因的工作段。
    }
}

排查标签消失时先看这张清单

第一，确认创建 goroutine 的那一刻，父 goroutine 是否已经处于目标标签环境；标签是在创建点继承，不是等 worker 真正开始执行时才回溯。第二，确认代码是否只调用了 WithLabels，却没有 Do 或 SetGoroutineLabels。第三，检查同一个 key 是否被后写入的值覆盖，因为 Labels 和 WithLabels 对同名 key 都采用覆盖语义。

还要区分“读取不到 context 标签”和“profile 里看不到标签”：前者用 pprof.Label、ForLabels 定位，后者要看 goroutine 创建边界、采样类型和 profile 输出是否支持该标签。不要用一张空 profile 直接证明传播失败。

相关问题

为什么把 context 传进 goroutine 仍然看不到 pprof 标签？

因为 context 是参数对象，pprof 的运行时标签是当前 goroutine 状态；传参不会自动执行 SetGoroutineLabels。

Do 和 SetGoroutineLabels 该怎么选？

能控制创建边界时优先 Do，它有回调范围并能恢复旧状态；只能控制 worker 入口时再用 SetGoroutineLabels。

子 goroutine 会继承后来修改的父标签吗？

不会。继承发生在新 goroutine 创建时，之后父 goroutine 的标签变化不会回写已经存在的子 goroutine。