module Datadog::Tracing::Contrib::ActionCable::Instrumentation::ActionCableChannel

def self.included(base)
  base.class_eval do
    set_callback(
      :subscribe,
      :around,
      ->(channel, block) { Tracer.trace(channel, :subscribe, &block) },
      prepend: true
    )
    set_callback(
      :unsubscribe,
      :around,
      ->(channel, block) { Tracer.trace(channel, :unsubscribe, &block) },
      prepend: true
    )
  end
end