Inject appsync trace

thundra-io · Nov 25, 2021 · de17311 · de17311
1 parent 36d1615
commit de17311
Show file tree

Hide file tree

Showing 4 changed files with 45 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -172,6 +172,12 @@ exports.handler = thundra((event, context) => {
 });
 ```
 
+### NOTES
+
+- In order to activate *AWS Step Functions* trace, `THUNDRA_AGENT_LAMBDA_AWS_STEPFUNCTIONS` environment variable should be set `true`.
+- In order to activate *AWS AppSync* trace, `THUNDRA_AGENT_LAMBDA_AWS_APPSYNC` environment variable should be set `true`.
+- For other integrations' configuration, please take a look environment variables table at the end.
+
 ## Frameworks
 
 The following frameworks are supported by Thundra:
@@ -332,6 +338,7 @@ Check out [this part](https://thundra.readme.io/docs/how-to-warmup) in our docs
 | THUNDRA_AGENT_LAMBDA_TRACE_FIREHOSE_REQUEST_ENABLE                  |  bool  |              false              |
 | THUNDRA_AGENT_LAMBDA_TRACE_CLOUDWATCHLOG_REQUEST_ENABLE             |  bool  |              false              |
 | THUNDRA_AGENT_LAMBDA_AWS_STEPFUNCTIONS                              |  bool  |              false              |
+| THUNDRA_AGENT_LAMBDA_AWS_APPSYNC                                    |  bool  |              false              |
 | THUNDRA_AGENT_APPLICATION_ID                                        | string |                -                |
 | THUNDRA_AGENT_APPLICATION_INSTANCEID                                | string |                -                |
 | THUNDRA_AGENT_APPLICATION_REGION                                    | string |                -                |

diff --git a/src/config/ConfigMetadata.ts b/src/config/ConfigMetadata.ts
@@ -189,6 +189,10 @@ export const ConfigMetadata: {[key: string]: { type: string, defaultValue?: any
         type: 'boolean',
         defaultValue: false,
     },
+    [ConfigNames.THUNDRA_LAMBDA_AWS_APPSYNC]: {
+        type: 'boolean',
+        defaultValue: false,
+    },
     [ConfigNames.THUNDRA_TRACE_INTEGRATIONS_HTTP_BODY_MASK]: {
         type: 'boolean',
         defaultValue: false,

diff --git a/src/config/ConfigNames.ts b/src/config/ConfigNames.ts
@@ -80,6 +80,8 @@ class ConfigNames {
         'thundra.agent.lambda.trace.cloudwatchlog.request.enable';
     public static readonly THUNDRA_LAMBDA_AWS_STEPFUNCTIONS: string =
         'thundra.agent.lambda.aws.stepfunctions';
+        public static readonly THUNDRA_LAMBDA_AWS_APPSYNC: string =
+            'thundra.agent.lambda.aws.appsync';
 
     /////////////////////////////////////////////////////////////////////////////
 

diff --git a/src/wrappers/lambda/LambdaExecutor.ts b/src/wrappers/lambda/LambdaExecutor.ts
@@ -248,6 +248,8 @@ export function finishInvocation(pluginContext: PluginContext, execContext: Exec
 
     // Inject step functions trace links
     injectStepFunctionInfo(originalEvent, response, execContext);
+    // Inject appsync trace id
+    injectAppsyncInfo(originalEvent, response, execContext);
 
     invocationData.finishTimestamp = finishTimestamp;
     invocationData.duration = finishTimestamp - startTimestamp;
@@ -333,6 +335,36 @@ function injectTriggerTags(span: ThundraSpan, pluginContext: PluginContext, exec
     }
 }
 
+function injectAppsyncInfo(request: any, response: any, execContext: ExecutionContext): any {
+    try {
+        const isAppsync = ConfigProvider.get<boolean>(ConfigNames.THUNDRA_LAMBDA_AWS_APPSYNC);
+        ThundraLogger.debug(
+            '<LambdaExecutor> Checked whether AWS Appsync support is enabled for transaction',
+            execContext.transactionId, ':', isAppsync);
+        if (isAppsync) {
+            const traceId = execContext.traceId;
+            if (typeof response === 'object' && response !== null) {
+                const thundraTraceKey = {
+                    trace_id: traceId,
+                };
+                ThundraLogger.debug(
+                    '<LambdaExecutor> Injected Thundra AppSync trace key for transaction',
+                    execContext.transactionId, ':', thundraTraceKey);
+                response[THUNDRA_TRACE_KEY] = thundraTraceKey;
+            } else {
+                ThundraLogger.debug(
+                    '<LambdaExecutor> Since response is not object, \
+                    skipped Thundra AppSync trace key injection for transaction',
+                    execContext.transactionId);
+            }
+
+            InvocationTraceSupport.addOutgoingTraceLink(traceId);
+        }
+    } catch (error) {
+        ThundraLogger.error('<LambdaExecutor> Failed to inject appsync trace id:', error);
+    }
+}
+
 function injectStepFunctionInfo(request: any, response: any, execContext: ExecutionContext): any {
     try {
         const isStepFunction = ConfigProvider.get<boolean>(ConfigNames.THUNDRA_LAMBDA_AWS_STEPFUNCTIONS);