diff --git a/src/main/java/com/datadoghq/trace/Sampler.java b/src/main/java/com/datadoghq/trace/Sampler.java index 6977bb8f4f..bf1d137b1c 100644 --- a/src/main/java/com/datadoghq/trace/Sampler.java +++ b/src/main/java/com/datadoghq/trace/Sampler.java @@ -1,7 +1,7 @@ package com.datadoghq.trace; -import com.datadoghq.trace.impl.DDSpan; +import io.opentracing.Span; /** * Main interface to sample a collection of traces. @@ -14,6 +14,6 @@ public interface Sampler { * @param span the parent span with its context * @return true when the trace/spans has to be reported/written */ - boolean sample(DDSpan span); + boolean sample(Span span); } diff --git a/src/main/java/com/datadoghq/trace/impl/AllSampler.java b/src/main/java/com/datadoghq/trace/impl/AllSampler.java index 9b5a148b4c..46448bc773 100644 --- a/src/main/java/com/datadoghq/trace/impl/AllSampler.java +++ b/src/main/java/com/datadoghq/trace/impl/AllSampler.java @@ -1,6 +1,7 @@ package com.datadoghq.trace.impl; import com.datadoghq.trace.Sampler; +import io.opentracing.Span; /** * Sampler that always says yes... @@ -8,7 +9,7 @@ import com.datadoghq.trace.Sampler; public class AllSampler implements Sampler { @Override - public boolean sample(DDSpan span) { + public boolean sample(Span span) { return true; } diff --git a/src/main/java/com/datadoghq/trace/impl/RateSampler.java b/src/main/java/com/datadoghq/trace/impl/RateSampler.java index be922f1a3d..afa5b49c66 100644 --- a/src/main/java/com/datadoghq/trace/impl/RateSampler.java +++ b/src/main/java/com/datadoghq/trace/impl/RateSampler.java @@ -2,6 +2,7 @@ package com.datadoghq.trace.impl; import com.datadoghq.trace.Sampler; +import io.opentracing.Span; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -41,7 +42,7 @@ public class RateSampler implements Sampler { } @Override - public boolean sample(DDSpan span) { + public boolean sample(Span span) { boolean sample = Math.random() <= this.sampleRate; logger.debug("{} - Span is sampled: {}", span, sample); return sample;