Expose profiling config in interface

Author: gabedos

src/ecs/fargate/README.md

@@ -184,6 +184,7 @@ For more general information, reference the [Datadog ECS Fargate Docs](https://d
 | `isEnabled`                  | `boolean` | Enables APM.                                                                                                                                                                                     |
 | `isSocketEnabled`            | `boolean` | Enables APM traces traffic over Unix Domain Socket. Falls back to TCP when false.                                                                                                                |
 | `traceInferredProxyServices` | `boolean` | Enables inferred spans for proxy services like AWS API Gateway. When enabled, the tracer will create spans for proxy services by using headers passed from the proxy service to the application. |
+| `isProfilingEnabled`         | `boolean` | Enables Profiling. (Requires APM SSI on application containers)                                                                                                                                  |
 
 ### CWSFeatureConfig
 

src/ecs/fargate/datadog-ecs-fargate.ts

@@ -218,6 +218,12 @@ export class DatadogECSFargateTaskDefinition extends ecs.FargateTaskDefinition {
       });
     }
 
+    if (this.datadogProps.apm!.isEnabled) {
+      if (this.datadogProps.apm!.isProfilingEnabled) {
+        container.addEnvironment("DD_PROFILING_ENABLED", "true");
+      }
+    }
+
     // Universal Service Tagging configuration:
     if (this.datadogProps.env) {
       container.addEnvironment("DD_ENV", this.datadogProps.env);

src/ecs/interfaces.ts

@@ -157,6 +157,11 @@ export interface APMFeatureConfig {
    * passed from the proxy service to the application.
    */
   readonly traceInferredProxyServices?: boolean;
+  /**
+   * Enables Profile collection.
+   * Requires Datadog APM SSI instrumentation on your application containers.
+   */
+  readonly isProfilingEnabled?: boolean;
 }
 
 /**

test/ecs/fargate/apm-dsd.spec.ts

@@ -169,6 +169,38 @@ describe("DatadogECSFargateTaskDefinition", () => {
     });
   });
 
+  it("should inject DD_PROFILING_ENABLED when profiling is enabled", () => {
+    datadogProps = {
+      ...datadogProps,
+      apm: {
+        isEnabled: true,
+        isProfilingEnabled: true,
+      },
+    };
+    const taskDefinition = new ecsDatadog.DatadogECSFargateTaskDefinition(scope, id, props, datadogProps);
+    taskDefinition.addContainer("app-container", {
+      containerName: "app-container",
+      image: ecs.ContainerImage.fromRegistry("amazon/amazon-ecs-sample"),
+      memoryLimitMiB: 512,
+    });
+    const template = Template.fromStack(stack);
+
+    // Validate that the ports are configured on the datadog-agent container
+    template.hasResourceProperties("AWS::ECS::TaskDefinition", {
+      ContainerDefinitions: Match.arrayWith([
+        Match.objectLike({
+          Name: "app-container",
+          Environment: Match.arrayWith([
+            Match.objectLike({
+              Name: "DD_PROFILING_ENABLED",
+              Value: "true",
+            }),
+          ]),
+        }),
+      ]),
+    });
+  });
+
   it("should enable origin detection when isOriginDetectionEnabled is true", () => {
     datadogProps = {
       ...datadogProps,