Skip to content

Commit 709a7b5

Browse files
catch invocation event tag extraction exceptions (#157)
1 parent 7ff4a00 commit 709a7b5

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,9 +135,9 @@ def __call__(self, event, context, **kwargs):
136135

137136
def _before(self, event, context):
138137
try:
139-
140138
set_cold_start()
141139
submit_invocations_metric(context)
140+
self.trigger_tags = extract_trigger_tags(event, context)
142141
# Extract Datadog trace context and source from incoming requests
143142
dd_context, trace_context_source = extract_dd_trace_context(
144143
event, context, extractor=self.trace_extractor

0 commit comments

Comments
 (0)