Add otel headers to health checks

Author: beefancohen

packages/node-opentelemetry/src/otel.ts

@@ -146,6 +146,41 @@ const hrtimeToMs = (hrtime: [number, number]) => {
   return hrtime[0] * 1e3 + hrtime[1] / 1e6;
 };
 
+/**
+ * Parses OTEL_EXPORTER_OTLP_HEADERS environment variable into a structured headers object.
+ * Format: "key1=value1,key2=value2" -> { key1: "value1", key2: "value2" }
+ */
+const parseOtlpHeaders = (headersString?: string): Record<string, string> => {
+  if (!headersString) {
+    return {};
+  }
+
+  const headers: Record<string, string> = {};
+  const pairs = headersString.split(',');
+
+  for (const pair of pairs) {
+    const trimmedPair = pair.trim();
+    if (!trimmedPair) {
+      continue;
+    }
+
+    const equalIndex = trimmedPair.indexOf('=');
+    if (equalIndex === -1) {
+      // Skip malformed pairs without '='
+      continue;
+    }
+
+    const key = trimmedPair.substring(0, equalIndex).trim();
+    const value = trimmedPair.substring(equalIndex + 1).trim();
+
+    if (key) {
+      headers[key] = value;
+    }
+  }
+
+  return headers;
+};
+
 const healthCheckUrl = async (
   ui: ora.Ora,
   url: string,
@@ -215,6 +250,13 @@ export const initSDK = (config: SDKConfig) => {
   });
   ui.succeed('Set default otel envs');
 
+  // Parse OTLP headers from environment variable
+  const otlpHeaders = parseOtlpHeaders(env.OTEL_EXPORTER_OTLP_HEADERS);
+  const healthCheckHeaders = {
+    'Content-Type': 'application/json',
+    ...otlpHeaders,
+  };
+
   const stopOnTerminationSignals =
     config.stopOnTerminationSignals ??
     DEFAULT_HDX_NODE_STOP_ON_TERMINATION_SIGNALS; // Stop by default
@@ -244,23 +286,17 @@ export const initSDK = (config: SDKConfig) => {
   Promise.all([
     healthCheckUrl(ui, DEFAULT_OTEL_TRACES_EXPORTER_URL, {
       method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-      },
+      headers: healthCheckHeaders,
       body: JSON.stringify({}),
     }),
     healthCheckUrl(ui, _logger.getExporterUrl(), {
       method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-      },
+      headers: healthCheckHeaders,
       body: JSON.stringify({}),
     }),
     healthCheckUrl(ui, DEFAULT_OTEL_METRICS_EXPORTER_URL, {
       method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-      },
+      headers: healthCheckHeaders,
       body: JSON.stringify({}),
     }),
   ]);