instana 1.193.2 → 1.193.6

data/lib/instana/instrumentation/action_controller.rb

@@ -0,0 +1,84 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+module Instana
+  module Instrumentation
+    module ActionController
+      def process_action(*args)
+        call_payload = {
+          actioncontroller: {
+            controller: self.class.name,
+            action: action_name
+          }
+        }
+
+        request.env['INSTANA_HTTP_PATH_TEMPLATE'] = matched_path_template
+        ::Instana::Tracer.trace(:actioncontroller, call_payload) { super(*args) }
+      end
+
+      def render(*args, &block)
+        call_payload = {
+          actionview: {
+            name: describe_render_options(args.first) || 'Default'
+          }
+        }
+
+        ::Instana::Tracer.trace(:actionview, call_payload) { super(*args, &block) }
+      end
+
+      private
+
+      def matched_path_template
+        Rails.application.routes.router.recognize(request) do |route, _, _|
+          path = route.path
+          return path.spec.to_s
+        end
+
+        nil
+      end
+
+      def describe_render_options(options)
+        return unless options.is_a?(Hash)
+
+        describe_layout(options[:layout]) ||
+          describe_direct(options)
+      end
+
+      def describe_layout(layout)
+        return unless layout
+
+        case layout
+        when FalseClass
+          'Without layout'
+        when String
+          layout
+        when Proc
+          'Proc'
+        else
+          'Default'
+        end
+      end
+
+      def describe_direct(options)
+        case options
+        when ->(o) { o.key?(:nothing) }
+          'Nothing'
+        when ->(o) { o.key?(:plain) }
+          'Plaintext'
+        when ->(o) { o.key?(:json) }
+          'JSON'
+        when ->(o) { o.key?(:xml) }
+          'XML'
+        when ->(o) { o.key?(:body) }
+          'Raw'
+        when ->(o) { o.key?(:js) }
+          'Javascript'
+        when ->(o) { o.key?(:template) }
+          options[:template]
+        when ->(o) { o.key?(:file) }
+          options[:file]
+        end
+      end
+    end
+  end
+end

data/lib/instana/instrumentation/action_view.rb

@@ -0,0 +1,30 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+module Instana
+  module Instrumentation
+    module ActionView
+      def render_partial(*args)
+        call_payload = {
+          render: {
+            type: :partial,
+            name: @options.is_a?(Hash) ? @options[:partial].to_s : 'Unknown'
+          }
+        }
+
+        ::Instana::Tracer.trace(:render, call_payload) { super(*args) }
+      end
+
+      def render_collection(*args)
+        call_payload = {
+          render: {
+            type: :collection,
+            name: @path.to_s
+          }
+        }
+
+        ::Instana::Tracer.trace(:render, call_payload) { super(*args) }
+      end
+    end
+  end
+end

data/lib/instana/instrumentation/active_record.rb

@@ -0,0 +1,50 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+module Instana
+  module Instrumentation
+    module ActiveRecord
+      IGNORED_NAMES = %w[SCHEMA EXPLAIN CACHE].freeze
+      IGNORED_SQL = %w[BEGIN COMMIT SET].freeze
+      SANITIZE_REGEXP = /('[\s\S][^']*'|\d*\.\d+|\d+|NULL)/i.freeze
+
+      def log(sql, name = 'SQL', binds = [], *args)
+        call_payload = {
+          activerecord: {
+            adapter: @config[:adapter],
+            host: @config[:host],
+            username: @config[:username],
+            db: @config[:database],
+            sql: maybe_sanitize(sql)
+          }
+        }
+
+        if binds.all? { |b| b.respond_to?(:value_before_type_cast) }
+          mapped = binds.map(&:value_before_type_cast)
+          call_payload[:activerecord][:binds] = mapped
+        end
+
+        maybe_trace(call_payload, name) { super(sql, name, binds, *args) }
+      end
+
+      private
+
+      def maybe_sanitize(sql)
+        ::Instana.config[:sanitize_sql] ? sql.gsub(SANITIZE_REGEXP, '?') : sql
+      end
+
+      def maybe_trace(call_payload, name, &blk)
+        if ::Instana.tracer.tracing? && !ignored?(call_payload, name)
+          ::Instana.tracer.trace(:activerecord, call_payload, &blk)
+        else
+          yield
+        end
+      end
+
+      def ignored?(call_payload, name)
+        IGNORED_NAMES.include?(name) ||
+          IGNORED_SQL.any? { |s| call_payload[:activerecord][:sql].upcase.start_with?(s) }
+      end
+    end
+  end
+end

