gitlab-org--gitlab-foss/lib/gitlab/marginalia.rb

# frozen_string_literal: true

module Gitlab
  module Marginalia
    cattr_accessor :enabled, default: false

    def self.set_application_name
      ::Marginalia.application_name = Gitlab.process_name
    end

    def self.enable_sidekiq_instrumentation
      if Sidekiq.server?
        ::Marginalia::SidekiqInstrumentation.enable!
      end
    end

    def self.set_enabled_from_feature_flag
      # During db:create and db:bootstrap skip feature query as DB is not available yet.
      return false unless Gitlab::Database.cached_table_exists?('features')

      self.enabled = Feature.enabled?(:marginalia, type: :ops)
    end
  end
end