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