feat: Set dd_cdk_construct version tag on state machine

DataDog · Nov 26, 2024 · df9fb18 · df9fb18
1 parent 8fcf1e1
commit df9fb18
Show file tree

Hide file tree

Showing 5 changed files with 79 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_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -116,6 +120,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -158,6 +166,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -304,6 +316,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -455,6 +471,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -497,6 +517,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"

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_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -116,6 +120,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -158,6 +166,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -305,6 +317,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -456,6 +472,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -498,6 +518,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"

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_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -124,6 +128,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -166,6 +174,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -350,6 +362,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -501,6 +517,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"
@@ -543,6 +563,10 @@
       "Key": "custom-tag-2",
       "Value": "tag-value-2"
      },
+     {
+      "Key": "dd_cdk_construct",
+      "Value": "vX.XX.X"
+     },
      {
       "Key": "DD_TRACE_ENABLED",
       "Value": "true"

diff --git a/src/tag.ts b/src/tag.ts
@@ -12,6 +12,8 @@ import * as sfn from "aws-cdk-lib/aws-stepfunctions";
 import log from "loglevel";
 import { TagKeys, DatadogLambdaProps, DatadogStepFunctionsProps } from "./index";
 
+const versionJson = require("../version.json");
+
 export function setTags(
   resource: lambda.Function | sfn.StateMachine,
   props: DatadogLambdaProps | DatadogStepFunctionsProps,
@@ -43,4 +45,5 @@ export function setTags(
 
 function setStepFunctionTags(stateMachine: sfn.StateMachine) {
   Tags.of(stateMachine).add(TagKeys.DD_TRACE_ENABLED, "true");
+  Tags.of(stateMachine).add(TagKeys.CDK, `v${versionJson.version}`);
 }
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_cdk_construct",
+          Value: `v${versionJson.version}`,
+        },
         {
           Key: "DD_TRACE_ENABLED",
           Value: "true",