Sha256: 0306d31b1f83825e461c171f8728b98e212c688cb6fa7b7ab78b20b58ac7629a
Contents?: true
Size: 1.72 KB
Versions: 10
Compression:
Stored size: 1.72 KB
Contents
# frozen_string_literal: true
require "json"
# collects stats from puma
module PrometheusExporter::Instrumentation
class Puma < PeriodicStats
def self.start(client: nil, frequency: 30, labels: {})
puma_collector = new(labels)
client ||= PrometheusExporter::Client.default
worker_loop do
metric = puma_collector.collect
client.send_json metric
end
super
end
def initialize(metric_labels = {})
@metric_labels = metric_labels
end
def collect
metric = {
pid: pid,
type: "puma",
hostname: ::PrometheusExporter.hostname,
metric_labels: @metric_labels
}
collect_puma_stats(metric)
metric
end
def pid
@pid = ::Process.pid
end
def collect_puma_stats(metric)
stats = JSON.parse(::Puma.stats)
if stats.key?("workers")
metric[:phase] = stats["phase"]
metric[:workers] = stats["workers"]
metric[:booted_workers] = stats["booted_workers"]
metric[:old_workers] = stats["old_workers"]
stats["worker_status"].each do |worker|
next if worker["last_status"].empty?
collect_worker_status(metric, worker["last_status"])
end
else
collect_worker_status(metric, stats)
end
end
private
def collect_worker_status(metric, status)
metric[:request_backlog] ||= 0
metric[:running_threads] ||= 0
metric[:thread_pool_capacity] ||= 0
metric[:max_threads] ||= 0
metric[:request_backlog] += status["backlog"]
metric[:running_threads] += status["running"]
metric[:thread_pool_capacity] += status["pool_capacity"]
metric[:max_threads] += status["max_threads"]
end
end
end
Version data entries
10 entries across 10 versions & 1 rubygems