Function collectPprof

support/support.go:1168–1225

(ctx context.Context, d *Deps, b *Bundle)

}

func collectPprof(ctx context.Context, d *Deps, b *Bundle) Pprof {
	var pprof Pprof

	// Check server version before attempting pprof collection.
	if b.Deployment.BuildInfo == nil {
		d.Log.Warn(ctx, "skipping pprof collection: build info not available")
		return pprof
	}
	if !VersionSupportsPprof(b.Deployment.BuildInfo.Version) {
		d.Log.Warn(ctx, "skipping pprof collection: server version too old",
			slog.F("version", b.Deployment.BuildInfo.Version),
			slog.F("min_version", minPprofVersion))
		return pprof
	}

	// Try the consolidated /debug/profile endpoint first. It
	// temporarily enables block/mutex profiling on the server and
	// returns a single tar.gz archive.
	serverPprof, err := PprofInfoFromArchive(ctx, d.Client, d.Log, 30*time.Second)
	if err != nil {
		d.Log.Warn(ctx, "consolidated profile endpoint unavailable, falling back to individual endpoints",
			slog.Error(err))
		// Fall back to the legacy per-profile endpoint approach.
		serverPprof = PprofInfo(ctx, d.Client, d.Log)
	}
	if serverPprof != nil {
		pprof.Server = serverPprof
	}

	if d.AgentID != uuid.Nil {
		conn, err := workspacesdk.New(d.Client).
			DialAgent(ctx, d.AgentID, &workspacesdk.DialAgentOptions{
				Logger:         d.Log.Named("dial-agent-pprof"),
				BlockEndpoints: false,
			})
		if err != nil {
			d.Log.Warn(ctx, "failed to dial agent for pprof collection", slog.Error(err))
		} else {
			defer func() {
				if err := conn.Close(); err != nil {
					d.Log.Error(ctx, "failed to close agent pprof connection", slog.Error(err))
				}
				<-conn.TailnetConn().Closed()
			}()

			if conn.AwaitReachable(ctx) {
				agentPprof := PprofInfoFromAgent(ctx, conn, d.Log)
				if agentPprof != nil {
					pprof.Agent = agentPprof
				}
			} else {
				d.Log.Warn(ctx, "agent not reachable for pprof collection")
			}
		}
	}

	return pprof
}