Merge pull request #1 from streamfold/dd-layer

Add DD layer test

Author: mheffner

.github/workflows/benchmark.yml

@@ -14,7 +14,11 @@ on:
       rotel-layer:
         description: 'Rotel Lambda Layer'
         required: true
-        default: 'arn:aws:lambda:us-east-1:418653438961:layer:rotel-extension-amd64-alpha:12'
+        default: 'arn:aws:lambda:us-east-1:418653438961:layer:rotel-extension-amd64-alpha:23'
+      datadog-layer:
+        description: 'Datadog Lambda Layer'
+        required: true
+        default: 'arn:aws:lambda:us-east-1:464622532012:layer:Datadog-Extension:53'
       python-version:
         description: 'Python version to use'
         required: true
@@ -56,6 +60,7 @@ jobs:
       - name: run
         env:
           AWS_ROLE_ARN: ${{ secrets.AWS_ROLE_ARN }}
+          DD_API_KEY: ${{ secrets.DD_API_KEY }}
           AWS_REGION: ${{ github.event.inputs.aws-region }}
           OTEL_LAYER: ${{ github.event.inputs.otel-layer }}
           ROTEL_LAYER: ${{ github.event.inputs.rotel-layer }}

README.md

@@ -5,6 +5,7 @@ Small repo to benchmark Lambda runtime execution across different layers and mem
 Tests:
 * [AWS OpenTelemetry Collector layer](https://github.com/open-telemetry/opentelemetry-lambda)
 * [Rotel collector layer](https://github.com/streamfold/rotel-lambda-extension)
+* [Datadog Go Layer](https://docs.datadoghq.com/serverless/aws_lambda/opentelemetry/?tab=python): While there is a new [Rust DD layer](https://www.datadoghq.com/blog/datadog-next-gen-lambda-extension/), the Go version is still required for OpenTelemetry support.
 
 ## Usage
 

scripts/benchmark-coldstart.sh

@@ -32,12 +32,23 @@ if [ -z "$ROTEL_LAYER" ]; then
   exit 1
 fi
 
+if [ -z "$DATADOG_LAYER" ]; then
+  echo "Must set DATADOG_LAYER"
+  exit 1
+fi
+
+if [ -z "$DD_API_KEY" ]; then
+  echo "Must set DD_API_KEY"
+  exit 1
+fi
+
 cat <<EOF
 Running benchmark with the following:
  - Memory size: $MEMORY_SIZES
  - AWS Region: $AWS_REGION
  - OTEL Layer: $OTEL_LAYER
  - Rotel Layer: $ROTEL_LAYER
+ - Datadog Layer: $DATADOG_LAYER
 EOF
 
 set -e
@@ -72,3 +83,12 @@ for MEMORY_SIZE in ${MEMORY_SIZES}; do
     --role-arn "$AWS_ROLE_ARN" --region "$AWS_REGION" --layer "$ROTEL_LAYER" --output "$OUTPUT" \
     --memory "$MEMORY_SIZE"
 done
+
+for MEMORY_SIZE in ${MEMORY_SIZES}; do
+  echo "Testing Datadog layer at $MEMORY_SIZE MB"
+
+  uv run main.py --path ../function.zip --count $COUNT --function-name benchmark-coldstart-datadog \
+    --environment DD_API_KEY=${DD_API_KEY},DD_OTLP_CONFIG_RECEIVER_PROTOCOLS_HTTP_ENDPOINT=localhost:4318 \
+    --role-arn "$AWS_ROLE_ARN" --region "$AWS_REGION" --layer "$DATADOG_LAYER" --output "$OUTPUT" \
+    --memory "$MEMORY_SIZE"
+done