Skip to content

Commit d787b3d

Browse files
catch errors if extracting tags from invocation event fails
1 parent f01ea24 commit d787b3d

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

datadog_lambda/wrapper.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,7 @@ def __init__(self, func):
9797
self.extractor_env = os.environ.get("DD_TRACE_EXTRACTOR", None)
9898
self.trace_extractor = None
9999
self.span = None
100+
self.response = None
100101

101102
if self.extractor_env:
102103
extractor_parts = self.extractor_env.rsplit(".", 1)
@@ -119,8 +120,6 @@ def __init__(self, func):
119120

120121
def __call__(self, event, context, **kwargs):
121122
"""Executes when the wrapped function gets called"""
122-
self.trigger_tags = extract_trigger_tags(event, context)
123-
self.response = None
124123
init_lambda_stats()
125124
self._before(event, context)
126125
try:
@@ -136,7 +135,7 @@ def __call__(self, event, context, **kwargs):
136135

137136
def _before(self, event, context):
138137
try:
139-
138+
self.trigger_tags = extract_trigger_tags(event, context)
140139
set_cold_start()
141140
submit_invocations_metric(context)
142141
# Extract Datadog trace context and source from incoming requests

0 commit comments

Comments
 (0)