Move samplers to SDK package (#1023)

Author: lzchen

docs/api/trace.rst

@@ -6,7 +6,6 @@ Submodules
 
 .. toctree::
 
-   trace.sampling
    trace.status
    trace.span
 

docs/api/trace.sampling.rst

@@ -7,6 +7,7 @@ Submodules
 .. toctree::
 
    trace.export
+   trace.sampling
    util.instrumentation
 
 .. automodule:: opentelemetry.sdk.trace

docs/sdk/trace.rst

@@ -0,0 +1,7 @@
+opentelemetry.sdk.trace.sampling
+==========================================
+
+.. automodule:: opentelemetry.sdk.trace.sampling
+    :members:
+    :undoc-members:
+    :show-inheritance:

docs/sdk/trace.sampling.rst

@@ -21,6 +21,7 @@
 from ddtrace.span import Span as DatadogSpan
 
 import opentelemetry.trace as trace_api
+from opentelemetry.sdk.trace import sampling
 from opentelemetry.sdk.trace.export import SpanExporter, SpanExportResult
 from opentelemetry.trace.status import StatusCanonicalCode
 
@@ -246,7 +247,7 @@ def _get_sampling_rate(span):
     return (
         span.sampler.rate
         if ctx.trace_flags.sampled
-        and isinstance(span.sampler, trace_api.sampling.ProbabilitySampler)
+        and isinstance(span.sampler, sampling.ProbabilitySampler)
         else None
     )
 

exporter/opentelemetry-exporter-datadog/src/opentelemetry/exporter/datadog/exporter.py

@@ -23,6 +23,7 @@
 from opentelemetry import trace as trace_api
 from opentelemetry.exporter import datadog
 from opentelemetry.sdk import trace
+from opentelemetry.sdk.trace import sampling
 from opentelemetry.sdk.util.instrumentation import InstrumentationInfo
 
 
@@ -497,7 +498,7 @@ def test_sampling_rate(self):
             is_remote=False,
             trace_flags=trace_api.TraceFlags(trace_api.TraceFlags.SAMPLED),
         )
-        sampler = trace_api.sampling.ProbabilitySampler(0.5)
+        sampler = sampling.ProbabilitySampler(0.5)
 
         span = trace.Span(
             name="sampled", context=context, parent=None, sampler=sampler

exporter/opentelemetry-exporter-datadog/tests/test_datadog_exporter.py

@@ -2,6 +2,9 @@
 
 ## Unreleased
 
+- Moved samplers from API to SDK
+  ([#1023](https://github.com/open-telemetry/opentelemetry-python/pull/1023))
+
 ## Version 0.12b0
 
 Released 2020-08-14

opentelemetry-api/CHANGELOG.md

@@ -2,11 +2,15 @@
 
 ## Unreleased
 
+- Moved samplers from API to SDK
+  ([#1023](https://github.com/open-telemetry/opentelemetry-python/pull/1023))
+
 ## Version 0.12b0
 
 Released 2020-08-14
 
 - Changed default Sampler to `ParentOrElse(AlwaysOn)`
+  ([#960](https://github.com/open-telemetry/opentelemetry-python/pull/960))
 - Update environment variable names, prefix changed from `OPENTELEMETRY` to `OTEL`
   ([#904](https://github.com/open-telemetry/opentelemetry-python/pull/904))
 - Implement Views in metrics SDK

opentelemetry-sdk/CHANGELOG.md

@@ -40,9 +40,10 @@
 from opentelemetry import trace as trace_api
 from opentelemetry.sdk import util
 from opentelemetry.sdk.resources import Resource
+from opentelemetry.sdk.trace import sampling
 from opentelemetry.sdk.util import BoundedDict, BoundedList
 from opentelemetry.sdk.util.instrumentation import InstrumentationInfo
-from opentelemetry.trace import SpanContext, sampling
+from opentelemetry.trace import SpanContext
 from opentelemetry.trace.propagation import SPAN_KEY
 from opentelemetry.trace.status import Status, StatusCanonicalCode
 from opentelemetry.util import time_ns, types
@@ -869,7 +870,7 @@ def use_span(
 class TracerProvider(trace_api.TracerProvider):
     def __init__(
         self,
-        sampler: sampling.Sampler = trace_api.sampling.DEFAULT_ON,
+        sampler: sampling.Sampler = sampling.DEFAULT_ON,
         resource: Resource = Resource.create_empty(),
         shutdown_on_exit: bool = True,
         active_span_processor: Union[

opentelemetry-sdk/src/opentelemetry/sdk/trace/__init__.py

@@ -36,12 +36,12 @@
 .. code:: python
 
     from opentelemetry import trace
-    from opentelemetry.trace.sampling import ProbabilitySampler
     from opentelemetry.sdk.trace import TracerProvider
     from opentelemetry.sdk.trace.export import (
         ConsoleSpanExporter,
         SimpleExportSpanProcessor,
     )
+    from opentelemetry.sdk.trace.sampling import ProbabilitySampler
 
     # sample 1 in every 1000 traces
     sampler = ProbabilitySampler(1/1000)