instana 1.193.4 → 1.195.2

data/lib/instana/activators/resque_client.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
 module Instana
   module Activators
     class ResqueClient < Activator

data/lib/instana/activators/resque_worker.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
 module Instana
   module Activators
     class ResqueWorker < Activator

data/lib/instana/activators/rest_client.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
 module Instana
   module Activators
     class RestClient < Activator

data/lib/instana/activators/roda.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
 module Instana
   module Activators
     class Roda < Activator

data/lib/instana/activators/shoryuken.rb

@@ -0,0 +1,24 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+module Instana
+  module Activators
+    class Shoryuken < Activator
+      def can_instrument?
+        defined?(::Shoryuken) && ::Shoryuken.respond_to?(:configure_server)
+      end
+
+      def instrument
+        require 'instana/instrumentation/shoryuken'
+
+        ::Shoryuken.configure_server do |config|
+          config.server_middleware do |chain|
+            chain.add ::Instana::Instrumentation::Shoryuken
+          end
+        end
+
+        true
+      end
+    end
+  end
+end

data/lib/instana/activators/sidekiq_client.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
 module Instana
   module Activators
     class SidekiqClient < Activator

data/lib/instana/activators/sidekiq_worker.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
 module Instana
   module Activators
     class SidekiqWorker < Activator

data/lib/instana/activators/sinatra.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
 module Instana
   module Activators
     class Sinatra < Activator

data/lib/instana/agent.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 require 'net/http'
 require 'socket'
 require 'sys/proctable'

data/lib/instana/agent/helpers.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2018
+
 module AgentHelpers
   # Attempts to determine if we're running inside a container.
   # The qualifications are:

data/lib/instana/agent/hooks.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2018
+
 module AgentHooks
   # Used post fork to re-initialize state and restart communications with
   # the host agent.

data/lib/instana/agent/tasks.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2018
+
 module AgentTasks
   # When request(s) are received by the host agent, it is sent here
   # for handling & processing.

data/lib/instana/base.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 require "logger"
 require "instana/version"
 require "instana/util"

data/lib/instana/collector.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 module Instana
   class Collector
     attr_accessor :collectors

data/lib/instana/collectors/gc.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 module Instana
   module Collectors
     class GC

data/lib/instana/collectors/memory.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 require 'get_process_mem'
 
 module Instana

data/lib/instana/collectors/thread.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 module Instana
   module Collectors
     class Thread

data/lib/instana/config.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 module Instana
   class Config
 
@@ -52,6 +55,9 @@ module Instana
       # ::Instana.config[:sanitize_sql] = false
       @config[:sanitize_sql] = true
 
+      # W3 Trace Context Support
+      @config[:w3_trace_correlation] = ENV.fetch('INSTANA_W3C_TRACE_CORRELATION', 'true').eql?('true')
+
       @config[:action_controller]  = { :enabled => true }
       @config[:action_view]        = { :enabled => true }
       @config[:active_record]      = { :enabled => true }

data/lib/instana/frameworks/cuba.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 require "instana/rack"
 
 module Instana

data/lib/instana/frameworks/rails.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 if ::Rails::VERSION::MAJOR < 3
   ::Rails.configuration.after_initialize do
     # In Rails, let's use the Rails logger

data/lib/instana/frameworks/roda.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 require "instana/rack"
 
 module Instana

data/lib/instana/frameworks/sinatra.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 require "instana/rack"
 
 # This instrumentation will insert Rack into Sinatra _and_ Padrino since

data/lib/instana/helpers.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 module Instana
   module Helpers
     EUM_SNIPPET= (File.read(File.dirname(__FILE__) + '/eum/eum.js.erb')).freeze

data/lib/instana/instrumentation/action_cable.rb

@@ -0,0 +1,53 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+module Instana
+  module Instrumentation
+    module ActionCableConnection
+      def instana_trace_context
+        @instana_trace_context
+      end
+
+      def process
+        @instana_trace_context ||= ::Instana.tracer.tracing? ? ::Instana.tracer.current_span.context : {}
+        super
+      end
+    end
+
+    module ActionCableChannel
+      def transmit(*args)
+        rpc_tags = {
+          service: ::Instana::Util.get_app_name,
+          rpc: {
+            flavor: :actioncable,
+            call: self.class.to_s,
+            call_type: :transmit,
+            host: Socket.gethostname
+          }
+        }
+
+        context = connection.instana_trace_context
+        ::Instana.tracer.start_or_continue_trace(:'rpc-server', rpc_tags, context) do
+          super(*args)
+        end
+      end
+
+      def dispatch_action(action, data)
+        rpc_tags = {
+          service: ::Instana::Util.get_app_name,
+          rpc: {
+            flavor: :actioncable,
+            call: "#{self.class}##{action}",
+            call_type: :action,
+            host: Socket.gethostname
+          }
+        }
+
+        context = connection.instana_trace_context
+        ::Instana.tracer.start_or_continue_trace(:'rpc-server', rpc_tags, context) do
+          super(action, data)
+        end
+      end
+    end
+  end
+end

data/lib/instana/instrumentation/action_controller.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
 module Instana
   module Instrumentation
     module ActionController

data/lib/instana/instrumentation/action_view.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
 module Instana
   module Instrumentation
     module ActionView

data/lib/instana/instrumentation/active_record.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
 module Instana
   module Instrumentation
     module ActiveRecord

data/lib/instana/instrumentation/aws_sdk_dynamodb.rb

