feat(node): Populate propagation context using env variables

Author: AbhiPrasad

packages/node/src/sdk.ts

@@ -6,13 +6,16 @@ import {
   initAndBind,
   Integrations as CoreIntegrations,
 } from '@sentry/core';
-import type { SessionStatus, StackParser } from '@sentry/types';
+import type { DynamicSamplingContext, PropagationContext, SessionStatus, StackParser } from '@sentry/types';
 import {
+  baggageHeaderToDynamicSamplingContext,
   createStackParser,
+  extractTraceparentData,
   GLOBAL_OBJ,
   logger,
   nodeStackLineParser,
   stackParserFromStackParserOptions,
+  uuid4,
 } from '@sentry/utils';
 
 import { setNodeAsyncContextStrategy } from './async';
@@ -129,8 +132,9 @@ export function init(options: NodeOptions = {}): void {
     options.dsn = process.env.SENTRY_DSN;
   }
 
-  if (options.tracesSampleRate === undefined && process.env.SENTRY_TRACES_SAMPLE_RATE) {
-    const tracesSampleRate = parseFloat(process.env.SENTRY_TRACES_SAMPLE_RATE);
+  const sentryTracesSampleRate = process.env.SENTRY_TRACES_SAMPLE_RATE;
+  if (options.tracesSampleRate === undefined && sentryTracesSampleRate) {
+    const tracesSampleRate = parseFloat(sentryTracesSampleRate);
     if (isFinite(tracesSampleRate)) {
       options.tracesSampleRate = tracesSampleRate;
     }
@@ -171,6 +175,8 @@ export function init(options: NodeOptions = {}): void {
   if (options.autoSessionTracking) {
     startSessionTracking();
   }
+
+  updateScopeFromEnvVariables();
 }
 
 /**
@@ -285,3 +291,35 @@ function startSessionTracking(): void {
     if (session && !terminalStates.includes(session.status)) hub.endSession();
   });
 }
+
+/**
+ * Update scope and propagation context based on environmental variables.
+ *
+ * See https://github.com/getsentry/rfcs/blob/main/text/0071-continue-trace-over-process-boundaries.md
+ * for more details.
+ */
+function updateScopeFromEnvVariables(): void {
+  const sentryUseEnvironment = process.env.SENTRY_USE_ENVIRONMENT;
+  if (
+    !['false', 'False', 'FALSE', 'n', 'no', 'No', 'NO', 'off', 'Off', 'OFF', '0'].includes(
+      sentryUseEnvironment as string,
+    )
+  ) {
+    const sentryTraceEnv = process.env.SENTRY_TRACE || '';
+    const baggageEnv = process.env.SENTRY_BAGGAGE;
+
+    const traceparentData = extractTraceparentData(sentryTraceEnv) || {};
+    const { traceId, parentSpanId, parentSampled } = traceparentData;
+    const propagationContext: PropagationContext = {
+      traceId: traceId || uuid4(),
+      spanId: uuid4().substring(16),
+      parentSpanId,
+      sampled: parentSampled === undefined ? false : parentSampled,
+    };
+    const dynamicSamplingContext = baggageHeaderToDynamicSamplingContext(baggageEnv);
+    if (dynamicSamplingContext) {
+      propagationContext.dsc = dynamicSamplingContext as DynamicSamplingContext;
+    }
+    getCurrentHub().getScope().setPropagationContext(propagationContext);
+  }
+}

packages/node/test/index.test.ts

@@ -488,4 +488,47 @@ describe('SentryNode initialization', () => {
       expect(instrumenter).toEqual('otel');
     });
   });
+
+  describe('propagation context', () => {
+    beforeEach(() => {
+      process.env.SENTRY_TRACE = '12312012123120121231201212312012-1121201211212012-0';
+      process.env.SENTRY_BAGGAGE = 'sentry-release=1.0.0,sentry-environment=production';
+
+      getCurrentHub().getScope().clear();
+    });
+
+    afterEach(() => {
+      delete process.env.SENTRY_TRACE;
+      delete process.env.SENTRY_BAGGAGE;
+    });
+
+    it('reads from environmental variables', () => {
+      init({ dsn });
+
+      // @ts-expect-error accessing private method for test
+      expect(getCurrentHub().getScope()._propagationContext).toEqual({
+        traceId: '12312012123120121231201212312012',
+        parentSpanId: '1121201211212012',
+        spanId: expect.any(String),
+        sampled: false,
+        dsc: {
+          release: '1.0.0',
+          environment: 'production',
+        },
+      });
+    });
+
+    it.each(['false', 'False', 'FALSE', 'n', 'no', 'No', 'NO', 'off', 'Off', 'OFF', '0'])(
+      'does not read from environmental variable if SENTRY_USE_ENVIRONMENT is set to %s',
+      useEnvValue => {
+        process.env.SENTRY_USE_ENVIRONMENT = useEnvValue;
+        init({ dsn });
+
+        // @ts-expect-error accessing private method for test
+        expect(getCurrentHub().getScope()._propagationContext.traceId).not.toEqual('12312012123120121231201212312012');
+
+        delete process.env.SENTRY_USE_ENVIRONMENT;
+      },
+    );
+  });
 });