summaryrefslogtreecommitdiff
path: root/lib/gitlab/metrics/sli.rb
blob: de73db0755d0f94cbbffab8a9fba795a4b44f93c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
# frozen_string_literal: true

module Gitlab
  module Metrics
    class Sli
      SliNotInitializedError = Class.new(StandardError)

      COUNTER_PREFIX = 'gitlab_sli'

      class << self
        INITIALIZATION_MUTEX = Mutex.new

        def [](name)
          known_slis[name] || initialize_sli(name, [])
        end

        def initialize_sli(name, possible_label_combinations)
          INITIALIZATION_MUTEX.synchronize do
            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

      attr_reader :name

      def initialize(name)
        @name = name
        @initialized_with_combinations = false
      end

      def initialize_counters(possible_label_combinations)
        @initialized_with_combinations = possible_label_combinations.any?
        possible_label_combinations.each do |label_combination|
          total_counter.get(label_combination)
          success_counter.get(label_combination)
        end
      end

      def increment(labels:, success:)
        total_counter.increment(labels)
        success_counter.increment(labels) if success
      end

      def initialized?
        @initialized_with_combinations
      end

      private

      def total_counter
        prometheus.counter(total_counter_name.to_sym, "Total number of measurements for #{name}")
      end

      def success_counter
        prometheus.counter(success_counter_name.to_sym, "Number of successful measurements for #{name}")
      end

      def total_counter_name
        "#{COUNTER_PREFIX}:#{name}:total"
      end

      def success_counter_name
        "#{COUNTER_PREFIX}:#{name}:success_total"
      end

      def prometheus
        Gitlab::Metrics
      end
    end
  end
end