File: local_aggregator.rb

package info (click to toggle)
ruby-sentry-ruby 5.18.2-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 564 kB
  • sloc: ruby: 4,701; makefile: 8; sh: 4
file content (53 lines) | stat: -rw-r--r-- 1,130 bytes parent folder | download
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
# frozen_string_literal: true

module Sentry
  module Metrics
    class LocalAggregator
      # exposed only for testing
      attr_reader :buckets

      def initialize
        @buckets = {}
      end

      def add(key, value)
        if @buckets[key]
          @buckets[key].add(value)
        else
          @buckets[key] = GaugeMetric.new(value)
        end
      end

      def to_hash
        return nil if @buckets.empty?

        @buckets.map do |bucket_key, metric|
          type, key, unit, tags = bucket_key

          payload_key = "#{type}:#{key}@#{unit}"
          payload_value = {
            tags: deserialize_tags(tags),
            min: metric.min,
            max: metric.max,
            count: metric.count,
            sum: metric.sum
          }

          [payload_key, payload_value]
        end.to_h
      end

      private

      def deserialize_tags(tags)
        tags.inject({}) do |h, tag|
          k, v = tag
          old = h[k]
          # make it an array if key repeats
          h[k] = old ? (old.is_a?(Array) ? old << v : [old, v]) : v
          h
        end
      end
    end
  end
end