Contents

# frozen_string_literal: true

module PrometheusExporter::Server
  class RunnerException < StandardError; end;
  class WrongInheritance < RunnerException; end;

  class Runner
    def initialize(options = {})
      options.each do |k, v|
        send("#{k}=", v) if self.class.method_defined?("#{k}=")
      end
    end

    def start
      PrometheusExporter::Metric::Base.default_prefix = prefix

      register_type_collectors

      unless collector.is_a?(PrometheusExporter::Server::CollectorBase)
        raise WrongInheritance, 'Collector class must be inherited from PrometheusExporter::Server::CollectorBase'
      end

      server = server_class.new port: port, collector: collector, verbose: verbose
      server.start
    end

    def prefix=(prefix)
      @prefix = prefix
    end

    def prefix
      @prefix || PrometheusExporter::DEFAULT_PREFIX
    end

    def port=(port)
      @port = port
    end

    def port
      @port || PrometheusExporter::DEFAULT_PORT
    end

    def collector_class=(collector_class)
      @collector_class = collector_class
    end

    def collector_class
      @collector_class || PrometheusExporter::Server::Collector
    end

    def type_collectors=(type_collectors)
      @type_collectors = type_collectors
    end

    def type_collectors
      @type_collectors || []
    end

    def verbose=(verbose)
      @verbose = verbose
    end

    def verbose
      return @verbose if defined? @verbose
      false
    end

    def server_class=(server_class)
      @server_class = server_class
    end

    def server_class
      @server_class || PrometheusExporter::Server::WebServer
    end

    def collector
      @_collector ||= collector_class.new
    end

    private

    def register_type_collectors
      type_collectors.each do |klass|
        collector.register_collector klass.new
        STDERR.puts "Registered TypeCollector: #{klass}" if verbose
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
prometheus_exporter-0.3.0	lib/prometheus_exporter/server/runner.rb
prometheus_exporter-0.2.0	lib/prometheus_exporter/server/runner.rb
prometheus_exporter-0.1.16	lib/prometheus_exporter/server/runner.rb
prometheus_exporter-0.1.15	lib/prometheus_exporter/server/runner.rb
prometheus_exporter-0.1.14	lib/prometheus_exporter/server/runner.rb