feat: use separate Axiom datasets for traces/logs and metrics

Traces and logs go to AXIOM_TRACES_DATASET (events), metrics go
to AXIOM_METRICS_DATASET (metrics). Replaces single AXIOM_DATASET.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

Author: AchoArnold

api/.env.docker

@@ -61,8 +61,10 @@ UPTRACE_DSN=
 # Axiom observability configuration
 # API token for Axiom (required for logging, traces, and metrics in production)
 AXIOM_TOKEN=
-# Dataset name in Axiom where logs, traces, and metrics are sent
-AXIOM_DATASET=
+# Dataset for logs and traces (e.g. "events")
+AXIOM_TRACES_DATASET=
+# Dataset for metrics (e.g. "metrics")
+AXIOM_METRICS_DATASET=
 
 
 # [optional] Websocket configuration for https://pusher.com if you will like to frontend to update in real time

api/pkg/di/container.go

@@ -1847,14 +1847,14 @@ func (container *Container) initializeGoogleTraceProvider(version string, namesp
 func (container *Container) initializeAxiomTraceProvider(version string, namespace string) func() {
 	container.logger.Debug("initializing axiom trace provider")
 
-	headers := map[string]string{
+	traceHeaders := map[string]string{
 		"Authorization":   "Bearer " + os.Getenv("AXIOM_TOKEN"),
-		"X-Axiom-Dataset": os.Getenv("AXIOM_DATASET"),
+		"X-Axiom-Dataset": os.Getenv("AXIOM_TRACES_DATASET"),
 	}
 
 	traceExporter, err := otlptracehttp.New(context.Background(),
 		otlptracehttp.WithEndpoint("us-east-1.aws.edge.axiom.co"),
-		otlptracehttp.WithHeaders(headers),
+		otlptracehttp.WithHeaders(traceHeaders),
 	)
 	if err != nil {
 		container.logger.Fatal(stacktrace.Propagate(err, "cannot create axiom OTLP trace exporter"))
@@ -1872,9 +1872,14 @@ func (container *Container) initializeAxiomTraceProvider(version string, namespa
 		propagation.Baggage{},
 	))
 
+	metricHeaders := map[string]string{
+		"Authorization":   "Bearer " + os.Getenv("AXIOM_TOKEN"),
+		"X-Axiom-Dataset": os.Getenv("AXIOM_METRICS_DATASET"),
+	}
+
 	metricExporter, err := otlpmetrichttp.New(context.Background(),
 		otlpmetrichttp.WithEndpoint("us-east-1.aws.edge.axiom.co"),
-		otlpmetrichttp.WithHeaders(headers),
+		otlpmetrichttp.WithHeaders(metricHeaders),
 	)
 	if err != nil {
 		container.logger.Fatal(stacktrace.Propagate(err, "cannot create axiom OTLP metric exporter"))
@@ -1969,7 +1974,7 @@ func axiomLogger(skipFrameCount int) *zerodriver.Logger {
 	zerolog.TimeFieldFormat = time.RFC3339Nano
 
 	axiomWriter, err := axiomzerolog.New(
-		axiomzerolog.SetDataset(os.Getenv("AXIOM_DATASET")),
+		axiomzerolog.SetDataset(os.Getenv("AXIOM_TRACES_DATASET")),
 	)
 	if err != nil {
 		// Fall back to stderr JSON if Axiom is not configured