gitlabhq/lib/gitlab/instrumentation/redis.rb

# frozen_string_literal: true

module Gitlab
  module Instrumentation
    # Aggregates Redis measurements from different request storage sources.
    class Redis
      # Actioncable has it's separate instrumentation, but isn't configurable
      # in the same way as all the other instances using a class.
      ActionCable = Class.new(RedisBase)

      STORAGES = (
        Gitlab::Redis::ALL_CLASSES.map do |redis_instance_class|
          instrumentation_class = Class.new(RedisBase)

          unless Gitlab::Redis::Queues.instances.value?(redis_instance_class)
            instrumentation_class.enable_redis_cluster_validation
          end

          const_set(redis_instance_class.store_name, instrumentation_class)
          instrumentation_class
        end << ActionCable
      ).freeze

      # Milliseconds represented in seconds
      QUERY_TIME_BUCKETS = [0.1, 0.25, 0.5].freeze

      class << self
        include ::Gitlab::Instrumentation::RedisPayload

        def storage_key
          nil
        end

        def shard_key
          nil
        end

        def payload
          super.merge(*STORAGES.flat_map(&:payload))
        end

        def storage_hash
          @storage_hash ||= STORAGES.index_by { |k| k.name.demodulize }
        end

        def detail_store
          STORAGES.flat_map do |storage|
            storage.detail_store.map { |details| details.merge(storage: storage.name.demodulize) }
          end
        end

        %i[get_request_count get_cross_slot_request_count get_allowed_cross_slot_request_count query_time read_bytes
           write_bytes].each do |method|
          define_method method do
            STORAGES.sum(&method)
          end
        end
      end
    end
  end
end