data/lib/instana/instrumentation/aws_sdk_dynamodb.rb

@@ -0,0 +1,29 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+module Instana
+  module Instrumentation
+    class DynamoDB < Seahorse::Client::Plugin
+      class Handler < Seahorse::Client::Handler
+        def call(context)
+          dynamo_tags = {
+            op: context.operation_name,
+            table: table_name_from(context)
+          }
+
+          ::Instana.tracer.trace(:dynamodb, dynamo_tags) { @handler.call(context) }
+        end
+
+        private
+
+        def table_name_from(context)
+          context.params[:table_name] || context.params[:global_table_name] || 'Unknown'
+        end
+      end
+
+      def add_handlers(handlers, _config)
+        handlers.add(Handler, step: :initialize)
+      end
+    end
+  end
+end

data/lib/instana/instrumentation/dalli.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2017
+
 module Instana
   module Instrumentation
     module Dalli
@@ -69,9 +72,3 @@ module Instana
     end
   end
 end
-
-if defined?(::Dalli) && ::Instana.config[:dalli][:enabled]
-  ::Instana.logger.debug "Instrumenting Dalli"
-  ::Dalli::Client.send(:prepend, ::Instana::Instrumentation::Dalli)
-  ::Dalli::Server.send(:prepend, ::Instana::Instrumentation::DalliServer)
-end

data/lib/instana/instrumentation/excon.rb

@@ -1,74 +1,71 @@
-if defined?(::Excon) && ::Instana.config[:excon][:enabled]
-  module Instana
-    module Instrumentation
-      class Excon < ::Excon::Middleware::Base
-        def request_call(datum)
-          return @stack.request_call(datum) unless ::Instana.tracer.tracing?
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
 
-          payload = { :http => {} }
-          path = datum[:path].split('?').first
-          payload[:http][:url] = ::Instana.secrets.remove_from_query("#{datum[:connection].instance_variable_get(:@socket_key)}#{path}")
-          payload[:http][:method] = datum[:method] if datum.key?(:method)
+module Instana
+  module Instrumentation
+    class Excon < ::Excon::Middleware::Base
+      def request_call(datum)
+        return @stack.request_call(datum) unless ::Instana.tracer.tracing? || !Instana.tracer.current_span.exit_span?
 
-          if datum[:pipeline] == true
-            # Pass the context along in the datum so we get back on response
-            # and can close out the async span
-            datum[:instana_span] = ::Instana.tracer.log_async_entry(:excon, payload)
-            t_context = datum[:instana_span].context
-          else
-            ::Instana.tracer.log_entry(:excon, payload)
-            t_context = ::Instana.tracer.context
-          end
+        payload = { :http => {} }
+        path = datum[:path].split('?').first
+        payload[:http][:url] = ::Instana.secrets.remove_from_query("#{datum[:connection].instance_variable_get(:@socket_key)}#{path}")
+        payload[:http][:method] = datum[:method] if datum.key?(:method)
 
