Skip to content

Commit 722c233

Browse files
committed
Add richer logs to the record forwarder
1 parent 360d56f commit 722c233

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

lambdas/recordforwarder/src/forwarding_batch_lambda.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,8 +81,11 @@ def forward_lambda_handler(event, _):
8181
kinesis_payload = record["kinesis"]["data"]
8282
decoded_payload = base64.b64decode(kinesis_payload).decode("utf-8")
8383
incoming_message_body = json.loads(decoded_payload, use_decimal=True)
84+
file_key = incoming_message_body.get("file_key")
85+
local_id = incoming_message_body.get("local_id")
8486

8587
if is_eof_message(incoming_message_body):
88+
logger.info("Received EOF message for file key: %s", file_key)
8689
filename_to_events_mapper.add_event(incoming_message_body)
8790
continue
8891

@@ -95,6 +98,7 @@ def forward_lambda_handler(event, _):
9598
"supplier": incoming_message_body.get("supplier"),
9699
"vaccine_type": incoming_message_body.get("vax_type"),
97100
}
101+
logger.info("Received message for file %s with local id: %s", file_key, local_id)
98102

99103
try:
100104
if incoming_diagnostics := incoming_message_body.get("diagnostics"):
@@ -119,7 +123,8 @@ def forward_lambda_handler(event, _):
119123
else:
120124
array_of_identifiers.append(identifier)
121125

122-
forward_request_to_dynamo(incoming_message_body, table, identifier_already_present, controller)
126+
imms_pk = forward_request_to_dynamo(incoming_message_body, table, identifier_already_present, controller)
127+
logger.info("Successfully processed message. Local id: %s, PK: %s", local_id, imms_pk)
123128

124129
except Exception as error: # pylint: disable = broad-exception-caught
125130
filename_to_events_mapper.add_event(

0 commit comments

Comments
 (0)