Sha256: 8c95043e2df4cb5b5331649072528fc62a866fa9b2ae6b3fec54df21ce42d7a6
Contents?: true
Size: 1.7 KB
Versions: 10
Compression:
Stored size: 1.7 KB
Contents
# frozen_string_literal: true
begin
require 'raindrops'
rescue LoadError
# No raindrops available, dont do anything
end
module PrometheusExporter::Instrumentation
# collects stats from unicorn
class Unicorn < PeriodicStats
def self.start(pid_file:, listener_address:, client: nil, frequency: 30)
unicorn_collector = new(pid_file: pid_file, listener_address: listener_address)
client ||= PrometheusExporter::Client.default
worker_loop do
metric = unicorn_collector.collect
client.send_json metric
end
super
end
def initialize(pid_file:, listener_address:)
@pid_file = pid_file
@listener_address = listener_address
@tcp = listener_address =~ /\A.+:\d+\z/
end
def collect
metric = {}
metric[:type] = 'unicorn'
collect_unicorn_stats(metric)
metric
end
def collect_unicorn_stats(metric)
stats = listener_address_stats
metric[:active_workers] = stats.active
metric[:request_backlog] = stats.queued
metric[:workers] = worker_process_count
end
private
def worker_process_count
return nil unless File.exist?(@pid_file)
pid = File.read(@pid_file).to_i
return nil if pid < 1
# find all processes whose parent is the unicorn master
# but we're actually only interested in the number of processes (= lines of output)
result = `pgrep -P #{pid} -f unicorn -a`
result.lines.count
end
def listener_address_stats
if @tcp
Raindrops::Linux.tcp_listener_stats([@listener_address])[@listener_address]
else
Raindrops::Linux.unix_listener_stats([@listener_address])[@listener_address]
end
end
end
end
Version data entries
10 entries across 10 versions & 1 rubygems