instana 1.12.0 → 1.193.3.pre1

data/lib/instana/frameworks/rails.rb

@@ -1,7 +1,6 @@
 require "instana/rack"
 
 if defined?(::Rails)
-
   if ::Rails::VERSION::MAJOR < 3
     ::Rails.configuration.after_initialize do
       # In Rails, let's use the Rails logger

data/lib/instana/frameworks/roda.rb

@@ -4,11 +4,11 @@ module Instana
   module RodaPathTemplateExtractor
     module RequestMethods
       TERM = defined?(::Roda) ? ::Roda::RodaPlugins::Base::RequestMethods::TERM : Object
-      
+
       def if_match(args, &blk)
         path = @remaining_path
         captures = @captures.clear
-  
+
         if match_all(args)
           (env['INSTANA_PATH_TEMPLATE_FRAGMENTS'] ||= []).concat(named_args(args, blk))
           block_result(blk.(*captures))
@@ -21,10 +21,10 @@ module Instana
           false
         end
       end
-      
-      def named_args(args, blk)  
-        parameters = blk.parameters      
-        args.map do |a| 
+
+      def named_args(args, blk)
+        parameters = blk.parameters
+        args.map do |a|
           case a
           when String
             a
@@ -35,13 +35,7 @@ module Instana
             "{#{name}}"
           end
         end.compact
-      end  
+      end
     end
   end
 end
-
-if defined?(::Roda)
-  ::Instana.logger.debug "Instrumenting Roda"
-  Roda.use ::Instana::Rack
-  Roda.plugin ::Instana::RodaPathTemplateExtractor
-end

data/lib/instana/frameworks/sinatra.rb

@@ -6,10 +6,9 @@ require "instana/rack"
 module Instana
   module SinatraPathTemplateExtractor
     def self.extended(base)
-      ::Instana.logger.debug "#{base} extended #{self}"
       base.store_path_template
     end
-    
+
     def store_path_template
       after do
         @env["INSTANA_HTTP_PATH_TEMPLATE"] = @env["sinatra.route"]
@@ -18,9 +17,3 @@ module Instana
     end
   end
 end
-
-if defined?(::Sinatra)
-  ::Instana.logger.debug "Instrumenting Sinatra"
-  ::Sinatra::Base.use ::Instana::Rack
-  ::Sinatra::Base.register ::Instana::SinatraPathTemplateExtractor
-end

data/lib/instana/instrumentation/dalli.rb

@@ -1,15 +1,10 @@
 module Instana
   module Instrumentation
     module Dalli
-      def self.included(klass)
-        ::Instana::Util.method_alias(klass, :perform)
-        ::Instana::Util.method_alias(klass, :get_multi)
-      end
-
-      def perform_with_instana(*args, &blk)
+      def perform(*args, &blk)
         if !::Instana.tracer.tracing? || ::Instana.tracer.tracing_span?(:memcache)
           do_skip = true
-          return perform_without_instana(*args, &blk)
+          return super(*args, &blk)
         end
 
         op, key, *_opts = args
@@ -22,7 +17,7 @@ module Instana
         ::Instana.tracer.log_entry(:memcache, entry_payload)
         exit_payload = { :memcache => {} }
 
-        result = perform_without_instana(*args, &blk)
+        result = super(*args, &blk)
 
         if op == :get
           exit_payload[:memcache][:hit] = result ? 1 : 0
@@ -36,7 +31,7 @@ module Instana
         ::Instana.tracer.log_exit(:memcache, exit_payload) unless do_skip
       end
 
-      def get_multi_with_instana(*keys)
+      def get_multi(*keys)
         entry_payload = { :memcache => {} }
         entry_payload[:memcache][:namespace] = @options[:namespace] if @options.key?(:namespace)
         entry_payload[:memcache][:command] = :get_multi
@@ -45,7 +40,7 @@ module Instana
         ::Instana.tracer.log_entry(:memcache, entry_payload)
         exit_payload = { :memcache => {} }
 
-        result = get_multi_without_instana(*keys)
+        result = super(*keys)
 
         exit_payload[:memcache][:hits] = result.length
         result
@@ -63,20 +58,14 @@ module Instana
         ::Instana::Util.method_alias(klass, :request)
       end
 
-      def request_with_instana(op, *args)
+      def request(op, *args)
         if ::Instana.tracer.tracing? || ::Instana.tracer.tracing_span?(:memcache)
           info_payload = { :memcache => {} }
           info_payload[:memcache][:server] = "#{@hostname}:#{@port}"
           ::Instana.tracer.log_info(info_payload)
         end
