feat: Set DD_TRACE_ENABLED tag on state machine

DataDog · Nov 25, 2024 · 6e22a20 · 6e22a20
1 parent 44e9dd0
commit 6e22a20
Show file tree

Hide file tree

Showing 6 changed files with 85 additions and 0 deletions.
diff --git a/integration_tests/snapshots/correct-step-function-stack-snapshot.json b/integration_tests/snapshots/correct-step-function-stack-snapshot.json
@@ -24,6 +24,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -112,6 +116,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -150,6 +158,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -292,6 +304,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -439,6 +455,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -477,6 +497,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"

diff --git a/integration_tests/snapshots/correct-step_functions_go_stack-snapshot.json b/integration_tests/snapshots/correct-step_functions_go_stack-snapshot.json
@@ -24,6 +24,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -112,6 +116,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -150,6 +158,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -293,6 +305,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -440,6 +456,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -478,6 +498,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"

diff --git a/integration_tests/snapshots/correct-step_functions_python_stack-snapshot.json b/integration_tests/snapshots/correct-step_functions_python_stack-snapshot.json
@@ -32,6 +32,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -120,6 +124,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -158,6 +166,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -338,6 +350,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -485,6 +501,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"
@@ -523,6 +543,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "DD_TRACE_ENABLED",
+      "Value": "true"
+     },
      {
       "Key": "env",
       "Value": "dev"

diff --git a/src/constants.ts b/src/constants.ts
@@ -51,6 +51,7 @@ export enum TagKeys {
   ENV = "env",
   SERVICE = "service",
   VERSION = "version",
+  DD_TRACE_ENABLED = "DD_TRACE_ENABLED",
 }
 
 export const runtimeLookup: { [key: string]: RuntimeType } = {

diff --git a/src/tag.ts b/src/tag.ts
@@ -35,4 +35,12 @@ export function setTags(
       }
     });
   }
+
+  if (resource instanceof sfn.StateMachine) {
+    setStepFunctionTags(resource);
+  }
+}
+
+function setStepFunctionTags(stateMachine: sfn.StateMachine) {
+  Tags.of(stateMachine).add(TagKeys.DD_TRACE_ENABLED, "true");
 }
diff --git a/test/tags.spec.ts b/test/tags.spec.ts
@@ -188,6 +188,10 @@ describe("setTags for Step Function", () => {
           Key: "custom-tag-2",
           Value: "tag-value-2",
         },
+        {
+          Key: "DD_TRACE_ENABLED",
+          Value: "true",
+        },
         {
           Key: "env",
           Value: "dev",