aboutsummaryrefslogtreecommitdiff
path: root/lib/longboat/metrics.rb
blob: 58b1cbee9e1e31225bbe88b715afeb7e260a3596 (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
module Longboat
  module Metrics
    class << self
      def report!(name, value, help: nil, type: nil, labels: {})
        @metrics ||= {}
        @metrics[name] ||= {help: help, type: type}
        @metrics[name][labels] = value
      end

      def to_s
        timestamp = (Time.now.to_f * 1000).to_i
        res = ""
        (@metrics || {}).each do |name, metric|
          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} #{timestamp}\n"
          end
        end
        res
      end

      def register!(collector)
        @collectors ||= []
        @collectors << collector
      end

      def collect!
        (@collectors || []).each do |collector|
          collector.collect!
        end
      end
    end
  end
end