lib/skylight.rb in skylight-5.3.5 vs lib/skylight.rb in skylight-6.0.0.beta

@@ -200,20 +200,21 @@
       instrumenter.instrument(category, title, desc, meta, &block)
     end
 
     instrumenter_method :config
 
-    instrumenter_method :mute, block: true
-    instrumenter_method :unmute, block: true
-    instrumenter_method :muted?
+    instrumenter_method :mute, wrapped_block: true
+    instrumenter_method :unmute, wrapped_block: true
+    instrumenter_method :tracing_muted?
+    instrumenter_method :endpoint_assignment_muted?
 
     # End a span
     instrumenter_method :done
 
     instrumenter_method :broken!
 
     # Temporarily disable
-    instrumenter_method :disable, block: true
+    instrumenter_method :disable, wrapped_block: true
 
     # Runs the shutdown procedure in the background.
     # This should do little more than unsubscribe from all ActiveSupport::Notifications
     def spawn_shutdown_thread!
       @shutdown_thread || const_get(:LOCK).synchronize { @shutdown_thread ||= Thread.new { @instrumenter&.shutdown } }