Skip to content

Commit 3d1a08e

Browse files
committed
feat(tracing): register W3C TraceContext propagator in agent and operator
1 parent f0c385b commit 3d1a08e

2 files changed

Lines changed: 14 additions & 0 deletions

File tree

cmd/agent/main.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@ import (
1818
"sigs.k8s.io/controller-runtime/pkg/log/zap"
1919

2020
impdevv1alpha1 "github.com/syscode-labs/imp/api/v1alpha1"
21+
"go.opentelemetry.io/otel"
22+
"go.opentelemetry.io/otel/propagation"
23+
2124
"github.com/syscode-labs/imp/internal/agent"
2225
"github.com/syscode-labs/imp/internal/agent/network"
2326
"github.com/syscode-labs/imp/internal/telemetry"
@@ -78,6 +81,10 @@ func main() {
7881
os.Exit(1)
7982
}
8083
defer func() { _ = shutdownTraces(context.Background()) }()
84+
otel.SetTextMapPropagator(propagation.NewCompositeTextMapPropagator(
85+
propagation.TraceContext{},
86+
propagation.Baggage{},
87+
))
8188
mc := agent.NewVMMetricsCollector(mp.Meter("imp.agent"), agentReg)
8289

8390
// IMP_STUB_DRIVER=true: StubDriver (CI, test clusters, no KVM needed).

cmd/operator/main.go

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@ import (
2626

2727
ctrlmetrics "sigs.k8s.io/controller-runtime/pkg/metrics"
2828

29+
"go.opentelemetry.io/otel"
30+
"go.opentelemetry.io/otel/propagation"
31+
2932
impv1alpha1 "github.com/syscode-labs/imp/api/v1alpha1"
3033
"github.com/syscode-labs/imp/internal/cnidetect"
3134
"github.com/syscode-labs/imp/internal/controller"
@@ -139,6 +142,10 @@ func main() {
139142
os.Exit(1)
140143
}
141144
defer func() { _ = shutdownTraces(context.Background()) }()
145+
otel.SetTextMapPropagator(propagation.NewCompositeTextMapPropagator(
146+
propagation.TraceContext{},
147+
propagation.Baggage{},
148+
))
142149
controller.InitMetrics(mp.Meter("imp.controller"))
143150

144151
cfg := ctrl.GetConfigOrDie()

0 commit comments

Comments
 (0)