42 lines
1 KiB
Ruby
42 lines
1 KiB
Ruby
|
module Gitlab
|
||
|
|
module Metrics
|
||
|
|
module Subscribers
|
||
|
|
# Class for tracking the total time spent in Rails cache calls
|
||
|
|
class RailsCache < ActiveSupport::Subscriber
|
||
|
|
attach_to :active_support
|
||
|
|
|
||
|
|
def cache_read(event)
|
||
|
|
increment(:cache_read, event.duration)
|
||
|
|
end
|
||
|
|
|
||
|
|
def cache_write(event)
|
||
|
|
increment(:cache_write, event.duration)
|
||
|
|
end
|
||
|
|
|
||
|
|
def cache_delete(event)
|
||
|
|
increment(:cache_delete, event.duration)
|
||
|
|
end
|
||
|
|
|
||
|
|
def cache_exist?(event)
|
||
|
|
increment(:cache_exists, event.duration)
|
||
|
|
end
|
||
|
|
|
||
|
|
def increment(key, duration)
|
||
|
|
return unless current_transaction
|
||
|
|
|
||
|
|
current_transaction.increment(:cache_duration, duration)
|
||
|
|
current_transaction.increment(:cache_count, 1)
|
||
|
|
current_transaction.increment("#{key}_duration".to_sym, duration)
|
||
|
|
current_transaction.increment("#{key}_count".to_sym, 1)
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def current_transaction
|
||
|
|
Transaction.current
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|