debian-mirror-gitlab/lib/gitlab/usage_data_counters/base_counter.rb
2020-05-24 23:13:21 +05:30

47 lines
957 B
Ruby

# frozen_string_literal: true
module Gitlab::UsageDataCounters
class BaseCounter
extend RedisCounter
UnknownEvent = Class.new(StandardError)
class << self
def redis_key(event)
Gitlab::ErrorTracking.track_and_raise_for_dev_exception(UnknownEvent.new, 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 { |event| [counter_key(event), read(event)] }.to_h
end
def fallback_totals
known_events.map { |event| [counter_key(event), -1] }.to_h
end
private
def counter_key(event)
"#{prefix}_#{event}".to_sym
end
def known_events
self::KNOWN_EVENTS
end
def prefix
self::PREFIX
end
end
end
end