open-telemetry
/
opentelemetry-java-instrumentation
mirror of https://github.com/open-telemetry/opentelemetry-java-instrumentation.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Rename KnuthSampler to DeterministicSampler

This commit is contained in:
Laplie Anderson 2019-11-22 13:50:55 -05:00
parent 10b0f35235
commit 83e78fb8f0
4 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
dd-trace-ot/src/main/java/datadog/trace/common/sampling/KnuthSampler.java → dd-trace-ot/src/main/java/datadog/trace/common/sampling/DeterministicSampler.java
View File

@ -6,8 +6,12 @@ import java.math.BigDecimal;
import java.math.BigInteger; import java.math.BigInteger;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
/**
* This implements the deterministic sampling algorithm used by the Datadog Agent as well as the
* tracers for other languages
*/
@Slf4j @Slf4j
public class KnuthSampler implements RateSampler { public class DeterministicSampler implements RateSampler {
private static final BigInteger KNUTH_FACTOR = new BigInteger("1111111111111111111"); private static final BigInteger KNUTH_FACTOR = new BigInteger("1111111111111111111");
private static final BigDecimal TRACE_ID_MAX_AS_BIG_DECIMAL = private static final BigDecimal TRACE_ID_MAX_AS_BIG_DECIMAL =
new BigDecimal(DDTracer.TRACE_ID_MAX); new BigDecimal(DDTracer.TRACE_ID_MAX);
@ -16,7 +20,7 @@ public class KnuthSampler implements RateSampler {
private final BigInteger cutoff; private final BigInteger cutoff;
private final double rate; private final double rate;
public KnuthSampler(final double rate) { public DeterministicSampler(final double rate) {
this.rate = rate; this.rate = rate;
cutoff = new BigDecimal(rate).multiply(TRACE_ID_MAX_AS_BIG_DECIMAL).toBigInteger(); cutoff = new BigDecimal(rate).multiply(TRACE_ID_MAX_AS_BIG_DECIMAL).toBigInteger();

2
dd-trace-ot/src/main/java/datadog/trace/common/sampling/RateByServiceSampler.java
View File

@ -107,6 +107,6 @@ public class RateByServiceSampler implements Sampler, PrioritySampler, ResponseL
sanitizedRate = sampleRate; sanitizedRate = sampleRate;
} }
return new KnuthSampler(sanitizedRate); return new DeterministicSampler(sanitizedRate);
} }
} }

6
dd-trace-ot/src/main/java/datadog/trace/common/sampling/RuleBasedSampler.java
View File

@ -45,7 +45,7 @@ public class RuleBasedSampler implements Sampler, PrioritySampler {
try { try {
final double rateForEntry = Double.parseDouble(entry.getValue()); final double rateForEntry = Double.parseDouble(entry.getValue());
final SamplingRule samplingRule = final SamplingRule samplingRule =
new ServiceSamplingRule(entry.getKey(), new KnuthSampler(rateForEntry)); new ServiceSamplingRule(entry.getKey(), new DeterministicSampler(rateForEntry));
samplingRules.add(samplingRule); samplingRules.add(samplingRule);
} catch (final NumberFormatException e) { } catch (final NumberFormatException e) {
log.error("Unable to parse rate for service: {}", entry, e); log.error("Unable to parse rate for service: {}", entry, e);
@ -58,7 +58,7 @@ public class RuleBasedSampler implements Sampler, PrioritySampler {
try { try {
final double rateForEntry = Double.parseDouble(entry.getValue()); final double rateForEntry = Double.parseDouble(entry.getValue());
final SamplingRule samplingRule = final SamplingRule samplingRule =
new OperationSamplingRule(entry.getKey(), new KnuthSampler(rateForEntry)); new OperationSamplingRule(entry.getKey(), new DeterministicSampler(rateForEntry));
samplingRules.add(samplingRule); samplingRules.add(samplingRule);
} catch (final NumberFormatException e) { } catch (final NumberFormatException e) {
log.error("Unable to parse rate for operation: {}", entry, e); log.error("Unable to parse rate for operation: {}", entry, e);
@ -68,7 +68,7 @@ public class RuleBasedSampler implements Sampler, PrioritySampler {
if (defaultRate != null) { if (defaultRate != null) {
final SamplingRule samplingRule = final SamplingRule samplingRule =
new AlwaysMatchesSamplingRule(new KnuthSampler(defaultRate)); new AlwaysMatchesSamplingRule(new DeterministicSampler(defaultRate));
samplingRules.add(samplingRule); samplingRules.add(samplingRule);
} }

10
dd-trace-ot/src/test/groovy/datadog/trace/api/sampling/KnuthSamplerTest.groovy → dd-trace-ot/src/test/groovy/datadog/trace/api/sampling/DeterministicSamplerTest.groovy
View File

@ -1,14 +1,14 @@
package datadog.trace.api.sampling package datadog.trace.api.sampling
import datadog.opentracing.DDSpan import datadog.opentracing.DDSpan
import datadog.trace.common.sampling.KnuthSampler import datadog.trace.common.sampling.DeterministicSampler
import datadog.trace.util.test.DDSpecification import datadog.trace.util.test.DDSpecification
class KnuthSamplerTest extends DDSpecification { class DeterministicSamplerTest extends DDSpecification {
def "test known values: #traceId"() { def "test known values: #traceId"() {
given: given:
KnuthSampler sampler = new KnuthSampler(0.5) DeterministicSampler sampler = new DeterministicSampler(0.5)
DDSpan span = Mock(DDSpan) { DDSpan span = Mock(DDSpan) {
getTraceId() >> traceId getTraceId() >> traceId
} }
@ -125,7 +125,7 @@ class KnuthSamplerTest extends DDSpecification {
def "test sampling none: #traceId"() { def "test sampling none: #traceId"() {
given: given:
KnuthSampler sampler = new KnuthSampler(0) DeterministicSampler sampler = new DeterministicSampler(0)
DDSpan span = Mock(DDSpan) { DDSpan span = Mock(DDSpan) {
getTraceId() >> traceId getTraceId() >> traceId
} }
@ -244,7 +244,7 @@ class KnuthSamplerTest extends DDSpecification {
def "test sampling all: #traceId"() { def "test sampling all: #traceId"() {
given: given:
KnuthSampler sampler = new KnuthSampler(1) DeterministicSampler sampler = new DeterministicSampler(1)
DDSpan span = Mock(DDSpan) { DDSpan span = Mock(DDSpan) {
getTraceId() >> traceId getTraceId() >> traceId
} }