gitlab-org--gitlab-foss/lib/gitlab/metrics/exporter/base_exporter.rb

# frozen_string_literal: true

module Gitlab
  module Metrics
    module Exporter
      class BaseExporter < Daemon
        attr_reader :server

        attr_accessor :additional_checks

        def enabled?
          settings.enabled
        end

        def settings
          raise NotImplementedError
        end

        def log_filename
          raise NotImplementedError
        end

        private

        def start_working
          logger = WEBrick::Log.new(log_filename)
          logger.time_format = "[%Y-%m-%dT%H:%M:%S.%L%z]"

          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_proc '/readiness' do |req, res|
            render_probe(readiness_probe, req, res)
          end
          server.mount_proc '/liveness' do |req, res|
            render_probe(liveness_probe, req, res)
          end
          server.mount '/', Rack::Handler::WEBrick, rack_app

          true
        end

        def run_thread
          server&.start
        rescue IOError
          # ignore forcibily closed servers
        end

        def stop_working
          if server
            # we close sockets if thread is not longer running
            # this happens, when the process forks
            server.listeners.each(&:close) unless thread.alive?
            server.shutdown
          end

          @server = nil
        end

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

        def readiness_probe
          ::Gitlab::HealthChecks::Probes::Readiness.new(*additional_checks)
        end

        def liveness_probe
          ::Gitlab::HealthChecks::Probes::Liveness.new
        end

        def render_probe(probe, req, res)
          result = probe.execute

          res.status = result.http_status
          res.content_type = 'application/json; charset=utf-8'
          res.body = result.json.to_json
        end
      end
    end
  end
end