datadog-lambda 2.23.0 → 3.24.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d7d5f92e39c4579b9ac265f60f354a128a2f02c348e98127142081b3593ea28c
-  data.tar.gz: 19821c8a70157f6738a9a47dd4ccf3d69ef853322e261e15596af68041eb8959
+  metadata.gz: 95cfdc9d348ccc91adda2193dc1c42c6c7533154b2c845a9f7517ca8f35e7d45
+  data.tar.gz: 9770541d230535fd72916d237d647c63447ac5c0b563589dd011e0a9b1199336
 SHA512:
-  metadata.gz: 48c653819b1eb3f353a183820310381e7ee5e67691d79988572bb9482054e74c71e683f359ce051d28c1239a0b8edd07c530e986f1f734b435b3d413a1d2f786
-  data.tar.gz: 3ce2f6c46406aaa8557f3a7ea01eeb3b62516edc0d4ee1d518141a444932dcdda6a0ad358a0c1a2c861be14a1d852602548074064552b9d89b902f141c3fd62a
+  metadata.gz: 13dec18ab59eb3bea0a73a5a674d58350e008fe6bb9e4fbaae3952da0efc8a8607dc919bc0078ead238cb41844867045c29ba8258f1ebe1a8e5bae8d5390626d
+  data.tar.gz: edeb3616cd4a0d2a43183113069b5ed334930e34ddbb639ad5eead60e2c4593b6ca50289da301a498f933c0f703b2bfd91f9b79be74b7591176f765ef43c1f1a

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

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-# `ddtrace`` is an optional dependency for the ruby package
+# `datadog`` is an optional dependency for the ruby package
 begin
-  require 'ddtrace'
+  require 'datadog'
 rescue LoadError
-  Datadog::Utils.logger.debug 'dd-trace unavailable'
+  Datadog::Utils.logger.debug 'datadog unavailable'
 end
 
 module Datadog

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

@@ -38,10 +38,9 @@ module Datadog
         context = Datadog::Trace.trace_context
         source = context[:source] if context
         options[:tags]['_dd.parent_source'] = source if source && source != 'ddtrace'
-        options[:resource] = @function_name
+        options[:resource] = 'dd-tracer-serverless-span'
         options[:service] = 'aws.lambda'
-        options[:span_type] = 'serverless'
-        Datadog::Trace.apply_datadog_trace_context(Datadog::Trace.trace_context)
+        options[:type] = 'serverless'
 
         trace_digest = Datadog::Utils.send_start_invocation_request(event:)
         # Only continue trace from a new one if it exist, or else,
@@ -49,11 +48,13 @@ module Datadog
         options[:continue_from] = trace_digest if trace_digest
 
         @trace = Datadog::Tracing.trace('aws.lambda', **options)
+
+        Datadog::Trace.apply_datadog_trace_context(Datadog::Trace.trace_context)
       end
       # rubocop:enable Metrics/AbcSize
 
       def on_end(response:)
-        Datadog::Utils.send_end_invocation_request(response:)
+        Datadog::Utils.send_end_invocation_request(response:, span_id: @trace.id)
         @trace&.finish
       end
 

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

@@ -19,6 +19,9 @@ module Datadog
     START_INVOCATION_PATH = '/lambda/start-invocation'
     END_INVOCATION_PATH = '/lambda/end-invocation'
 
+    DD_SPAN_ID_HEADER = 'x-datadog-span-id'
+    DD_PARENT_ID_HEADER = Datadog::Tracing::Distributed::Datadog::PARENT_ID_KEY
+
     START_INVOCATION_URI = URI(EXTENSION_BASE_URL + START_INVOCATION_PATH).freeze
     END_INVOCATION_URI = URI(EXTENSION_BASE_URL + END_INVOCATION_PATH).freeze
 
@@ -49,7 +52,8 @@ module Datadog
       Datadog::Utils.logger.debug "failed on start invocation request to extension: #{e}"
     end
 
