datadog-lambda 0.7.0 → 1.12.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 300ec79cee959b800acfbb7db97f2495cd73357dbbf2982e3fa471912c21c4eb
-  data.tar.gz: 8854a16bbafcaba922d387f6923fde1340ba788284c72ec6affc1afa373a1184
+  metadata.gz: f40ca893546c79c836081f4fb4a63aa6862e213f91453a7b630c25c59ca73ba3
+  data.tar.gz: 01e664333a8a5f114363a6b79d47efa46ba464d1c36372ab4178a35010058245
 SHA512:
-  metadata.gz: 02aa89e21441a5d66ac6af0568e1fb9e441f04d392744ee0848b2b229a9e47305ac08120b8f18b0b68889c99c4e618ecad812667ff34244185e4ebd1fadb9f03
-  data.tar.gz: fc9da6f32e2a9802bee2cfe29953ace582fbeda26a94fde8e6a1064ee80021e78783562d21d66b29c040fe6540c02808bcecc1594af73e1537c66f9f6009c0d4
+  metadata.gz: 6330bddce249fd6263c3858d447fbb4789f5f280517ad9a25fa669e0ce33728ed6a46e4ee03757d954c8de366d27e6db9b2602b5629f4cc0bcee3bbb18c0d777
+  data.tar.gz: b3e1c27fc86ce76afc6a7ceaed3161b26b34e21fac1eb38e211ec7c5c9668f5eb4ea4175ea88a3805c723054f9d2e1b0462951d9f08245296f39ad7e248b446f

data/lib/datadog/lambda.rb

@@ -7,12 +7,14 @@
 # This product includes software developed at Datadog (https://www.datadoghq.com/).
 # Copyright 2019 Datadog, Inc.
 #
+# rubocop:disable Metrics/ModuleLength
 
 require 'datadog/lambda/trace/listener'
 require 'datadog/lambda/utils/logger'
 require 'datadog/lambda/trace/patch_http'
 require 'json'
 require 'time'
+require 'datadog/lambda/version'
 
 module Datadog
   # Instruments AWS Lambda functions with Datadog distributed tracing and
@@ -36,6 +38,7 @@ module Datadog
         c.tracer writer: Datadog::SyncWriter.new(
           transport: Datadog::Transport::IO.default
         )
+        c.tags = { "_dd.origin": 'lambda' }
         yield(c) if block_given?
       end
     end
@@ -67,7 +70,6 @@ module Datadog
     # Gets the current tracing context
     def self.trace_context
       context = Hash[Datadog::Trace.trace_context]
-      context.delete(:source)
       context
     end
 
@@ -81,34 +83,65 @@ module Datadog
       raise 'value must be a number' unless value.is_a?(Numeric)
 
       time ||= Time.now
-      tag_list = ['dd_lambda_layer:datadog-ruby25']
+      time_ms = time.to_f.to_i
+
+      tag_list = ["dd_lambda_layer:datadog-ruby#{dd_lambda_layer_tag}"]
       tags.each do |tag|
         tag_list.push("#{tag[0]}:#{tag[1]}")
       end
-      time_ms = time.to_f.to_i
-      metric = { e: time_ms, m: name, t: tag_list, v: value }.to_json
-      puts metric
+      metric = { e: time_ms, m: name, t: tag_list, v: value }
+      puts metric.to_json
+    end
+
+    def self.dd_lambda_layer_tag
+      RUBY_VERSION[0, 3].tr('.', '')
     end
 
     # Generate tags for enhanced metrics
     # @param context [Object] https://docs.aws.amazon.com/lambda/latest/dg/ruby-context.html
     # @return [hash] a hash of the enhanced metrics tags
+    # rubocop:disable Metrics/AbcSize, Metrics/MethodLength
     def self.gen_enhanced_tags(context)
       arn_parts = context.invoked_function_arn.split(':')
