blob: 10e2f8492f5c8094917b53010119df266f395d09 (
plain)
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
46
47
48
49
50
51
52
53
54
|
module Longboat
class Collector
def initialize(config)
@metrics = {}
@config = config
end
def report!(name, value, help: nil, type: nil, labels: {}, timestamp: Time.now)
name = prefix(name)
@metrics[name] ||= {help: help, type: type}
@metrics[name][labels] = {value: value, timestamp: timestamp}
end
def redact!(name, labels: nil)
name = prefix(name)
if labels.nil?
@metrics.delete(name)
else
@metrics[name].delete(labels)
end
end
def prometheus_metrics
res = ""
@metrics.each do |name, metric|
res << ?\n unless res.empty?
res << "#HELP #{name} #{metric[:help]}\n" unless metric[:help].nil?
res << "#TYPE #{name} #{metric[:type]}\n" unless metric[:type].nil?
metric.each do |labels, value|
next if labels == :help
next if labels == :type
labellist = []
labels.each do |k, v|
labellist << "#{k}=\"#{v}\""
end
labellist = labellist.join(",")
res << "#{name}{#{labellist}} #{value[:value]} #{(value[:timestamp].to_f * 1000).to_i}\n"
end
end
res
end
private
def prefix(name)
"#{@config[:metric_prefix]}#{name}"
end
end
end
|