Skip to content

Commit a1da811

Browse files
committed
feat: add processEventPayload function to handle event payloads
1 parent 9e44844 commit a1da811

File tree

1 file changed

+20
-10
lines changed

1 file changed

+20
-10
lines changed

src/function.py

Lines changed: 20 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -174,6 +174,24 @@ def _format_error(e, text):
174174
raise MaxRetriesException()
175175

176176

177+
def processEventPayload(event):
178+
"""
179+
Processes event payload for all formats.
180+
"""
181+
id, timestamp, message = event["id"], event["timestamp"], event["message"]
182+
reconstructed_message = message.split('\t')
183+
if len(reconstructed_message) >= 4:
184+
185+
message = reconstructed_message[3]
186+
event = {
187+
"id": id,
188+
"timestamp": timestamp,
189+
"message": message
190+
}
191+
192+
return event
193+
194+
177195
def _filter_log_lines(log_entry):
178196
"""
179197
The EntryType.LAMBDA check guarantees that we'll be left with at least one log after filtering
@@ -182,16 +200,8 @@ def _filter_log_lines(log_entry):
182200
for event in log_entry["logEvents"]:
183201
message = event["message"]
184202
if REPORT_PATTERN.match(message) or _is_lambda_message(message):
185-
id, timestamp, message = event["id"], event["timestamp"], event["message"]
186-
reconstructed_message = message.split('\t')
187-
if len(reconstructed_message) >= 4:
188-
message = reconstructed_message[3]
189-
event = {
190-
"id": id,
191-
"timestamp": timestamp,
192-
"message": message
193-
}
194-
final_log_events.append(event)
203+
updated_event = processEventPayload(event)
204+
final_log_events.append(updated_event)
195205

196206
ret = log_entry.copy()
197207
ret["logEvents"] = final_log_events

0 commit comments

Comments
 (0)