Suppress internal spans by default

Author: jterapin

instrumentation/aws_sdk/Gemfile

@@ -22,6 +22,11 @@ group :test do
   gem 'rspec-mocks'
   gem 'opentelemetry-instrumentation-base', path: '../base'
   gem 'webrick', '~> 1.7'
+  gem 'aws-sdk-core', '~> 3'
+  gem 'aws-sdk-lambda', '~> 1'
+  gem 'aws-sdk-dynamodb', '~> 1'
+  gem 'aws-sdk-sns', '~> 1'
+  gem 'aws-sdk-sqs', '~> 1'
   if RUBY_VERSION >= '3.4'
     gem 'base64'
     gem 'mutex_m'

instrumentation/aws_sdk/README.md

@@ -20,7 +20,7 @@ To install the instrumentation, call `use` with the name of the instrumentation.
 OpenTelemetry::SDK.configure do |c|
   c.use 'OpenTelemetry::Instrumentation::AwsSdk', {
     inject_messaging_context: true,
-    suppress_internal_instrumentation: true
+	enable_internal_instrumentation: true
   }
 end
 ```
@@ -36,8 +36,10 @@ end
 This instrumentation offers the following configuration options: 
 * `:inject_messaging_context` (default: `false`): When set to `true`, adds context key/value 
  to Message Attributes for SQS/SNS messages.
-* `suppress_internal_instrumentation` (default: `false`): When set to `true`, any spans with 
- span kind of `internal` are suppressed from traces.
+* `:enable_internal_instrumentation` (default: `false`): When set to `true`, any spans with 
+ span kind of `internal` are traced.
+* `:suppress_internal_instrumentation`: **Deprecated**. This configuration has been 
+ deprecated in a favor of `:enable_internal_instrumentation`
 
 ## Integration with SDK V3's Telemetry support
 AWS SDK for Ruby V3 added support for Observability which includes a new configuration, 

instrumentation/aws_sdk/lib/opentelemetry/instrumentation/aws_sdk/handler.rb

@@ -23,7 +23,7 @@ def call(context)
           ) do |span|
             MessagingHelper.inject_context_if_supported(context, client_method, service_id)
 
-            if HandlerHelper.instrumentation_config[:suppress_internal_instrumentation]
+            if HandlerHelper.skip_internal_instrumentation?
               OpenTelemetry::Common::Utilities.untraced { super }
             else
               super
@@ -32,7 +32,6 @@ def call(context)
                 OpenTelemetry::SemanticConventions::Trace::HTTP_STATUS_CODE,
                 context.http_response.status_code
               )
-
               if (err = response.error)
                 span.record_exception(err)
                 span.status = Trace::Status.error(err.to_s)

instrumentation/aws_sdk/lib/opentelemetry/instrumentation/aws_sdk/handler_helper.rb

@@ -14,6 +14,10 @@ def instrumentation_config
             AwsSdk::Instrumentation.instance.config
           end
 
+          def skip_internal_instrumentation?
+            instrumentation_config[:enable_internal_instrumentation] == false
+          end
+
           def client_method(service_id, context)
             "#{service_id}.#{context.operation.name}".delete(' ')
           end

instrumentation/aws_sdk/lib/opentelemetry/instrumentation/aws_sdk/instrumentation.rb

@@ -19,18 +19,20 @@ module AwsSdk
       # - `false` **(default)** - Context key/value will not be added.
       # - `true` - Context key/value will be added.
       #
-      # ### `:suppress_internal_instrumentation`
+      # ### `:enable_internal_instrumentation`
+      # Enables tracing of spans of `internal` span kind.
       #
-      # Disables tracing of spans of `internal` span kind.
+      # - `false` **(default)** - Internal spans are not traced
+      # - `true` - Internal spans are traced.
       #
-      # - `false` **(default)** - Internal spans are traced.
-      # - `true` - Internal spans are not traced.
+      # ### `:suppress_internal_instrumentation` (deprecated)
+      # This configuration has been deprecated in a favor of `:enable_internal_instrumentation`
       #
-      # @example An explicit default configuration
+      # @example An explicit default configurations
       #   OpenTelemetry::SDK.configure do |c|
       #     c.use 'OpenTelemetry::Instrumentation::AwsSdk', {
       #       inject_messaging_context: false,
-      #       suppress_internal_instrumentation: false
+      #       enable_internal_instrumentation: false
       #     }
       #   end
       class Instrumentation < OpenTelemetry::Instrumentation::Base
@@ -51,7 +53,7 @@ class Instrumentation < OpenTelemetry::Instrumentation::Base
         end
 
         option :inject_messaging_context, default: false, validate: :boolean
-        option :suppress_internal_instrumentation, default: false, validate: :boolean
+        option :enable_internal_instrumentation, default: false, validate: :boolean
 
         def gem_version
           if Gem.loaded_specs['aws-sdk']

instrumentation/aws_sdk/lib/opentelemetry/instrumentation/aws_sdk/patches/telemetry.rb

@@ -26,7 +26,7 @@ def span_wrapper(context, &)
             ) do |span|
               MessagingHelper.inject_context_if_supported(context, client_method, service_id)
 
-              if HandlerHelper.instrumentation_config[:suppress_internal_instrumentation]
+              if HandlerHelper.skip_internal_instrumentation?
                 OpenTelemetry::Common::Utilities.untraced { super }
               else
                 yield span

instrumentation/aws_sdk/test/opentelemetry/handler_test.rb

@@ -4,7 +4,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-require 'test_helper'
+require_relative '../test_helper'
 
 describe OpenTelemetry::Instrumentation::AwsSdk do
   describe 'AwsSdk Plugin' do

instrumentation/aws_sdk/test/opentelemetry/instrumentation_test.rb

@@ -65,4 +65,11 @@
       instrumentation.instance_variable_set(:@installed, false)
     end
   end
+
+  describe '#install with default options' do
+    it 'with default options' do
+      _(instrumentation.config[:inject_messaging_context]).must_equal(false)
+      _(instrumentation.config[:enable_internal_instrumentation]).must_equal(false)
+    end
+  end
 end