Skip to content

Commit 6781193

Browse files
committed
refactor(co): append Exit to SpanCodeOriginProcessor
1 parent 710e171 commit 6781193

File tree

4 files changed

+10
-10
lines changed

4 files changed

+10
-10
lines changed

ddtrace/debugging/_origin/span.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -231,10 +231,10 @@ def disable(cls):
231231

232232

233233
@dataclass
234-
class SpanCodeOriginProcessor(SpanProcessor):
234+
class SpanCodeOriginProcessorExit(SpanProcessor):
235235
__uploader__ = LogsIntakeUploaderV1
236236

237-
_instance: t.Optional["SpanCodeOriginProcessor"] = None
237+
_instance: t.Optional["SpanCodeOriginProcessorExit"] = None
238238

239239
def on_span_start(self, span: Span) -> None:
240240
if span.span_type not in EXIT_SPAN_TYPES:

ddtrace/debugging/_products/code_origin/span.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,11 @@ def post_preload():
1616

1717
def start():
1818
if config.span.enabled:
19-
from ddtrace.debugging._origin.span import SpanCodeOriginProcessor
19+
from ddtrace.debugging._origin.span import SpanCodeOriginProcessorExit
2020
from ddtrace.debugging._origin.span import SpanCodeOriginProcessorEntry
2121

2222
SpanCodeOriginProcessorEntry.enable()
23-
SpanCodeOriginProcessor.enable()
23+
SpanCodeOriginProcessorExit.enable()
2424
# If dynamic instrumentation is enabled, and code origin for spans is not explicitly disabled,
2525
# we'll enable entry spans only.
2626
elif product_manager.is_enabled(DI_PRODUCT_KEY) and config.value_source(CO_ENABLED) == ValueSource.DEFAULT:
@@ -35,11 +35,11 @@ def restart(join=False):
3535

3636
def stop(join=False):
3737
if config.span.enabled:
38-
from ddtrace.debugging._origin.span import SpanCodeOriginProcessor
38+
from ddtrace.debugging._origin.span import SpanCodeOriginProcessorExit
3939
from ddtrace.debugging._origin.span import SpanCodeOriginProcessorEntry
4040

4141
SpanCodeOriginProcessorEntry.disable()
42-
SpanCodeOriginProcessor.disable()
42+
SpanCodeOriginProcessorExit.disable()
4343
elif product_manager.is_enabled(DI_PRODUCT_KEY):
4444
from ddtrace.debugging._origin.span import SpanCodeOriginProcessorEntry
4545

tests/debugging/live/test_live_debugger.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import typing as t
33

44
import ddtrace
5-
from ddtrace.debugging._origin.span import SpanCodeOriginProcessor
5+
from ddtrace.debugging._origin.span import SpanCodeOriginProcessorExit
66
from ddtrace.debugging._probe.model import ProbeEvalTiming
77
from ddtrace.internal import core
88
from tests.debugging.mocking import MockLogsIntakeUploaderV1
@@ -12,7 +12,7 @@
1212
from tests.utils import TracerTestCase
1313

1414

15-
class MockSpanCodeOriginProcessor(SpanCodeOriginProcessor):
15+
class MockSpanCodeOriginProcessor(SpanCodeOriginProcessorExit):
1616
__uploader__ = MockLogsIntakeUploaderV1
1717

1818
@classmethod

tests/debugging/origin/test_span.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
import typing as t
33

44
import ddtrace
5-
from ddtrace.debugging._origin.span import SpanCodeOriginProcessor
5+
from ddtrace.debugging._origin.span import SpanCodeOriginProcessorExit
66
from ddtrace.debugging._origin.span import SpanCodeOriginProcessorEntry
77
from ddtrace.debugging._session import Session
88
from ddtrace.ext import SpanTypes
@@ -19,7 +19,7 @@ def get_uploader(cls) -> MockLogsIntakeUploaderV1:
1919
return t.cast(MockLogsIntakeUploaderV1, cls.__uploader__._instance)
2020

2121

22-
class MockSpanCodeOriginProcessor(SpanCodeOriginProcessor):
22+
class MockSpanCodeOriginProcessor(SpanCodeOriginProcessorExit):
2323
__uploader__ = MockLogsIntakeUploaderV1
2424

2525
@classmethod

0 commit comments

Comments
 (0)