coder/coderd/pproflabel/pproflabel.go

package pproflabel

import (
	"context"
	"runtime/pprof"
)

// Go is just a convince wrapper to set off a labeled goroutine.
func Go(ctx context.Context, labels pprof.LabelSet, f func(context.Context)) {
	go pprof.Do(ctx, labels, f)
}

func Do(ctx context.Context, labels pprof.LabelSet, f func(context.Context)) {
	pprof.Do(ctx, labels, f)
}

const (
	// ServiceTag should not collide with the pyroscope built-in tag "service".
	// Use `coder_` to avoid collisions.
	ServiceTag = "coder_service"

	ServiceHTTPServer           = "http-api"
	ServiceLifecycles           = "lifecycle-executor"
	ServicePrebuildReconciler   = "prebuilds-reconciler"
	ServiceTerraformProvisioner = "terraform-provisioner"
	ServiceDBPurge              = "db-purge"
	ServiceNotifications        = "notifications"
	ServiceReplicaSync          = "replica-sync"
	// ServiceMetricCollector collects metrics from insights in the database and
	// exports them in a prometheus collector format.
	ServiceMetricCollector = "metrics-collector"
	// ServiceAgentMetricAggregator merges agent metrics and exports them in a
	// prometheus collector format.
	ServiceAgentMetricAggregator = "agent-metrics-aggregator"
	// ServiceTallymanPublisher publishes usage events to coder/tallyman.
	ServiceTallymanPublisher = "tallyman-publisher"
	ServiceUsageEventCron    = "usage-event-cron"
	// ServiceWorkspaceBuildOrchestrator fulfills workspace build
	// orchestrations once their parent build reaches a terminal state.
	ServiceWorkspaceBuildOrchestrator = "workspace-build-orchestrator"
	ServiceUsageEventGenerator        = "usage-event-generator"

	RequestTypeTag = "coder_request_type"
)

func Service(name string, pairs ...string) pprof.LabelSet {
	return pprof.Labels(append([]string{ServiceTag, name}, pairs...)...)
}