metrics wip

sentry-ruby/lib/sentry/metrics.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+require 'sentry/metrics/metric'
+require 'sentry/metrics/counter_metric'
+require 'sentry/metrics/distribution_metric'
+require 'sentry/metrics/gauge_metric'
+require 'sentry/metrics/set_metric'
+
+module Sentry
+ module Metrics
+ class << self
+ # TODO-neel-metrics define units, maybe symbols
+ def incr(key, value: 1.0, unit: 'none', tags: nil, timestamp: nil)
+ end
+ end
+ end
+end

sentry-ruby/lib/sentry/metrics/aggregator.rb

@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+
+module Sentry
+ module Metrics
+ class Aggregator
+ include LoggingHelper
+
+ FLUSH_INTERVAL = 5
+ ROLLUP_IN_SECONDS = 10
+
+ def initialize(configuration, client)
+ @client = client
+ @logger = configuration.logger
+ @release = configuration.release
+ @environment = configuration.environment
+
+ @thread = nil
+ @exited = false
+
+ @buckets = {}
+ end
+
+ def add(type,
+ key,
+ value,
+ unit,
+ tags: nil,
+ timestamp: nil)
+ return unless ensure_thread
+
+ timestamp = timestamp.to_i if timestamp.is_a?(Time)
+ timestamp ||= Sentry.utc_now.to_i
+
+ # this is integer division and thus takes the floor of the division
+ # and buckets into 10 second intervals
+ bucket_timestamp = (timestamp / ROLLUP_IN_SECONDS) * ROLLUP_IN_SECONDS
+ serialized_tags = serialize_tags(tags)
+ bucket_key = [type, key, unit, serialized_tags]
+
+ # TODO lock and add to bucket
+ 42
+ end
+
+ def flush
+ # TODO
+ end
+
+ def kill
+ log_debug("[Metrics::Aggregator] killing thread")
+
+ @exited = true
+ @thread&.kill
+ end
+
+ private
+
+ def ensure_thread
+ return false if @exited
+ return true if @thread&.alive?
+
+ @thread = Thread.new do
+ loop do
+ # TODO use event for force flush later
+ sleep(FLUSH_INTERVAL)
+ flush
+ end
+ end
+
+ true
+ rescue ThreadError
+ log_debug("[Metrics::Aggregator] thread creation failed")
+ @exited = true
+ false
+ end
+
+ def serialize_tags(tags)
+ # TODO support array tags
+ return [] unless tags
+ tags.map { |k, v| [k.to_s, v.to_s] }
+ end
+ end
+ end
+end

sentry-ruby/lib/sentry/metrics/counter_metric.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Sentry
+ module Metrics
+ class CounterMetric < Metric
+ attr_reader :value
+
+ def initialize(value)
+ @value = value.to_f
+ end
+
+ def add(value)
+ @value += value.to_f
+ end
+
+ def serialize
+ [value]
+ end
+
+ def weight
+ 1
+ end
+ end
+ end
+end

sentry-ruby/lib/sentry/metrics/distribution_metric.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Sentry
+ module Metrics
+ class DistributionMetric < Metric
+ attr_reader :value
+
+ def initialize(value)
+ @value = [value.to_f]
+ end
+
+ def add(value)
+ @value << value.to_f
+ end
+
+ def serialize
+ value
+ end
+
+ def weight
+ value.size
+ end
+ end
+ end
+end

sentry-ruby/lib/sentry/metrics/gauge_metric.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module Sentry
+ module Metrics
+ class GaugeMetric < Metric
+ attr_reader :last, :min, :max, :sum, :count
+
+ def initialize(value)
+ value = value.to_f
+ @last = value
+ @min = value
+ @max = value
+ @sum = value
+ @count = 1
+ end
+
+ def add(value)
+ value = value.to_f
+ @last = value
+ @min = [@min, value].min
+ @max = [@max, value].max
+ @sum += value
+ @count += 1
+ end
+
+ def serialize
+ [last, min, max, sum, count]
+ end
+
+ def weight
+ 5
+ end
+ end
+ end
+end

sentry-ruby/lib/sentry/metrics/metric.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module Sentry
+ module Metrics
+ class Metric
+ def add(value)
+ raise NotImplementedError
+ end
+
+ def serialize
+ raise NotImplementedError
+ end
+
+ def weight
+ raise NotImplementedError
+ end
+ end
+ end
+end

sentry-ruby/lib/sentry/metrics/set_metric.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require 'set'
+require 'zlib'
+
+module Sentry
+ module Metrics
+ class SetMetric < Metric
+ attr_reader :value
+
+ def initialize(value)
+ @value = Set[value.to_f]
+ end
+
+ def add(value)
+ @value << value
+ end
+
+ def serialize
+ value.map do |v|
+ x.is_a?(String) ? Zlib.crc32(x) : x.to_i
+ end
+ end
+
+ def weight
+ value.size
+ end
+ end
+ end
+end