gitlab-org--gitlab-foss/lib/gitlab/metrics/sli.rb

# 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|
          legacy_total_counter.get(label_combination)
          legacy_numerator_counter.get(label_combination)

          if ::Feature.enabled?(:gitlab_sli_new_counters)
            total_counter.get(label_combination)
            numerator_counter.get(label_combination)
          end
        end
      end

      def increment(labels:, increment_numerator:)
        legacy_total_counter.increment(labels)
        legacy_numerator_counter.increment(labels) if increment_numerator

        if ::Feature.enabled?(:gitlab_sli_new_counters)
          total_counter.increment(labels)
          numerator_counter.increment(labels) if increment_numerator
        end
      end

      def initialized?
        @initialized_with_combinations
      end

      private

      def total_counter
        prometheus.counter(counter_name('total', '_'), "Total number of measurements for #{name}")
      end

      def legacy_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, separator)
          [COUNTER_PREFIX, "#{name}_apdex", suffix].join(separator).to_sym
        end

        def numerator_counter
          prometheus.counter(counter_name('success_total', '_'), "Number of successful measurements for #{name}")
        end

        def legacy_numerator_counter
          prometheus.counter(counter_name('success_total', ':'), "Legacy 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, separator)
          [COUNTER_PREFIX, name, suffix].join(separator).to_sym
        end

        def numerator_counter
          prometheus.counter(counter_name('error_total', '_'), "Number of error measurements for #{name}")
        end

        def legacy_numerator_counter
          prometheus.counter(counter_name('error_total', ':'), "Number of error measurements for #{name}")
        end
      end
    end
  end
end