class Atatus::Metrics::Registry

@api private

Constants

TIMEOUT_INTERVAL

Attributes

callback[R]
config[R]
sets[R]

Public Class Methods

new(config, &block) click to toggle source
# File lib/atatus/metrics.rb, line 37
def initialize(config, &block)
  @config = config
  @callback = block
end

Public Instance Methods

collect() click to toggle source
# File lib/atatus/metrics.rb, line 120
def collect
  sets.each_value.each_with_object([]) do |set, arr|
    samples = set.collect
    next unless samples
    arr.concat(samples)
  end
end
collect_and_send() click to toggle source
# File lib/atatus/metrics.rb, line 111
def collect_and_send
  return unless @config.recording?
  metricsets = collect
  metricsets.compact!
  metricsets.each do |m|
    callback.call(m)
  end
end
get(key) click to toggle source
# File lib/atatus/metrics.rb, line 107
def get(key)
  sets.fetch(key)
end
handle_forking!() click to toggle source
# File lib/atatus/metrics.rb, line 95
def handle_forking!
  # Note that ideally we would be able to check if the @timer_task died
  # and restart it. You can't simply check @timer_task.running? because
  # it will only return the state of the TimerTask, not whether the
  # internal thread used to manage the execution interval has died.
  # This is a limitation of the Concurrent::TimerTask object.
  # Therefore, our only option when forked is to stop and start.
  # ~estolfo
  stop
  start
end
running?() click to toggle source
# File lib/atatus/metrics.rb, line 91
def running?
  !!@running
end
start() click to toggle source
# File lib/atatus/metrics.rb, line 43
def start
  unless config.collect_metrics?
    debug 'Skipping metrics'
    return
  end

  debug 'Starting metrics'

  # Only set the @sets once, in case we stop
  # and start again.
  @sets ||= {
    system: CpuMemSet,
    # vm: VMSet,
    breakdown: BreakdownSet,
    transaction: TransactionSet
  }.each_with_object({}) do |(key, kls), sets|
    debug "Adding metrics collector '#{kls}'"
    sets[key] = kls.new(config)
  end

  @timer_task = Concurrent::TimerTask.execute(
    run_now: true,
    execution_interval: config.metrics_interval,
    timeout_interval: TIMEOUT_INTERVAL
  ) do
    begin
      debug 'Collecting metrics'
      collect_and_send
      true
    rescue StandardError => e
      error 'Error while collecting metrics: %e', e.inspect
      debug { e.backtrace.join("\n") }
      false
    end
  end

  @running = true
end
stop() click to toggle source
# File lib/atatus/metrics.rb, line 82
def stop
  return unless running?

  debug 'Stopping metrics'

  @timer_task.shutdown
  @running = false
end