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

require 'webrick'
require 'prometheus/client/rack/exporter'

module Gitlab
  module Metrics
    class SidekiqMetricsExporter < Daemon
      LOG_FILENAME = File.join(Rails.root, 'log', 'sidekiq_exporter.log')

      def enabled?
        Gitlab::Metrics.metrics_folder_present? && settings.enabled
      end

      def settings
        Settings.monitoring.sidekiq_exporter
      end

      private

      attr_reader :server

      def start_working
        logger = WEBrick::Log.new(LOG_FILENAME)
        access_log = [
          [logger, WEBrick::AccessLog::COMBINED_LOG_FORMAT]
        ]

        @server = ::WEBrick::HTTPServer.new(Port: settings.port, BindAddress: settings.address,
                                            Logger: logger, AccessLog: access_log)
        server.mount "/", Rack::Handler::WEBrick, rack_app
        server.start
      end

      def stop_working
        server.shutdown if server
        @server = nil
      end

      def rack_app
        Rack::Builder.app do
          use Rack::Deflater
          use ::Prometheus::Client::Rack::Exporter
          run -> (env) { [404, {}, ['']] }
        end
      end
    end
  end
end