-
Notifications
You must be signed in to change notification settings - Fork 153
/
sidekiq_queue_collector.rb
45 lines (37 loc) · 1.13 KB
/
sidekiq_queue_collector.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# frozen_string_literal: true
module PrometheusExporter::Server
class SidekiqQueueCollector < TypeCollector
MAX_METRIC_AGE = 60
SIDEKIQ_QUEUE_GAUGES = {
'backlog' => 'Size of the sidekiq queue.',
'latency_seconds' => 'Latency of the sidekiq queue.',
}.freeze
attr_reader :sidekiq_metrics, :gauges
def initialize
@sidekiq_metrics = MetricsContainer.new(ttl: MAX_METRIC_AGE)
@gauges = {}
end
def type
'sidekiq_queue'
end
def metrics
SIDEKIQ_QUEUE_GAUGES.each_key { |name| gauges[name]&.reset! }
sidekiq_metrics.map do |metric|
labels = metric.fetch("labels", {})
SIDEKIQ_QUEUE_GAUGES.map do |name, help|
if (value = metric[name])
gauge = gauges[name] ||= PrometheusExporter::Metric::Gauge.new("sidekiq_queue_#{name}", help)
gauge.observe(value, labels)
end
end
end
gauges.values
end
def collect(object)
object['queues'].each do |queue|
queue["labels"].merge!(object['custom_labels']) if object['custom_labels']
@sidekiq_metrics << queue
end
end
end
end