Method initializeAxiomTraceProvider

api/pkg/di/container.go:1837–1894

(version string, namespace string)

}

func (container *Container) initializeAxiomTraceProvider(version string, namespace string) func() {
	container.logger.Debug("initializing axiom trace provider")

	traceHeaders := map[string]string{
		"Authorization":   "Bearer " + os.Getenv("AXIOM_TOKEN"),
		"X-Axiom-Dataset": os.Getenv("AXIOM_DATASET_EVENTS"),
	}

	traceExporter, err := otlptracehttp.New(
		context.Background(),
		otlptracehttp.WithEndpoint("us-east-1.aws.edge.axiom.co"),
		otlptracehttp.WithHeaders(traceHeaders),
	)
	if err != nil {
		container.logger.Fatal(stacktrace.Propagate(err, "cannot create axiom OTLP trace exporter"))
	}

	tp := trace.NewTracerProvider(
		trace.WithBatcher(traceExporter),
		trace.WithSampler(trace.AlwaysSample()),
		trace.WithResource(container.OtelResources(version, namespace)),
	)
	otel.SetTracerProvider(tp)

	otel.SetTextMapPropagator(propagation.NewCompositeTextMapPropagator(
		propagation.TraceContext{},
		propagation.Baggage{},
	))

	metricHeaders := map[string]string{
		"Authorization":   "Bearer " + os.Getenv("AXIOM_TOKEN"),
		"X-Axiom-Dataset": os.Getenv("AXIOM_DATASET_METRICS"),
	}

	metricExporter, err := otlpmetrichttp.New(
		context.Background(),
		otlpmetrichttp.WithEndpoint("us-east-1.aws.edge.axiom.co"),
		otlpmetrichttp.WithHeaders(metricHeaders),
	)
	if err != nil {
		container.logger.Fatal(stacktrace.Propagate(err, "cannot create axiom OTLP metric exporter"))
	}

	meterProvider := metric.NewMeterProvider(
		metric.WithReader(metric.NewPeriodicReader(metricExporter)),
		metric.WithResource(container.OtelResources(version, namespace)),
	)
	otel.SetMeterProvider(meterProvider)

	return func() {
		if err := tp.Shutdown(context.Background()); err != nil {
			container.logger.Error(stacktrace.Propagate(err, "cannot shutdown axiom trace provider"))
		}
		if err := meterProvider.Shutdown(context.Background()); err != nil {
			container.logger.Error(stacktrace.Propagate(err, "cannot shutdown axiom meter provider"))
		}
	}
}