34 lines
1 KiB
Ruby
34 lines
1 KiB
Ruby
|
# 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
|
||
|
|
|
||
|
|
def initialize(max_traces_per_second: 10)
|
||
|
|
if max_traces_per_second < 0.0
|
||
|
|
raise "max_traces_per_second must not be negative, got #{max_traces_per_second}"
|
||
|
|
end
|
||
|
|
|
||
|
|
@rate_limiter = RateLimiter.new(
|
||
|
|
credits_per_second: max_traces_per_second,
|
||
|
|
max_balance: [max_traces_per_second, 1.0].max
|
||
|
|
)
|
||
|
|
@tags = {
|
||
|
|
'sampler.type' => 'ratelimiting',
|
||
|
|
'sampler.param' => max_traces_per_second
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
def sample?(*)
|
||
|
|
[@rate_limiter.check_credit(1.0), @tags]
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|