lib/cassandra_object/log_subscriber.rb
module CassandraObject
class LogSubscriber < ActiveSupport::LogSubscriber
def self.runtime=(value)
Thread.current["cassandra_object_request_runtime"] = value
end
def self.runtime
Thread.current["cassandra_object_request_runtime"] ||= 0
end
def self.reset_runtime
rt, self.runtime = runtime, 0
rt
end
def initialize
super
@odd_or_even = false
end
def cql(event)
self.class.runtime += event.duration
payload = event.payload
name = '%s (%.1fms)' % [payload[:name], event.duration]
cql = payload[:cql].squeeze(' ')
if odd?
name = color(name, CYAN, true)
cql = color(cql, nil, true)
else
name = color(name, MAGENTA, true)
end
debug " #{name} #{cql}"
end
def odd?
@odd_or_even = !@odd_or_even
end
end
end
CassandraObject::LogSubscriber.attach_to :cassandra_object