Replace opentracing with opentelemetry

faust/utils/tracing.py

@@ -6,8 +6,8 @@
 from functools import wraps
 from typing import Any, Callable, Optional, Tuple
 
-import opentracing
 from mode import shortlabel
+from opentelemetry.trace.span import Span
 
 __all__ = [
     "current_span",
@@ -20,28 +20,29 @@
 ]
 
 if typing.TYPE_CHECKING:
-    _current_span: ContextVar[opentracing.Span]
+    _current_span: ContextVar[Span]
 _current_span = ContextVar("current_span")
 
 
-def current_span() -> Optional[opentracing.Span]:
+def current_span() -> Optional[Span]:
     """Get the current span for this context (if any)."""
     return _current_span.get(None)
 
 
-def set_current_span(span: opentracing.Span) -> None:
+def set_current_span(span: Span) -> None:
     """Set the current span for the current context."""
     _current_span.set(span)
 
 
-def noop_span() -> opentracing.Span:
-    """Return a span that does nothing when traced."""
-    return opentracing.Tracer()._noop_span
+def noop_span() -> Span:
+    """
+    Return a span that does nothing when traced.
+    OpenTelemetry Spans function as no-ops when not recording.
+    """
+    return Span
 
 
-def finish_span(
-    span: Optional[opentracing.Span], *, error: BaseException = None
-) -> None:
+def finish_span(span: Optional[Span], *, error: BaseException = None) -> None:
     """Finish span, and optionally set error tag."""
     if span is not None:
         if error:
@@ -65,7 +66,7 @@ def operation_name_from_fun(fun: Any) -> str:
 
 
 def traced_from_parent_span(
-    parent_span: opentracing.Span = None,
+    parent_span: Span = None,
     callback: Optional[Callable] = None,
     **extra_context: Any,
 ) -> Callable:
@@ -97,16 +98,14 @@ def _inner(*args: Any, **kwargs: Any) -> Any:
     return _wrapper
 
 
-def _restore_span(
-    span: opentracing.Span, expected_current_span: opentracing.Span
-) -> None:
+def _restore_span(span: Span, expected_current_span: Span) -> None:
     current = current_span()
     assert current is expected_current_span
     set_current_span(span)
 
 
 def call_with_trace(
-    span: opentracing.Span,
+    span: Span,
     fun: Callable,
     callback: Optional[Tuple[Callable, Tuple[Any, ...]]],
     *args: Any,

requirements/requirements.txt

@@ -3,7 +3,7 @@ aiohttp_cors>=0.7,<2.0
 aiokafka>=0.8.0,<0.9.0
 click>=6.7,<8.2
 mode-streaming>=0.3.0
-opentracing>=1.3.0,<=2.4.0
+opentelemetry-api==1.16.0
 terminaltables>=3.1,<4.0
 yarl>=1.0,<2.0
 croniter>=0.3.16