40 lines
754 B
Ruby
40 lines
754 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab::UsageDataCounters
|
||
|
|
class BaseCounter
|
||
|
|
extend RedisCounter
|
||
|
|
|
||
|
|
UnknownEvent = Class.new(StandardError)
|
||
|
|
|
||
|
|
class << self
|
||
|
|
def redis_key(event)
|
||
|
|
Gitlab::Sentry.track_exception(UnknownEvent, extra: { event: event }) unless known_events.include?(event.to_s)
|
||
|
|
|
||
|
|
"USAGE_#{prefix}_#{event}".upcase
|
||
|
|
end
|
||
|
|
|
||
|
|
def count(event)
|
||
|
|
increment(redis_key event)
|
||
|
|
end
|
||
|
|
|
||
|
|
def read(event)
|
||
|
|
total_count(redis_key event)
|
||
|
|
end
|
||
|
|
|
||
|
|
def totals
|
||
|
|
known_events.map { |e| ["#{prefix}_#{e}".to_sym, read(e)] }.to_h
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def known_events
|
||
|
|
self::KNOWN_EVENTS
|
||
|
|
end
|
||
|
|
|
||
|
|
def prefix
|
||
|
|
self::PREFIX
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|