lib/rails/tracer.rb in rails-tracer-0.4.3 vs lib/rails/tracer.rb in rails-tracer-0.5.0

@@ -1,4 +1,25 @@
 require "rails/span_helpers"
 require "rails/rack/tracer"
 require "rails/active_record/tracer"
 require "rails/active_support/cache/tracer"
+
+module Rails
+  module Tracer
+    class << self
+      def instrument(tracer: OpenTracing.global_tracer, active_span: nil,
+                     rack: false, middlewares: Rails.configuration.middleware,
+                     active_record: true,
+                     active_support_cache: true, dalli: false)
+        Rails::Rack::Tracer.instrument(tracer: tracer, middlewares: middlewares) if rack
+        ActiveRecord::Tracer.instrument(tracer: tracer, active_span: active_span) if active_record
+        ActiveSupport::Cache::Tracer.instrument(tracer: tracer, active_span: active_span, dalli: dalli) if active_support_cache
+      end
+
+      def disable
+        ActiveRecord::Tracer.disable
+        ActiveSupport::Cache::Tracer.disable
+        Rails::Rack::Tracer.disable
+      end
+    end
+  end
+end