@@ -0,0 +1,48 @@
+# (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: format_operation(context.operation_name),
+            table: table_name_from(context)
+          }
+
+          ::Instana.tracer.trace(:dynamodb, {dynamodb: dynamo_tags}) { @handler.call(context) }
+        end
+
+        private
+
+        def table_name_from(context)
+          context.params[:table_name] || context.params[:global_table_name] || 'Unknown'
+        end
+
+        def format_operation(name)
+          case name
+          when :create_table
+            'create'
+          when :list_tables
+            'list'
+          when :get_item
+            'get'
+          when :put_item
+            'put'
+          when :update_item
+            'update'
+          when :delete_item
+            'delete'
+          else
+            name.to_s
+          end
+        end
+      end
+
+      def add_handlers(handlers, _config)
+        handlers.add(Handler, step: :initialize)
+      end
+    end
+  end
+end

data/lib/instana/instrumentation/aws_sdk_s3.rb

@@ -0,0 +1,55 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+module Instana
+  module Instrumentation
+    class S3 < Seahorse::Client::Plugin
+      class Handler < Seahorse::Client::Handler
+        def call(context)
+          s3_tags = {
+            op: format_operation(context.operation_name),
+            bucket: bucket_name_from(context),
+            key: key_from_context(context)
+          }.compact
+
+          ::Instana.tracer.trace(:s3, {s3: s3_tags}) { @handler.call(context) }
+        end
+
+        private
+
+        def bucket_name_from(context)
+          context.params[:bucket] || 'Unknown'
+        end
+
+        def key_from_context(context)
+          context.params[:key]
+        end
+
+        def format_operation(name)
+          case name
+          when :create_bucket
+            'createBucket'
+          when :delete_bucket
+            'deleteBucket'
+          when :delete_object
+            'delete'
+          when :get_object
+            'get'
+          when :head_object
+            'metadata'
+          when :list_objects
+            'list'
+          when :put_object
+            'list'
+          else
+            name.to_s
+          end
+        end
+      end
+
+      def add_handlers(handlers, _config)
+        handlers.add(Handler, step: :initialize)
+      end
+    end
+  end
+end

data/lib/instana/instrumentation/aws_sdk_sns.rb

@@ -0,0 +1,29 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+module Instana
+  module Instrumentation
+    class SNS < Seahorse::Client::Plugin
+      class Handler < Seahorse::Client::Handler
+        def call(context)
+          sns_tags = {
+            topic: context.params[:topic_arn],
+            target: context.params[:target_arn],
+            phone: context.params[:phone_number],
+            subject: context.params[:subject]
+          }.compact
+
+          if context.operation_name == :publish
+            ::Instana.tracer.trace(:sns, {sns: sns_tags}) { @handler.call(context) }
+          else
+            @handler.call(context)
+          end
+        end
+      end
+
+      def add_handlers(handlers, _config)
+        handlers.add(Handler, step: :initialize)
+      end
+    end
+  end
+end

data/lib/instana/instrumentation/aws_sdk_sqs.rb

@@ -0,0 +1,98 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+module Instana
+  module Instrumentation
+    class SQS < Seahorse::Client::Plugin
+      class Handler < Seahorse::Client::Handler
+        SPAN_FORMING_OPERATIONS = [:send_message, :send_message_batch, :get_queue_url, :create_queue, :delete_message, :delete_message_batch].freeze
+
+        def call(context)
+          is_tracing = ::Instana.tracer.tracing?
+          unless is_tracing && SPAN_FORMING_OPERATIONS.include?(context.operation_name)
+            return @handler.call(context)
+          end
+
+          span_tags = tags_for(context.operation_name, context.params).compact
+
+          ::Instana.tracer.trace(:sqs, {sqs: span_tags}) do |span|
+            case context.operation_name
+            when :send_message
+              inject_instana_headers(span, context.params)
+            when :send_message_batch
+              context.params[:entries].each { |e| inject_instana_headers(span, e) }
+            end
+
+            response = @handler.call(context)
+
+            span_tags[:queue] = response.queue_url if response.respond_to?(:queue_url)
+            span.set_tags(sqs: span_tags)
+
+            response
+          end
+        end
+
+        private
+
+        def inject_instana_headers(span, params)
+          params[:message_attributes] ||= {}
+          params[:message_attributes].merge!({
+                                               "X_INSTANA_T" => {data_type: 'String', string_value: span.context.trace_id},
+                                               "X_INSTANA_S" => {data_type: 'String', string_value: span.context.span_id},
+                                               "X_INSTANA_L" => {data_type: 'String', string_value: span.context.level.to_s}
+                                             })
+        end
+
+        def tags_for(operation_name, params)
+          case operation_name
+          when :create_queue
+            {
+              sort: 'exit',
+              type: 'create.queue',
+              queue: params[:queue_name]
+            }
+          when :get_queue_url
+            {
+              sort: 'exit',
+              type: 'get.queue',
+              queue: params[:queue_name]
+            }
+          when :send_message
+            {
+              sort: 'exit',
+              type: 'single.sync',
+              queue: params[:queue_url],
+              group: params[:message_group_id]
+            }
+          when :send_message_batch
+            {
+              sort: 'exit',
+              type: 'single.sync',
+              queue: params[:queue_url],
+              size: params[:entries].count
+            }
+          when :delete_message
+            {
+              sort: 'exit',
+              type: 'delete.single.sync',
+              queue: params[:queue_url]
+            }
+          when :delete_message_batch
+            {
+              sort: 'exit',
+              type: 'delete.batch.sync',
+              queue: params[:queue_url],
+              size: params[:entries].count
+            }
+          else
+            {}
+          end
+        end
+      end
+
+      def add_handlers(handlers, _config)
+        handlers.add(Handler, step: :initialize)
+      end
+    end
+  end
+end