debian-mirror-gitlab/lib/gitlab/metrics/sli.rb

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

109 lines
2.6 KiB
Ruby
Raw Normal View History

2021-11-18 22:05:49 +05:30
# frozen_string_literal: true
module Gitlab
module Metrics
2022-07-16 23:28:13 +05:30
module Sli
2021-11-18 22:05:49 +05:30
COUNTER_PREFIX = 'gitlab_sli'
2022-07-16 23:28:13 +05:30
module ClassMethods
2021-11-18 22:05:49 +05:30
INITIALIZATION_MUTEX = Mutex.new
def [](name)
known_slis[name] || initialize_sli(name, [])
end
def initialize_sli(name, possible_label_combinations)
INITIALIZATION_MUTEX.synchronize do
2022-07-16 23:28:13 +05:30
next known_slis[name] if initialized?(name)
2021-11-18 22:05:49 +05:30
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
2022-07-16 23:28:13 +05:30
def self.included(mod)
mod.extend(ClassMethods)
end
2021-11-18 22:05:49 +05:30
attr_reader :name
def initialize(name)
@name = name
@initialized_with_combinations = false
end
def initialize_counters(possible_label_combinations)
2022-07-16 23:28:13 +05:30
# This module is effectively an abstract class
@initialized_with_combinations = possible_label_combinations.any? # rubocop:disable Gitlab/ModuleWithInstanceVariables
2021-11-18 22:05:49 +05:30
possible_label_combinations.each do |label_combination|
total_counter.get(label_combination)
2022-07-16 23:28:13 +05:30
numerator_counter.get(label_combination)
2021-11-18 22:05:49 +05:30
end
end
2022-07-16 23:28:13 +05:30
def increment(labels:, increment_numerator:)
2021-11-18 22:05:49 +05:30
total_counter.increment(labels)
2022-07-16 23:28:13 +05:30
numerator_counter.increment(labels) if increment_numerator
2021-11-18 22:05:49 +05:30
end
def initialized?
@initialized_with_combinations
end
private
def total_counter
2022-07-16 23:28:13 +05:30
prometheus.counter(counter_name('total'), "Total number of measurements for #{name}")
2021-11-18 22:05:49 +05:30
end
2022-07-16 23:28:13 +05:30
def counter_name(suffix)
:"#{COUNTER_PREFIX}:#{name}_#{self.class.name.demodulize.underscore}:#{suffix}"
2021-11-18 22:05:49 +05:30
end
2022-07-16 23:28:13 +05:30
def prometheus
Gitlab::Metrics
2021-11-18 22:05:49 +05:30
end
2022-07-16 23:28:13 +05:30
class Apdex
include Sli
def increment(labels:, success:)
super(labels: labels, increment_numerator: success)
end
private
def numerator_counter
prometheus.counter(counter_name('success_total'), "Number of successful measurements for #{name}")
end
2021-11-18 22:05:49 +05:30
end
2022-07-16 23:28:13 +05:30
class ErrorRate
include Sli
def increment(labels:, error:)
super(labels: labels, increment_numerator: error)
end
private
def numerator_counter
prometheus.counter(counter_name('error_total'), "Number of error measurements for #{name}")
end
2021-11-18 22:05:49 +05:30
end
end
end
end