-          # Set request headers; encode IDs as hexadecimal strings
-          datum[:headers]['X-Instana-T'] = t_context.trace_id_header
-          datum[:headers]['X-Instana-S'] = t_context.span_id_header
-
-          @stack.request_call(datum)
+        if datum[:pipeline] == true
+          # Pass the context along in the datum so we get back on response
+          # and can close out the async span
+          datum[:instana_span] = ::Instana.tracer.log_async_entry(:excon, payload)
+          t_context = datum[:instana_span].context
+        else
+          ::Instana.tracer.log_entry(:excon, payload)
+          t_context = ::Instana.tracer.context
         end
 
-        def error_call(datum)
-          return @stack.error_call(datum) unless ::Instana.tracer.tracing?
+        # Set request headers; encode IDs as hexadecimal strings
+        datum[:headers]['X-Instana-T'] = t_context.trace_id_header
+        datum[:headers]['X-Instana-S'] = t_context.span_id_header
+
+        @stack.request_call(datum)
+      end
 
-          if datum[:pipeline] == true
-            ::Instana.tracer.log_async_error(datum[:error], datum[:instana_span])
-          else
-            ::Instana.tracer.log_error(datum[:error])
-          end
-          @stack.error_call(datum)
+      def error_call(datum)
+        return @stack.error_call(datum) unless ::Instana.tracer.tracing? || !Instana.tracer.current_span.exit_span?
+
+        if datum[:pipeline] == true
+          ::Instana.tracer.log_async_error(datum[:error], datum[:instana_span])
+        else
+          ::Instana.tracer.log_error(datum[:error])
         end
+        @stack.error_call(datum)
+      end
 
-        def response_call(datum)
-          # FIXME: Will connect exceptions call a response?
-          #
-          return @stack.response_call(datum) unless ::Instana.tracer.tracing?
+      def response_call(datum)
+        # FIXME: Will connect exceptions call a response?
+        #
+        return @stack.response_call(datum) unless ::Instana.tracer.tracing? || !Instana.tracer.current_span.exit_span?
 
-          result =  @stack.response_call(datum)
+        result =  @stack.response_call(datum)
 
-          status = datum[:status]
-          if !status && datum.key?(:response) && datum[:response].is_a?(Hash)
-            status = datum[:response][:status]
-          end
+        status = datum[:status]
+        if !status && datum.key?(:response) && datum[:response].is_a?(Hash)
+          status = datum[:response][:status]
+        end
 
-          if status.between?(500, 511)
-            # Because of the 5xx response, we flag this span as errored but
-            # without a backtrace (no exception)
-            ::Instana.tracer.log_error(nil)
-          end
+        if status.between?(500, 511)
+          # Because of the 5xx response, we flag this span as errored but
+          # without a backtrace (no exception)
+          ::Instana.tracer.log_error(nil)
+        end
 
-          if datum[:pipeline] == true
-            # Pickup context of this async span from datum[:instana_span]
-            ::Instana.tracer.log_async_exit(:excon, { :http => {:status => status } }, datum[:instana_span])
-          else
-            ::Instana.tracer.log_exit(:excon, { :http => {:status => status } })
-          end
-          result
+        if datum[:pipeline] == true
+          # Pickup context of this async span from datum[:instana_span]
+          ::Instana.tracer.log_async_exit(:excon, { :http => {:status => status } }, datum[:instana_span])
+        else
+          ::Instana.tracer.log_exit(:excon, { :http => {:status => status } })
         end
+        result
       end
     end
   end
-
-  ::Instana.logger.debug "Instrumenting Excon"
-  ::Excon.defaults[:middlewares].unshift(::Instana::Instrumentation::Excon)
 end
-

data/lib/instana/instrumentation/graphql.rb

@@ -1,5 +1,8 @@
-if defined?(GraphQL::Schema) && defined?(GraphQL::Tracing::PlatformTracing) && ::Instana.config[:graphql][:enabled]
-  module Instana
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+module Instana
+  module Instrumentation
     class GraphqlTracing < GraphQL::Tracing::PlatformTracing
       self.platform_keys = {
         'lex' => 'lex.graphql',
@@ -11,26 +14,26 @@ if defined?(GraphQL::Schema) && defined?(GraphQL::Tracing::PlatformTracing) && :
         'execute_query' => 'execute.graphql',
         'execute_query_lazy' => 'execute.graphql',
       }
