Function PprofInfoFromAgent

support/support.go:887–999

(ctx context.Context, conn workspacesdk.AgentConn, log slog.Logger)

}

func PprofInfoFromAgent(ctx context.Context, conn workspacesdk.AgentConn, log slog.Logger) *PprofCollection {
	if conn == nil {
		return nil
	}

	var (
		p  PprofCollection
		eg errgroup.Group
	)

	p.EndpointURL = "agent"
	p.CollectedAt = time.Now()

	// Define agent pprof endpoints - these go through the agent connection
	endpoints := map[string]func([]byte){
		"/debug/pprof/allocs": func(data []byte) {
			p.Allocs = compressData(data)
		},
		"/debug/pprof/heap": func(data []byte) {
			p.Heap = compressData(data)
		},
		"/debug/pprof/profile?seconds=30": func(data []byte) {
			p.Profile = compressData(data)
		},
		"/debug/pprof/block": func(data []byte) {
			p.Block = compressData(data)
		},
		"/debug/pprof/mutex": func(data []byte) {
			p.Mutex = compressData(data)
		},
		"/debug/pprof/goroutine": func(data []byte) {
			p.Goroutine = compressData(data)
		},
		"/debug/pprof/threadcreate": func(data []byte) {
			p.Threadcreate = compressData(data)
		},
		"/debug/pprof/trace?seconds=30": func(data []byte) {
			p.Trace = compressData(data)
		},
		"/debug/pprof/cmdline": func(data []byte) {
			p.Cmdline = string(data)
		},
		"/debug/pprof/symbol": func(data []byte) {
			p.Symbol = string(data)
		},
	}

	// Collect each endpoint in parallel
	for endpoint, setter := range endpoints {
		eg.Go(func() error {
			// Set longer timeout for profile and trace endpoints (they take 30 seconds)
			timeout := 10 * time.Second
			if strings.Contains(endpoint, "seconds=30") {
				timeout = 45 * time.Second
			}

			ctx, cancel := context.WithTimeout(ctx, timeout)
			defer cancel()