Sha256: a17ab07bcc02df18bdbce15a4cdd49005cb5d8e5698dd77e36a1ba0e295f3dba
Contents?: true
Size: 1.69 KB
Versions: 11
Compression:
Stored size: 1.69 KB
Contents
require 'json'
# collects stats from puma
module PrometheusExporter::Instrumentation
class Puma
def self.start(client: nil, frequency: 30)
puma_collector = new
client ||= PrometheusExporter::Client.default
Thread.new do
while true
begin
metric = puma_collector.collect
client.send_json metric
rescue => e
STDERR.puts("Prometheus Exporter Failed To Collect Puma Stats #{e}")
ensure
sleep frequency
end
end
end
end
def collect
metric = {}
metric[:type] = "puma"
collect_puma_stats(metric)
metric
end
def collect_puma_stats(metric)
stats = JSON.parse(::Puma.stats)
if stats.key? 'workers'
metric[:phase] = stats["phase"]
metric[:workers_total] = stats["workers"]
metric[:booted_workers_total] = stats["booted_workers"]
metric[:old_workers_total] = 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_total] ||= 0
metric[:running_threads_total] ||= 0
metric[:thread_pool_capacity_total] ||= 0
metric[:max_threads_total] ||= 0
metric[:request_backlog_total] += status["backlog"]
metric[:running_threads_total] += status["running"]
metric[:thread_pool_capacity_total] += status["pool_capacity"]
metric[:max_threads_total] += status["max_threads"]
end
end
end
Version data entries
11 entries across 11 versions & 1 rubygems