Sha256: 636ebf544c12b5b363c7078bbe198d1cc7d5088ed13c37ab65fbbc0b2069f5b8
Contents?: true
Size: 1.72 KB
Versions: 13
Compression:
Stored size: 1.72 KB
Contents
# frozen_string_literal: true
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
13 entries across 13 versions & 1 rubygems