datadog-lambda 2.23.1 → 3.25.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9cddc5701e2fd24ab4ae01089f2eb7a1beee75da183f2540060639a898279637
-  data.tar.gz: 94ba04342ac432d6ab95d40bbfbbbfb44b2e186a84c8571e1e2c93514448b29b
+  metadata.gz: ea012fe059be249b92847c0f657f87b650df56dfff0f52fc6b81cd4c874ad0ee
+  data.tar.gz: 61b1e147bc6c2f48859d88204c47a69b6699374e3342feb3206b5015b3bdad02
 SHA512:
-  metadata.gz: 784af632d4b1a84667eb6ff68a32cea62c92d3861237bc414eee868f6c39c9d257153330d691eb53c6e496ce5dadc366e18f03b4e79550b3ded5718692b49d64
-  data.tar.gz: 720cfbdba885607139157881b78a3ce398368e96828d896b9782d57a5d75b45f8fe27e348f097732db65e36c7f5c250cc6160ae5db4ffebff9e7d6de1d8aa001
+  metadata.gz: 5b94452dabcf5384f90701f00772dfa28ac2445ff8bb72aad866e1ce91aae6ee066a7e62c8fffaa409d2b1573ef88335b237417d1db83b5f526a95a7c829237f
+  data.tar.gz: 66cd0290bd69aaff8b579e4424ced2077752d963a644aea79a45079fae884001b954a68852c7057fa4ec5494581071f284e088f056c963d81a95624d99a0b895

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,9 +11,9 @@
 module Datadog
   module Lambda
     module VERSION
-      MAJOR = 2
-      MINOR = 23
-      PATCH = 1
+      MAJOR = 3
+      MINOR = 25
+      PATCH = 0
       PRE = nil
 
       STRING = [MAJOR, MINOR, PATCH, PRE].compact.join('.')

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.1
+  version: 3.25.0
 platform: ruby
 authors:
 - Datadog, Inc.
-autorequire:
+autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-09-25 00:00:00.000000000 Z
+date: 2025-03-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-xray-sdk
@@ -39,33 +39,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '5.0'
 - !ruby/object:Gem::Dependency
-  name: nokogiri
+  name: datadog
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.16.0
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 1.16.0
-- !ruby/object:Gem::Dependency
-  name: ddtrace
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 1.23.3
+        version: '2.12'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.23.3
+        version: '2.12'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -122,20 +108,6 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1'
-- !ruby/object:Gem::Dependency
-  name: solargraph
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.47.2
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.47.2
 description: |
   datadog-lambda is Datadog's AWS Lambda integration for ruby. It is used to perform
   distributed tracing between serverful and serverless environments, and send
@@ -162,7 +134,7 @@ licenses:
 - Apache-2.0
 metadata:
   allowed_push_host: https://rubygems.org
-post_install_message:
+post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -177,8 +149,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.3
-signing_key:
+rubygems_version: 3.3.5
+signing_key: 
 specification_version: 4
 summary: Instruments your Ruby AWS Lambda functions with Datadog
 test_files: []