debian-mirror-gitlab/lib/gitlab/metrics/sli.rb
2022-08-13 15:12:31 +05:30

112 lines
2.7 KiB
Ruby

# frozen_string_literal: true
module Gitlab
module Metrics
module Sli
COUNTER_PREFIX = 'gitlab_sli'
module ClassMethods
INITIALIZATION_MUTEX = Mutex.new
def [](name)
known_slis[name] || initialize_sli(name, [])
end
def initialize_sli(name, possible_label_combinations)
INITIALIZATION_MUTEX.synchronize do
next known_slis[name] if initialized?(name)
sli = new(name)
sli.initialize_counters(possible_label_combinations)
known_slis[name] = sli
end
end
def initialized?(name)
known_slis.key?(name) && known_slis[name].initialized?
end
private
def known_slis
@known_slis ||= {}
end
end
def self.included(mod)
mod.extend(ClassMethods)
end
attr_reader :name
def initialize(name)
@name = name
@initialized_with_combinations = false
end
def initialize_counters(possible_label_combinations)
# This module is effectively an abstract class
@initialized_with_combinations = possible_label_combinations.any? # rubocop:disable Gitlab/ModuleWithInstanceVariables
possible_label_combinations.each do |label_combination|
total_counter.get(label_combination)
numerator_counter.get(label_combination)
end
end
def increment(labels:, increment_numerator:)
total_counter.increment(labels)
numerator_counter.increment(labels) if increment_numerator
end
def initialized?
@initialized_with_combinations
end
private
def total_counter
prometheus.counter(counter_name('total'), "Total number of measurements for #{name}")
end
def prometheus
Gitlab::Metrics
end
class Apdex
include Sli
def increment(labels:, success:)
super(labels: labels, increment_numerator: success)
end
private
def counter_name(suffix)
[COUNTER_PREFIX, "#{name}_apdex", suffix].join('_').to_sym
end
def numerator_counter
prometheus.counter(counter_name('success_total'), "Number of successful measurements for #{name}")
end
end
class ErrorRate
include Sli
def increment(labels:, error:)
super(labels: labels, increment_numerator: error)
end
private
def counter_name(suffix)
[COUNTER_PREFIX, name, suffix].join('_').to_sym
end
def numerator_counter
prometheus.counter(counter_name('error_total'), "Number of error measurements for #{name}")
end
end
end
end
end