-    def self.send_end_invocation_request(response:)
+    # rubocop:disable Metrics/AbcSize
+    def self.send_end_invocation_request(response:, span_id:)
       return unless extension_running?
 
       request = Net::HTTP::Post.new(END_INVOCATION_URI)
@@ -59,12 +63,20 @@ module Datadog
       trace_digest = Datadog::Tracing.active_trace&.to_digest
 
       PROPAGATOR.inject!(trace_digest, request)
+      # 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
+      # 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}"
+
       Net::HTTP.start(END_INVOCATION_URI.host, END_INVOCATION_URI.port) do |http|
         http.request(request)
       end
     rescue StandardError => e
       Datadog::Utils.logger.debug "failed on end invocation request to extension: #{e}"
     end
+    # rubocop:enable Metrics/AbcSize
 
     def self.request_headers
       {

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

@@ -14,21 +14,19 @@ module Datadog
   # Utils contains utility functions shared between modules
   module Utils
     def self.logger
-      @logger ||= Logger.new($stdout)
-    end
-
-    def self.update_log_level
-      log_level = (ENV['DD_LOG_LEVEL'] || 'error').downcase
-      logger.level = case log_level
-                     when 'debug'
-                       Logger::DEBUG
-                     when 'info'
-                       Logger::INFO
-                     when 'warn'
-                       Logger::WARN
-                     else
-                       Logger::ERROR
-                     end
+      @logger ||= Logger.new($stderr).tap do |logger|
+        log_level = (ENV['DD_LOG_LEVEL'] || 'error').downcase
+        logger.level = case log_level
+                       when 'debug'
+                         Logger::DEBUG
+                       when 'info'
+                         Logger::INFO
+                       when 'warn'
+                         Logger::WARN
+                       else
+                         Logger::ERROR
+                       end
+      end
     end
   end
 end

data/lib/datadog/lambda/version.rb

@@ -11,8 +11,8 @@
 module Datadog
   module Lambda
     module VERSION
-      MAJOR = 2
-      MINOR = 23
+      MAJOR = 3
+      MINOR = 24
       PATCH = 0
       PRE = nil
 

data/lib/datadog/lambda.rb

@@ -56,7 +56,6 @@ module Datadog
     # @param context [Object] lambda context
     # @param block [Proc] implementation of the handler function.
     def self.wrap(event, context, &block)
-      Datadog::Utils.update_log_level
       @listener ||= initialize_listener
       record_enhanced('invocations', context)
       begin
@@ -115,9 +114,9 @@ module Datadog
         datadog_lambda: Datadog::Lambda::VERSION::STRING.to_sym
       }
       begin
-        tags[:dd_trace] = Gem.loaded_specs['ddtrace'].version
+        tags[:dd_trace] = Gem.loaded_specs['datadog'].version
       rescue StandardError
-        Datadog::Utils.logger.debug 'dd-trace unavailable'
+        Datadog::Utils.logger.debug 'datadog unavailable'
       end
       # If we have an alias...
       unless function_alias.nil?
@@ -188,7 +187,7 @@ module Datadog
 
       # Only initialize listener if Tracing enabled.
       unless Datadog::Tracing.enabled?
-        Datadog::Utils.logger.debug 'dd-trace unavailable'
+        Datadog::Utils.logger.debug 'datadog unavailable'
         return nil
       end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: datadog-lambda
 version: !ruby/object:Gem::Version
-  version: 2.23.0
+  version: 3.24.0
 platform: ruby
 authors:
 - Datadog, Inc.
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-09-18 00:00:00.000000000 Z
+date: 2025-01-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-xray-sdk
@@ -44,28 +44,28 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.15.6
+        version: 1.16.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.15.6
+        version: 1.16.0
 - !ruby/object:Gem::Dependency
-  name: ddtrace
+  name: datadog
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.23.3
+        version: '2.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.23.3
+        version: '2.0'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -177,7 +177,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.18
+rubygems_version: 3.5.3
 signing_key:
 specification_version: 4
 summary: Instruments your Ruby AWS Lambda functions with Datadog