chore: add otel headers config to support axiom (#724)

Co-authored-by: ben <ben@prologe.io>
un · Aug 27, 2024 · 038d143 · 038d143
1 parent fb32c17
commit 038d143
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 3 deletions.
diff --git a/.env.local.example b/.env.local.example
@@ -105,8 +105,10 @@ WORKER_ACCESS_KEY=secretsecretsecretsecretsecretsecret
 ######################### OpenTelemetry Variables ################################
 OTEL_ENABLED=false
 OTEL_EXPORTER_TRACES_ENDPOINT=
-OTEL_EXPORTER_METRICS_ENDPOINT=
+OTEL_EXPORTER_TRACES_HEADERS=
 OTEL_EXPORTER_LOGS_ENDPOINT=
+OTEL_EXPORTER_LOGS_HEADERS=
+OTEL_EXPORTER_METRICS_ENDPOINT=
 
 
 ########################## POSTHOG VARIABLES ################################

diff --git a/packages/otel/env.ts b/packages/otel/env.ts
@@ -1,14 +1,29 @@
 import { createEnv } from '@t3-oss/env-core';
 import { z } from 'zod';
 
+const stringToJSON = z.string().transform((str, ctx) => {
+  try {
+    return JSON.parse(str) as unknown;
+  } catch (e) {
+    ctx.addIssue({ code: 'custom', message: 'Invalid JSON' });
+    return z.NEVER;
+  }
+});
+
 export const env = createEnv({
   server: {
     OTEL_ENABLED: z
       .string()
       .optional()
       .transform((v) => v === 'true'),
     OTEL_EXPORTER_TRACES_ENDPOINT: z.string().url().optional(),
+    OTEL_EXPORTER_TRACES_HEADERS: stringToJSON
+      .pipe(z.record(z.string()))
+      .optional(),
     OTEL_EXPORTER_LOGS_ENDPOINT: z.string().url().optional(),
+    OTEL_EXPORTER_LOGS_HEADERS: stringToJSON
+      .pipe(z.record(z.string()))
+      .optional(),
     OTEL_EXPORTER_METRICS_ENDPOINT: z.string().url().optional(),
     NODE_ENV: z.enum(['development', 'production']).default('development')
   },

diff --git a/packages/otel/setup.ts b/packages/otel/setup.ts
@@ -32,7 +32,8 @@ export const setupOpentelemetry = ({
   traceProvider.addSpanProcessor(
     new BatchSpanProcessor(
       new OTLPTraceExporter({
-        url: env.OTEL_EXPORTER_TRACES_ENDPOINT
+        url: env.OTEL_EXPORTER_TRACES_ENDPOINT,
+        headers: env.OTEL_EXPORTER_TRACES_HEADERS
       })
     )
   );
@@ -42,7 +43,8 @@ export const setupOpentelemetry = ({
   loggerProvider.addLogRecordProcessor(
     new BatchLogRecordProcessor(
       new OTLPLogExporter({
-        url: env.OTEL_EXPORTER_LOGS_ENDPOINT
+        url: env.OTEL_EXPORTER_LOGS_ENDPOINT,
+        headers: env.OTEL_EXPORTER_LOGS_HEADERS
       })
     )
   );