-      {
+      # Check if we have an alias or version
+      function_alias = arn_parts[7].nil? ? nil : arn_parts[7]
+
+      tags = {
         functionname: context.function_name,
         region: arn_parts[3],
         account_id: arn_parts[4],
         memorysize: context.memory_limit_in_mb,
         cold_start: @is_cold_start,
-        runtime: "Ruby #{RUBY_VERSION}"
+        runtime: "Ruby #{RUBY_VERSION}",
+        resource: context.function_name,
+        datadog_lambda: Datadog::Lambda::VERSION::STRING.to_sym
       }
+      begin
+        tags[:dd_trace] = Gem.loaded_specs['ddtrace'].version
+      rescue StandardError
+        Datadog::Utils.logger.debug 'dd-trace unavailable'
+      end
+      # If we have an alias...
+      unless function_alias.nil?
+        # If the alis version is $Latest, drop the $ for ddog tag convention.
+        if function_alias.start_with?('$')
+          function_alias[0] = ''
+          # If the alias is not a version number add the executed version tag
+        elsif !/\A\d+\z/.match(function_alias)
+          tags[:executedversion] = context.function_version
+        end
+        # Append the alias to the resource tag
+        tags[:resource] = context.function_name + ':' + function_alias
+      end
+
+      tags
     rescue StandardError => e
       Datadog::Utils.logger.error 'Unable to parse Lambda context' \
       "#{context}: #{e}"
       {}
     end
 
+    # rubocop:enable Metrics/AbcSize, Metrics/MethodLength
     # Format and add tags to enhanced metrics
     # This method wraps the metric method, checking the DD_ENHANCED_METRICS
     # environment variable, adding 'aws.lambda.enhanced' to the metric name,
@@ -116,7 +149,7 @@ module Datadog
     # @param metric_name [String] basic name of the metric
     # @param context [Object] AWS Ruby Lambda Context
     # @return [boolean] false if the metric was not added for some reason,
-    #   true otherwise (for ease of testing)
+    #   true otherwise (for ease of testing
 
     def self.record_enhanced(metric_name, context)
       return false unless do_enhanced_metrics?
@@ -153,3 +186,4 @@ module Datadog
     end
   end
 end
+# rubocop:enable Metrics/ModuleLength

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

@@ -37,22 +37,39 @@ module Datadog
       def on_end; end
 
       def on_wrap(request_context:, cold_start:, &block)
+        options = get_option_tags(
+          request_context: request_context,
+          cold_start: cold_start
+        )
+        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[:service] = 'aws.lambda'
+        options[:span_type] = 'serverless'
+        Datadog::Trace.apply_datadog_trace_context(Datadog::Trace.trace_context)
+        Datadog::Trace.wrap_datadog(options) do
+          block.call
+        end
+      end
+
+      private
+
+      def get_option_tags(request_context:, cold_start:)
+        function_arn = request_context.invoked_function_arn.downcase
+        tk = function_arn.split(':')
+        function_arn = tk.length > 7 ? tk[0, 7].join(':') : function_arn
+        function_version = tk.length > 7 ? tk[7] : '$LATEST'
         options = {
           tags: {
             cold_start: cold_start,
-            function_arn: request_context.invoked_function_arn.downcase,
+            function_arn: function_arn,
+            function_version: function_version,
             request_id: request_context.aws_request_id,
             resource_names: request_context.function_name
           }
         }
-
-        options[:resource] = @handler_name
-        options[:service] =  @function_name
-        options[:span_type] = 'serverless'
-        Datadog::Trace.apply_datadog_trace_context(Datadog::Trace.trace_context)
-        Datadog::Trace.wrap_datadog(options) do
-          block.call
-        end
+        options
       end
     end
   end

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

@@ -30,25 +30,31 @@ module Datadog
     # NetExtensions contains patches which add tracing context to http calls
     module NetExtensions
       def request(req, body = nil, &block)
-        logger = Datadog::Utils.logger
         begin
           context = Datadog::Trace.current_trace_context(
             Datadog::Trace.trace_context
           )
 
-          req[Datadog::Trace::DD_SAMPLING_PRIORITY_HEADER.to_sym] =
-            context[:sample_mode]
-          req[Datadog::Trace::DD_PARENT_ID_HEADER.to_sym] = context[:parent_id]
-          req[Datadog::Trace::DD_TRACE_ID_HEADER.to_sym] = context[:trace_id]
-          logger.debug("added context #{context} to request")
+          req = add_ctx_to_req(req, context)
         rescue StandardError => e
           trace = e.backtrace.join("\n ")
-          logger.debug(
+          Datadog::Utils.logger.debug(
             "couldn't add tracing context #{context} to request #{e}:\n#{trace}"
           )
         end
         super(req, body, &block)
       end
+
+      private
+
+      def add_ctx_to_req(req, context)
+        req[Datadog::Trace::DD_SAMPLING_PRIORITY_HEADER.to_sym] =
+          context[:sample_mode]
+        req[Datadog::Trace::DD_PARENT_ID_HEADER.to_sym] = context[:parent_id]
+        req[Datadog::Trace::DD_TRACE_ID_HEADER.to_sym] = context[:trace_id]
+        Datadog::Utils.logger.debug("added context #{context} to request")
+        req
+      end
     end
   end
 end

data/lib/datadog/lambda/version.rb

@@ -11,8 +11,8 @@
 module Datadog
   module Lambda
     module VERSION
-      MAJOR = 0
-      MINOR = 7
+      MAJOR = 1
+      MINOR = 12
       PATCH = 0
       PRE = nil
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: datadog-lambda
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 1.12.0
 platform: ruby
 authors:
 - Datadog, Inc.
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-04-02 00:00:00.000000000 Z
+date: 2020-12-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-xray-sdk
@@ -30,14 +30,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.32'
+        version: '0.4'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.32'
+        version: '0.4'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -113,7 +113,7 @@ files:
 - lib/datadog/lambda/trace/xray.rb
 - lib/datadog/lambda/utils/logger.rb
 - lib/datadog/lambda/version.rb
-homepage: https://github.com/DataDog/dd-lambda-rb
+homepage: https://github.com/DataDog/datadog-lambda-rb
 licenses:
 - Apache-2.0
 metadata:
@@ -133,8 +133,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.7.6
+rubygems_version: 3.0.3
 signing_key: 
 specification_version: 4
 summary: Instruments your Ruby AWS Lambda functions with Datadog