eprothro/cassie

module Cassie::ConnectionHandler::Sessions
  module Instrumentation

    protected

    def initialize_session(*args)
      Cassie.instrumenter.instrument("cassie.session.connect") do |payload|
        super(*args).tap do |session|
          payload[:keyspace] = session.keyspace
        end
      end
    end
  end
end