diff --git a/synapse/logging/opentracing.py b/synapse/logging/opentracing.py
index b240d2d21d..d25f25ecb5 100644
--- a/synapse/logging/opentracing.py
+++ b/synapse/logging/opentracing.py
@@ -443,10 +443,14 @@ def start_active_span(
start_time=None,
ignore_active_span=False,
finish_on_close=True,
+ *,
+ tracer=None,
):
- """Starts an active opentracing span. Note, the scope doesn't become active
- until it has been entered, however, the span starts from the time this
- message is called.
+ """Starts an active opentracing span.
+
+ Records the start time for the span, and sets it as the "active span" in the
+ scope manager.
+
Args:
See opentracing.tracer
Returns:
@@ -456,7 +460,11 @@ def start_active_span(
if opentracing is None:
return noop_context_manager() # type: ignore[unreachable]
- return opentracing.tracer.start_active_span(
+ if tracer is None:
+ # use the global tracer by default
+ tracer = opentracing.tracer
+
+ return tracer.start_active_span(
operation_name,
child_of=child_of,
references=references,
@@ -468,7 +476,11 @@ def start_active_span(
def start_active_span_follows_from(
- operation_name: str, contexts: Collection, inherit_force_tracing=False
+ operation_name: str,
+ contexts: Collection,
+ *,
+ inherit_force_tracing=False,
+ tracer=None,
):
"""Starts an active opentracing span, with additional references to previous spans
@@ -477,12 +489,17 @@ def start_active_span_follows_from(
contexts: the previous spans to inherit from
inherit_force_tracing: if set, and any of the previous contexts have had tracing
forced, the new span will also have tracing forced.
+ tracer: override the opentracing tracer. By default the global tracer is used.
"""
if opentracing is None:
return noop_context_manager() # type: ignore[unreachable]
references = [opentracing.follows_from(context) for context in contexts]
- scope = start_active_span(operation_name, references=references)
+ scope = start_active_span(
+ operation_name,
+ references=references,
+ tracer=tracer,
+ )
if inherit_force_tracing and any(
is_context_forced_tracing(ctx) for ctx in contexts
|
