lib/ddtrace/tracer.rb in ddtrace-0.51.1 vs lib/ddtrace/tracer.rb in ddtrace-0.52.0

@@ -1,5 +1,6 @@
+# typed: true
 require 'logger'
 require 'pathname'
 
 require 'ddtrace/ext/environment'
 require 'ddtrace/span'
@@ -82,17 +83,16 @@
                          Datadog::ContextFlush::Finished.new
                        end
 
       @default_service = options[:default_service]
       @enabled = options.fetch(:enabled, true)
-      @provider = options.fetch(:context_provider, Datadog::DefaultContextProvider.new)
+      @provider = options[:context_provider] || Datadog::DefaultContextProvider.new
       @sampler = options.fetch(:sampler, Datadog::AllSampler.new)
       @tags = options.fetch(:tags, {})
       @writer = options.fetch(:writer) { Datadog::Writer.new }
 
       # Instance variables
       @mutex = Mutex.new
-      @provider ||= Datadog::DefaultContextProvider.new # @provider should never be nil
 
       # Enable priority sampling by default
       activate_priority_sampling!(@sampler)
     end