-        request_without_instana(op, *args)
+        super(op, *args)
       end
     end
   end
 end
-
-if defined?(::Dalli) && ::Instana.config[:dalli][:enabled]
-  ::Instana.logger.debug "Instrumenting Dalli"
-  ::Dalli::Client.send(:include, ::Instana::Instrumentation::Dalli)
-  ::Dalli::Server.send(:include, ::Instana::Instrumentation::DalliServer)
-end

data/lib/instana/instrumentation/excon.rb

@@ -1,74 +1,68 @@
-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?
+module Instana
+  module Instrumentation
+    class Excon < ::Excon::Middleware::Base
+      def request_call(datum)
+        return @stack.request_call(datum) unless ::Instana.tracer.tracing?
 
-          payload = { :http => {} }
-          path = datum[:path].split('?').first
-          payload[:http][:url] = "#{datum[:connection].instance_variable_get(:@socket_key)}#{path}"
-          payload[:http][:method] = datum[:method] if datum.key?(:method)
+        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)
 
-          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
+        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
 
-          # 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
+        # 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
+        @stack.request_call(datum)
+      end
 
-        def error_call(datum)
-          return @stack.error_call(datum) unless ::Instana.tracer.tracing?
+      def error_call(datum)
+        return @stack.error_call(datum) unless ::Instana.tracer.tracing?
 
-          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)
+        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?
 
-          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

@@ -0,0 +1,75 @@
+module Instana
+  module Instrumentation
+    class GraphqlTracing < GraphQL::Tracing::PlatformTracing
+      self.platform_keys = {
+        'lex' => 'lex.graphql',
+        'parse' => 'parse.graphql',
+        'validate' => 'validate.graphql',
+        'analyze_query' => 'analyze.graphql',
+        'analyze_multiplex' => 'analyze.graphql',
+        'execute_multiplex' => 'execute.graphql',
+        'execute_query' => 'execute.graphql',
+        'execute_query_lazy' => 'execute.graphql',
+      }
+
+      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))
+        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
+        rescue Exception => e
+          ::Instana.tracer.log_error(e)
+          raise e
+        ensure
+          ::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
+
+      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] }
+          .map { |name, p| [name, p.map { |f| f[:field] }] }
+          .to_h
+      end
+    end
+  end
+end

data/lib/instana/instrumentation/grpc.rb

@@ -1,84 +1,88 @@
 call_types = [:request_response, :client_streamer, :server_streamer, :bidi_streamer]
 
-if defined?(GRPC::ActiveCall) && ::Instana.config[:grpc][:enabled]
-  call_types.each do |call_type|
-    GRPC::ClientStub.class_eval <<-RUBY, __FILE__, __LINE__ + 1
-      def #{call_type}_with_instana(method, *others, **options)
-        kvs = { rpc: {} }
+module Instana
+  module Instrumentation
+    module GRPCCientInstrumentation
+      CALL_TYPES = [:request_response, :client_streamer, :server_streamer, :bidi_streamer]
 
-        unless ::Instana.tracer.tracing?
-          return #{call_type}_without_instana(method, *others, **options)
-        end
+      CALL_TYPES.each do |call_type|
+        define_method(call_type) do |method, *others, **options|
+          begin
+            kvs = { rpc: {} }
 
-        kvs[:rpc][:flavor] = :grpc
-        kvs[:rpc][:host] = @host
-        kvs[:rpc][:call] = method
-        kvs[:rpc][:call_type] = :#{call_type}
+            unless ::Instana.tracer.tracing?
+              return super(method, *others, **options)
+            end
 
-        ::Instana.tracer.log_entry(:'rpc-client', kvs)
+            kvs[:rpc][:flavor] = :grpc
+            kvs[:rpc][:host] = @host
+            kvs[:rpc][:call] = method
+            kvs[:rpc][:call_type] = call_type
 
-        context = ::Instana.tracer.context
-        if context
-          options[:metadata] = (options[:metadata] || {}).merge(
-            'x-instana-t' => context.trace_id_header,
-            'x-instana-s' => context.span_id_header
-          )
-        end
+            ::Instana.tracer.log_entry(:'rpc-client', kvs)
 
