chore(aws): Remove manual span creation

Author: msonnb

dev-packages/e2e-tests/test-applications/aws-serverless-esm/tests/basic.test.ts

@@ -43,7 +43,7 @@ test('AWS Serverless SDK sends events in ESM mode', async ({ request }) => {
     trace_id: expect.stringMatching(/[a-f0-9]{32}/),
   });
 
-  expect(transactionEvent.spans).toHaveLength(3);
+  expect(transactionEvent.spans).toHaveLength(2);
 
   // shows that the Otel Http instrumentation is working
   expect(transactionEvent.spans).toContainEqual(
@@ -58,19 +58,6 @@ test('AWS Serverless SDK sends events in ESM mode', async ({ request }) => {
     }),
   );
 
-  expect(transactionEvent.spans).toContainEqual(
-    expect.objectContaining({
-      data: {
-        'sentry.op': 'function.aws.lambda',
-        'sentry.origin': 'auto.function.serverless',
-        'sentry.source': 'component',
-      },
-      description: 'my-lambda',
-      op: 'function.aws.lambda',
-      origin: 'auto.function.serverless',
-    }),
-  );
-
   // shows that the manual span creation is working
   expect(transactionEvent.spans).toContainEqual(
     expect.objectContaining({

packages/aws-serverless/src/sdk.ts

@@ -1,33 +1,24 @@
-import type { Integration, Options, Scope, Span } from '@sentry/core';
-import {
-  applySdkMetadata,
-  debug,
-  getSDKSource,
-  SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN,
-  SEMANTIC_ATTRIBUTE_SENTRY_SOURCE,
-} from '@sentry/core';
+import type { Integration, Options, Scope } from '@sentry/core';
+import { applySdkMetadata, debug, getSDKSource } from '@sentry/core';
 import type { NodeClient, NodeOptions } from '@sentry/node';
 import {
   captureException,
   captureMessage,
-  continueTrace,
   flush,
   getCurrentScope,
   getDefaultIntegrationsWithoutPerformance,
   initWithoutDefaultIntegrations,
-  startSpanManual,
   withScope,
 } from '@sentry/node';
 import type { Context, Handler } from 'aws-lambda';
 import { existsSync } from 'fs';
-import { hostname } from 'os';
 import { basename, resolve } from 'path';
 import { performance } from 'perf_hooks';
 import { types } from 'util';
 import { DEBUG_BUILD } from './debug-build';
 import { awsIntegration } from './integration/aws';
 import { awsLambdaIntegration } from './integration/awslambda';
-import { getAwsTraceData, markEventUnhandled } from './utils';
+import { markEventUnhandled } from './utils';
 
 const { isPromise } = types;
 
@@ -54,12 +45,6 @@ export interface WrapperOptions {
    * @default false
    */
   captureAllSettledReasons: boolean;
-  /**
-   * Automatically trace all handler invocations.
-   * You may want to disable this if you use express within Lambda.
-   * @default true
-   */
-  startTrace: boolean;
 }
 
 /**
@@ -207,18 +192,6 @@ function enhanceScopeWithEnvironmentData(scope: Scope, context: Context, startTi
   });
 }
 
-/**
- * Adds additional transaction-related information from the environment and AWS Context to the Sentry Scope.
- *
- * @param scope Scope that should be enhanced
- * @param context AWS Lambda context that will be used to extract some part of the data
- */
-function enhanceScopeWithTransactionData(scope: Scope, context: Context): void {
-  scope.setTransactionName(context.functionName);
-  scope.setTag('server_name', process.env._AWS_XRAY_DAEMON_ADDRESS || process.env.SENTRY_NAME || hostname());
-  scope.setTag('url', `awslambda:///${context.functionName}`);
-}
-
 /**
  * Wraps a lambda handler adding it error capture and tracing capabilities.
  *
@@ -237,7 +210,6 @@ export function wrapHandler<TEvent, TResult>(
     captureTimeoutWarning: true,
     timeoutWarningLimit: 500,
     captureAllSettledReasons: false,
-    startTrace: true,
     ...wrapOptions,
   };
   let timeoutWarningTimer: NodeJS.Timeout;
@@ -293,7 +265,7 @@ export function wrapHandler<TEvent, TResult>(
       }, timeoutWarningDelay) as unknown as NodeJS.Timeout;
     }
 
-    async function processResult(span?: Span): Promise<TResult> {
+    async function processResult(): Promise<TResult> {
       const scope = getCurrentScope();
 
       let rv: TResult;
@@ -314,60 +286,15 @@ export function wrapHandler<TEvent, TResult>(
         throw e;
       } finally {
         clearTimeout(timeoutWarningTimer);
-        if (span?.isRecording()) {
-          span.end();
-        }
         await flush(options.flushTimeout).catch(e => {
           DEBUG_BUILD && debug.error(e);
         });
       }
       return rv;
     }
 
-    // Only start a trace and root span if the handler is not already wrapped by Otel instrumentation
-    // Otherwise, we create two root spans (one from otel, one from our wrapper).
-    // If Otel instrumentation didn't work or was filtered by users, we still want to trace the handler.
-    // TODO: Since bumping the OTEL Instrumentation, this is likely not needed anymore, we can possibly remove this (can be done whenever since it would be non-breaking)
-    if (options.startTrace && !isWrappedByOtel(handler)) {
-      const traceData = getAwsTraceData(event as { headers?: Record<string, string> }, context);
-
-      return continueTrace({ sentryTrace: traceData['sentry-trace'], baggage: traceData.baggage }, () => {
-        return startSpanManual(
-          {
-            name: context.functionName,
-            op: 'function.aws.lambda',
-            attributes: {
-              [SEMANTIC_ATTRIBUTE_SENTRY_SOURCE]: 'component',
-              [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: 'auto.function.serverless',
-            },
-          },
-          span => {
-            enhanceScopeWithTransactionData(getCurrentScope(), context);
-
-            return processResult(span);
-          },
-        );
-      });
-    }
-
     return withScope(async () => {
-      return processResult(undefined);
+      return processResult();
     });
   };
 }
-
-/**
- * Checks if Otel's AWSLambda instrumentation successfully wrapped the handler.
- * Check taken from @opentelemetry/core
- */
-function isWrappedByOtel(
-  // eslint-disable-next-line @typescript-eslint/ban-types
-  handler: Function & { __original?: unknown; __unwrap?: unknown; __wrapped?: boolean },
-): boolean {
-  return (
-    typeof handler === 'function' &&
-    typeof handler.__original === 'function' &&
-    typeof handler.__unwrap === 'function' &&
-    handler.__wrapped === true
-  );
-}