gitlab-org--gitlab-foss/lib/gitlab/usage_data_counters/base_counter.rb
Mark Chao 4a6f959ab8 Record usage on snippet usage
Generalize wiki page counter for other page types to extend to.
2019-08-08 13:18:57 +00:00

39 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