-        #{call_type}_without_instana(method, *others, **options)
-      rescue => e
-        kvs[:rpc][:error] = true
-        ::Instana.tracer.log_info(kvs)
-        ::Instana.tracer.log_error(e)
-        raise
-      ensure
-        ::Instana.tracer.log_exit(:'rpc-client', {})
-      end
+            context = ::Instana.tracer.context
+            if context
+              options[:metadata] = (options[:metadata] || {}).merge(
+                'x-instana-t' => context.trace_id_header,
+                'x-instana-s' => context.span_id_header
+              )
+            end
 
-      alias #{call_type}_without_instana #{call_type}
-      alias #{call_type} #{call_type}_with_instana
-    RUBY
+            super(method, *others, **options)
+          rescue => e
+            kvs[:rpc][:error] = true
+            ::Instana.tracer.log_info(kvs)
+            ::Instana.tracer.log_error(e)
+            raise
+          ensure
+            ::Instana.tracer.log_exit(:'rpc-client', {})
+          end
+        end
+      end
+    end
   end
-  ::Instana.logger.debug 'Instrumenting GRPC client'
 end
 
-if defined?(GRPC::RpcDesc) && ::Instana.config[:grpc][:enabled]
-  call_types.each do |call_type|
-    GRPC::RpcDesc.class_eval <<-RUBY, __FILE__, __LINE__ + 1
-      def handle_#{call_type}_with_instana(active_call, mth, *others)
-        kvs = { rpc: {} }
-        metadata = active_call.metadata
+module Instana
+  module Instrumentation
+    module GRPCServerInstrumentation
+      CALL_TYPES = [:request_response, :client_streamer, :server_streamer, :bidi_streamer]
 
-        incoming_context = {}
-        if metadata.key?('x-instana-t')
-          incoming_context[:trace_id]  = ::Instana::Util.header_to_id(metadata['x-instana-t'])
-          incoming_context[:span_id]   = ::Instana::Util.header_to_id(metadata['x-instana-s']) if metadata.key?('x-instana-s')
-          incoming_context[:level]     = metadata['x-instana-l'] if metadata.key?('x-instana-l')
-        end
+      CALL_TYPES.each do |call_type|
+        define_method(:"handle_#{call_type}") do |active_call, mth, *others|
+          begin
+            kvs = { rpc: {} }
+            metadata = active_call.metadata
 
-        kvs[:rpc][:flavor] = :grpc
-        kvs[:rpc][:host] = Socket.gethostname
-        kvs[:rpc][:call] = "/\#{mth.owner.service_name}/\#{name}"
-        kvs[:rpc][:call_type] = :#{call_type}
-        kvs[:rpc][:peer] = { address: active_call.peer }
+            incoming_context = {}
+            if metadata.key?('x-instana-t')
+              incoming_context[:trace_id]  = ::Instana::Util.header_to_id(metadata['x-instana-t'])
+              incoming_context[:span_id]   = ::Instana::Util.header_to_id(metadata['x-instana-s']) if metadata.key?('x-instana-s')
+              incoming_context[:level]     = metadata['x-instana-l'] if metadata.key?('x-instana-l')
+            end
 
-        ::Instana.tracer.log_start_or_continue(
-          :'rpc-server', kvs, incoming_context
-        )
+            kvs[:rpc][:flavor] = :grpc
+            kvs[:rpc][:host] = Socket.gethostname
+            kvs[:rpc][:call] = "/#{mth.owner.service_name}/#{name}"
+            kvs[:rpc][:call_type] = call_type
+            kvs[:rpc][:peer] = { address: active_call.peer }
 
-        handle_#{call_type}_without_instana(active_call, mth, *others)
-      rescue => e
-        kvs[:rpc][:error] = true
-        ::Instana.tracer.log_info(kvs)
-        ::Instana.tracer.log_error(e)
-        raise
-      ensure
-        ::Instana.tracer.log_end(:'rpc-server', {}) if ::Instana.tracer.tracing?
-      end
+            ::Instana.tracer.log_start_or_continue(
+              :'rpc-server', kvs, incoming_context
+            )
 
-      alias handle_#{call_type}_without_instana handle_#{call_type}
-      alias handle_#{call_type} handle_#{call_type}_with_instana
-    RUBY
+            super(active_call, mth, *others)
+          rescue => e
+            kvs[:rpc][:error] = true
+            ::Instana.tracer.log_info(kvs)
+            ::Instana.tracer.log_error(e)
+            raise
+          ensure
+            ::Instana.tracer.log_end(:'rpc-server', {}) if ::Instana.tracer.tracing?
+          end
+        end
+      end
+    end
   end
-  ::Instana.logger.debug 'Instrumenting GRPC server'
 end