class PrometheusExporter::Server::ActiveRecordCollector

Constants

ACTIVE_RECORD_GAUGES
MAX_ACTIVERECORD_METRIC_AGE

Public Class Methods

new() click to toggle source
# File lib/prometheus_exporter/server/active_record_collector.rb, line 15
def initialize
  @active_record_metrics = []
end

Public Instance Methods

collect(obj) click to toggle source
# File lib/prometheus_exporter/server/active_record_collector.rb, line 44
def collect(obj)
  now = ::Process.clock_gettime(::Process::CLOCK_MONOTONIC)

  obj["created_at"] = now

  @active_record_metrics.delete_if do |current|
    (obj["pid"] == current["pid"] && obj["hostname"] == current["hostname"] &&
     obj["metric_labels"]["pool_name"] == current["metric_labels"]["pool_name"]) ||
      (current["created_at"] + MAX_ACTIVERECORD_METRIC_AGE < now)
  end

  @active_record_metrics << obj
end
metrics() click to toggle source
# File lib/prometheus_exporter/server/active_record_collector.rb, line 23
def metrics
  return [] if @active_record_metrics.length == 0

  metrics = {}

  @active_record_metrics.map do |m|
    metric_key = (m["metric_labels"] || {}).merge("pid" => m["pid"])
    metric_key.merge!(m["custom_labels"]) if m["custom_labels"]

    ACTIVE_RECORD_GAUGES.map do |k, help|
      k = k.to_s
      if v = m[k]
        g = metrics[k] ||= PrometheusExporter::Metric::Gauge.new("active_record_connection_pool_#{k}", help)
        g.observe(v, metric_key)
      end
    end
  end

  metrics.values
end
type() click to toggle source
# File lib/prometheus_exporter/server/active_record_collector.rb, line 19
def type
  "active_record"
end