aboutsummaryrefslogtreecommitdiff
path: root/lib/longboat/collector.rb
blob: ccf103b1399321d3db70dc4fe5bc8fb6d7ca5210 (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
module Longboat
  class Collector
    def initialize
      @metrics = {}
      @jobs = []
    end

    def report!(name, value, help: nil, type: nil, labels: {})
      @metrics[name] ||= {help: help, type: type}
      @metrics[name][labels] = value
    end

    def metrics
      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!(job)
      @jobs << job
    end

    def collect!
      @jobs.each do |job|
        job.run
      end
    end
  end
end