feat: Add createTraceSpan support

Author: Mmarzex

node/packages/aws-lambda-sdk/package.json

@@ -4,7 +4,7 @@
   "version": "0.15.10",
   "author": "Serverless, Inc.",
   "dependencies": {
-    "@serverless/sdk": "^0.5.21",
+    "@serverless/sdk": "^0.5.23",
     "@serverless/sdk-schema": "^0.15.6",
     "d": "^1.0.1",
     "ext": "^1.7.0",

node/packages/aws-lambda-sdk/test/fixtures/lambdas/sdk-create-trace-span.js

@@ -0,0 +1,27 @@
+'use strict';
+
+const sdk = require('@serverless/sdk');
+
+module.exports.handler = async (event) => {
+  if (!sdk) throw new Error('SDK not exported');
+
+  if (!event.isTriggeredByUnitTest) {
+    // eslint-disable-next-line import/no-unresolved
+    const sdkMirror = require('@serverless/aws-lambda-sdk');
+    if (sdk !== sdkMirror) throw new Error('SDK exports mismatch');
+  }
+
+  const span = sdk.createTraceSpan('user.parent');
+
+  sdk.createTraceSpan('user.child.one', () => {});
+
+  await sdk.createTraceSpan('user.child.two', async () => {});
+
+  span.close();
+
+  return {
+    name: sdk.name,
+    version: sdk.version,
+    rootSpanName: sdk.traceSpans.root.name,
+  };
+};

node/packages/aws-lambda-sdk/test/integration/index.test.js

@@ -482,6 +482,23 @@ describe('integration', function () {
     },
   };
 
+  const sdkCreateTraceSpanTestConfig = {
+    isCustomResponse: true,
+    test: ({ invocationsData }) => {
+      for (const [, { trace }] of invocationsData.entries()) {
+        const { spans } = trace;
+        const parentSpan = spans.find(({ name }) => name === 'user.parent');
+        const childOneSpan = spans.find(({ name }) => name === 'user.child.one');
+        const childTwoSpan = spans.find(({ name }) => name === 'user.child.two');
+        expect(parentSpan).to.exist;
+        expect(childOneSpan).to.exist;
+        expect(childTwoSpan).to.exist;
+        expect(childOneSpan.parentSpanId.toString()).to.equal(parentSpan.id.toString());
+        expect(childTwoSpan.parentSpanId.toString()).to.equal(parentSpan.id.toString());
+      }
+    },
+  };
+
   const sdkTestConfig = {
     isCustomResponse: true,
     capturedEvents: [
@@ -2638,6 +2655,17 @@ describe('integration', function () {
         config: sdkTestConfig,
       },
     ],
+    [
+      'sdk-create-trace-span',
+      {
+        variants: new Map([
+          ['v14', { configuration: { Runtime: 'nodejs14.x' } }],
+          ['v16', { configuration: { Runtime: 'nodejs16.x' } }],
+          ['v18', { configuration: { Runtime: 'nodejs18.x' } }],
+        ]),
+        config: sdkCreateTraceSpanTestConfig,
+      },
+    ],
     [
       'structured-logging-events',
       {