Skip to content

Commit

Permalink
simpler span proxying code
Browse files Browse the repository at this point in the history
  • Loading branch information
emmettbutler committed Feb 7, 2025
1 parent 2f07b12 commit 94a7494
Showing 1 changed file with 5 additions and 13 deletions.
18 changes: 5 additions & 13 deletions ddtrace/contrib/internal/dd_trace_api/patch.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,19 +19,11 @@

def _proxy_span_arguments(args: List, kwargs: Dict) -> Tuple[List, Dict]:
"""Convert all dd_trace_api.Span objects in the args/kwargs collections to their held ddtrace.Span objects"""
proxied_args = []
for arg in args:
if isinstance(arg, dd_trace_api.Span):
proxied_args.append(_STUB_TO_REAL[arg])
else:
proxied_args.append(arg)
proxied_kwargs = {}
for name, kwarg in kwargs.items():
if isinstance(kwarg, dd_trace_api.Span):
proxied_kwargs[name] = _STUB_TO_REAL[kwarg]
else:
proxied_kwargs[name] = kwarg
return proxied_args, proxied_kwargs

def convert(arg):
return _STUB_TO_REAL[arg] if isinstance(arg, dd_trace_api.Span) else arg

return [convert(arg) for arg in args], {name: convert(kwarg) for name, kwarg in kwargs.items()}


def _call_on_real_instance(
Expand Down

0 comments on commit 94a7494

Please sign in to comment.