Skip to content

Commit 3b04b06

Browse files
committed
remove private import of ddtrace functionality
1 parent 3bf5778 commit 3b04b06

1 file changed

Lines changed: 9 additions & 9 deletions

File tree

datadog_lambda/wrapper.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
from importlib import import_module
1010
from time import time_ns
1111

12-
from ddtrace.internal._exceptions import BlockingException
1312
from datadog_lambda.extension import should_use_extension, flush_extension
1413
from datadog_lambda.cold_start import (
1514
set_cold_start,
@@ -191,16 +190,17 @@ def __call__(self, event, context, **kwargs):
191190
if self.blocking_response:
192191
return self.blocking_response
193192
self.response = self.func(event, context, **kwargs)
194-
except BlockingException:
195-
self.blocking_response = get_asm_blocked_response(self.event_source)
196-
except Exception:
197-
from datadog_lambda.metric import submit_errors_metric
193+
except Exception as e:
194+
if type(e).__name__ == "BlockingException":
195+
self.blocking_response = get_asm_blocked_response(self.event_source)
196+
else:
197+
from datadog_lambda.metric import submit_errors_metric
198198

199-
submit_errors_metric(context)
199+
submit_errors_metric(context)
200200

201-
if self.span:
202-
self.span.set_traceback()
203-
raise
201+
if self.span:
202+
self.span.set_traceback()
203+
raise
204204
finally:
205205
self._after(event, context)
206206
if self.blocking_response:

0 commit comments

Comments
 (0)