Sha256: c664182aa88dea1b40f33228ebc074b84be9e62b2f73ba81e891a9a3ffb81c17

Contents?: true

Size: 1.54 KB

Versions: 6

Compression:

Stored size: 1.54 KB

Contents

# frozen_string_literal: true

module Jaeger
  module Samplers
    # Samples at most max_traces_per_second. The distribution of sampled
    # traces follows burstiness of the service, i.e. a service with uniformly
    # distributed requests will have those requests sampled uniformly as
    # well, but if requests are bursty, especially sub-second, then a number
    # of sequential requests can be sampled each second.
    class RateLimiting
      attr_reader :tags, :max_traces_per_second

      def initialize(max_traces_per_second: 10)
        update(max_traces_per_second: max_traces_per_second)
      end

      def update(max_traces_per_second:)
        if max_traces_per_second < 0.0
          raise "max_traces_per_second must not be negative, got #{max_traces_per_second}"
        end

        return false if max_traces_per_second == @max_traces_per_second

        @tags = {
          'sampler.type' => 'ratelimiting',
          'sampler.param' => max_traces_per_second
        }
        @max_traces_per_second = max_traces_per_second
        max_balance = [max_traces_per_second, 1.0].max

        if @rate_limiter
          @rate_limiter.update(
            credits_per_second: max_traces_per_second,
            max_balance: max_balance
          )
        else
          @rate_limiter = RateLimiter.new(
            credits_per_second: max_traces_per_second,
            max_balance: [max_traces_per_second, 1.0].max
          )
        end

        true
      end

      def sample(*)
        [@rate_limiter.check_credit(1.0), @tags]
      end
    end
  end
end

Version data entries

6 entries across 6 versions & 3 rubygems

Version Path
dox-jaeger-client-2.0.0 lib/jaeger/samplers/rate_limiting.rb
jaeger-client-with-ruby-32-support-2.0.0 lib/jaeger/samplers/rate_limiting.rb
jaeger-client-1.3.0 lib/jaeger/samplers/rate_limiting.rb
jaeger-client-1.2.0 lib/jaeger/samplers/rate_limiting.rb
jaeger-client-1.1.0 lib/jaeger/samplers/rate_limiting.rb
jaeger-client-1.0.0 lib/jaeger/samplers/rate_limiting.rb