-      
-      def platform_trace(platform_key, key, data) 
-        return yield unless key == 'execute_query'    
+
+      def platform_trace(platform_key, key, data)
+        return yield unless key == 'execute_query'
         operation = data[:query].selected_operation
-            
+
         arguments = []
         fields = []
-        
+
         operation.selections.each do |field|
           arguments.concat(walk_fields(field, :arguments))
-          fields.concat(walk_fields(field, :selections))    
+          fields.concat(walk_fields(field, :selections))
         end
-            
+
         payload = {
           operationName: data[:query].operation_name || 'anonymous',
           operationType: operation.operation_type,
           arguments: grouped_fields(arguments),
           fields: grouped_fields(fields),
         }
-                
+
         begin
           ::Instana.tracer.log_entry(:'graphql.server')
           yield
@@ -41,11 +44,11 @@ if defined?(GraphQL::Schema) && defined?(GraphQL::Tracing::PlatformTracing) && :
           ::Instana.tracer.log_exit(:'graphql.server', {graphql: payload})
         end
       end
-    
+
       def platform_field_key(type, field)
         "#{type.graphql_name}.#{field.graphql_name}"
       end
-      
+
       def platform_authorized_key(type)
         "#{type.graphql_name}.authorized.graphql"
       end
@@ -53,17 +56,17 @@ if defined?(GraphQL::Schema) && defined?(GraphQL::Tracing::PlatformTracing) && :
       def platform_resolve_type_key(type)
         "#{type.graphql_name}.resolve_type.graphql"
       end
-      
+
       private
-      
+
       def walk_fields(parent, method)
         return [] unless parent.respond_to?(method)
-        
+
         parent.send(method).map do |field|
           [{object: parent.name, field: field.name}] + walk_fields(field, method)
         end.flatten
       end
-      
+
       def grouped_fields(fields)
         fields
           .group_by { |p| p[:object] }
@@ -72,6 +75,4 @@ if defined?(GraphQL::Schema) && defined?(GraphQL::Tracing::PlatformTracing) && :
       end
     end
   end
-  
-  ::GraphQL::Schema.use(::Instana::GraphqlTracing)
 end

data/lib/instana/instrumentation/grpc.rb

@@ -1,7 +1,10 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2017
+
 call_types = [:request_response, :client_streamer, :server_streamer, :bidi_streamer]
 
-if defined?(GRPC::ActiveCall) && ::Instana.config[:grpc][:enabled]
-  module Instana
+module Instana
+  module Instrumentation
     module GRPCCientInstrumentation
       CALL_TYPES = [:request_response, :client_streamer, :server_streamer, :bidi_streamer]
 
@@ -42,13 +45,10 @@ if defined?(GRPC::ActiveCall) && ::Instana.config[:grpc][:enabled]
       end
     end
   end
-
-  ::Instana.logger.debug 'Instrumenting GRPC client'
-  ::GRPC::ClientStub.prepend(::Instana::GRPCCientInstrumentation)
 end
 
-if defined?(GRPC::RpcDesc) && ::Instana.config[:grpc][:enabled]
-  module Instana
+module Instana
+  module Instrumentation
     module GRPCServerInstrumentation
       CALL_TYPES = [:request_response, :client_streamer, :server_streamer, :bidi_streamer]
 
@@ -88,7 +88,4 @@ if defined?(GRPC::RpcDesc) && ::Instana.config[:grpc][:enabled]
       end
     end
   end
-
-  ::Instana.logger.debug 'Instrumenting GRPC server'
-  ::GRPC::RpcDesc.prepend(::Instana::GRPCServerInstrumentation)
 end