datadog-lambda 3.26.0 → 3.27.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 235f4d57b486d61592d478b2e12ea42d13ea00146520a973b75534a6f778f803
-  data.tar.gz: 84bd271dbb4d5507f442c4bce2a426f8a25f922e06f35e7350577eb465a4ddc2
+  metadata.gz: 327f22524c28715a772eae6ea159c4b39c2bdc5ce8ceca2c5f665f45d3ecaa72
+  data.tar.gz: 325f5f3d40f965f34b9474df80580e6e8664dfedcfa8e072c24bcf065d0992e0
 SHA512:
-  metadata.gz: 1e3357b9471d5f3da25b344f4b23e3eb1c123d9f8308e6eee61d398cb9ddc0fb683c0cc9099b3f0bc27e329cd607d64d1471acf1d0487c83686085bd9cc2335a
-  data.tar.gz: ef9828a5509e0aa5940cf45e706734bc5ec077e32c9387b5c4aace5699e82cdd09a48fbdd0bf7318394d8172f6ecc953f0d1de4eb2ff5ee665fc459081067d3c
+  metadata.gz: 1f2d0d594085a1c719d62329369706e0febc232aab59c8bb3fd56de050d6419dc74f07d692b294443ab80a6693d756515c09f71909741a9cd693632a330f9b8c
+  data.tar.gz: cf47002396c9583e3afe62c4bb1ea654c4644e1b073ec13356b17c87e5f44bea8bda1a6335c3e2dfe71bb56a429a476f837b4ab89af0559b8810f1c2d69a5d70

data/lib/datadog/lambda/trace/listener.rb

@@ -42,7 +42,7 @@ module Datadog
         options[:service] = 'aws.lambda'
         options[:type] = 'serverless'
 
-        trace_digest = Datadog::Utils.send_start_invocation_request(event:)
+        trace_digest = Datadog::Utils.send_start_invocation_request(event:, request_context:)
         # Only continue trace from a new one if it exist, or else,
         # it will create a new trace, which is not ideal here.
         options[:continue_from] = trace_digest if trace_digest
@@ -53,8 +53,8 @@ module Datadog
       end
       # rubocop:enable Metrics/AbcSize
 
-      def on_end(response:)
-        Datadog::Utils.send_end_invocation_request(response:, span_id: @trace.id)
+      def on_end(response:, request_context:)
+        Datadog::Utils.send_end_invocation_request(response:, span_id: @trace.id, request_context:)
         @trace&.finish
       end
 

data/lib/datadog/lambda/utils/extension.rb

@@ -22,6 +22,8 @@ module Datadog
     DD_SPAN_ID_HEADER = 'x-datadog-span-id'
     DD_PARENT_ID_HEADER = Datadog::Tracing::Distributed::Datadog::PARENT_ID_KEY
 
+    LAMBDA_RUNTIME_AWS_REQUEST_HEADER_ID = 'lambda-runtime-aws-request-id'
+
     START_INVOCATION_URI = URI(EXTENSION_BASE_URL + START_INVOCATION_PATH).freeze
     END_INVOCATION_URI = URI(EXTENSION_BASE_URL + END_INVOCATION_PATH).freeze
 
@@ -40,10 +42,13 @@ module Datadog
       File.exist?(EXTENSION_PATH)
     end
 
-    def self.send_start_invocation_request(event:)
+    def self.send_start_invocation_request(event:, request_context:)
       return unless extension_running?
 
-      response = Net::HTTP.post(START_INVOCATION_URI, event.to_json, request_headers)
+      headers = request_headers
+      headers[LAMBDA_RUNTIME_AWS_REQUEST_HEADER_ID] = request_context.aws_request_id
+      response = Net::HTTP.post(START_INVOCATION_URI, event.to_json, headers)
+
       # Add origin, since tracer expects it for extraction
       response[Datadog::Trace::DD_ORIGIN] = 'lambda'
 
@@ -53,7 +58,7 @@ module Datadog
     end
 
     # rubocop:disable Metrics/AbcSize
-    def self.send_end_invocation_request(response:, span_id:)
+    def self.send_end_invocation_request(response:, span_id:, request_context:)
       return unless extension_running?
 
       request = Net::HTTP::Post.new(END_INVOCATION_URI)
@@ -66,6 +71,9 @@ module Datadog
       # Propagator doesn't inject span_id, so we do it manually
       # It is needed for the extension to take this span id
       request[DD_SPAN_ID_HEADER] = span_id.to_s
+
+      request[LAMBDA_RUNTIME_AWS_REQUEST_HEADER_ID] = request_context.aws_request_id
+
       # Remove Parent ID if it is the same as the Span ID
       request.delete(DD_PARENT_ID_HEADER) if request[DD_PARENT_ID_HEADER] == span_id.to_s
       Datadog::Utils.logger.debug "End invocation request headers: #{request.to_hash}"

data/lib/datadog/lambda/version.rb

@@ -12,7 +12,7 @@ module Datadog
   module Lambda
     module VERSION
       MAJOR = 3
-      MINOR = 26
+      MINOR = 27
       PATCH = 0
       PRE = nil
 

data/lib/datadog/lambda.rb

@@ -66,7 +66,7 @@ module Datadog
         record_enhanced('errors', context)
         raise e
       ensure
-        @listener&.on_end(response: @response)
+        @listener&.on_end(response: @response, request_context: context)
         @is_cold_start = false
         @metrics_client.close
       end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: datadog-lambda
 version: !ruby/object:Gem::Version
-  version: 3.26.0
+  version: 3.27.0
 platform: ruby
 authors:
 - Datadog, Inc.
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2025-07-25 00:00:00.000000000 Z
+date: 2025-12-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-xray-sdk