event

Author: nhsdevws

lambdas/id_sync/src/id_sync.py

@@ -1,6 +1,8 @@
 from common.clients import logger
 from common.clients import STREAM_NAME
 from common.log_decorator import logging_decorator
+from common.aws_lambda_event import AwsLambdaEvent
+from common.aws_lambda_sqs_event_record import AwsLambdaSqsEventRecord
 from record_processor import process_record
 
 '''
@@ -10,31 +12,29 @@
 
 
 @logging_decorator(prefix="id_sync", stream_name=STREAM_NAME)
-def handler(event, _):
+def handler(event_data, _):
 
     try:
-        if "Records" in event:
-            logger.info("Processing S3 event with %d records", len(event.get('Records', [])))
-            record_count = len(event.get('Records', []))
-            if record_count == 0:
-                logger.info("No records found in event")
-                return {"status": "success", "message": "No records found in event"}
+        event = AwsLambdaEvent(event_data)
+        record_count = len(event.records)
+        if record_count > 0:
+            logger.info("Processing SQS event with %d records", record_count)
+            error_count = 0
+            file_keys = []
+            for record in event.records:
+                record: AwsLambdaSqsEventRecord
+                record_result = process_record(record, None)
+                file_keys.append(record_result["file_key"])
+                if record_result["status"] == "error":
+                    error_count += 1
+            if error_count > 0:
+                logger.error("Processed %d records with %d errors", record_count, error_count)
+                return {"status": "error", "message": f"Processed {record_count} records with {error_count} errors",
+                        "file_keys": file_keys}
             else:
-                error_count = 0
-                file_keys = []
-                for record in event.get('Records', []):
-                    record_result = process_record(record, None)
-                    file_keys.append(record_result["file_key"])
-                    if record_result["status"] == "error":
-                        error_count += 1
-                if error_count > 0:
-                    logger.error("Processed %d records with %d errors", record_count, error_count)
-                    return {"status": "error", "message": f"Processed {record_count} records with {error_count} errors",
-                            "file_keys": file_keys}
-                else:
-                    logger.info("Successfully processed all %d records", record_count)
-                    return {"status": "success", "message": f"Successfully processed {record_count} records",
-                            "file_keys": file_keys}
+                logger.info("Successfully processed all %d records", record_count)
+                return {"status": "success", "message": f"Successfully processed {record_count} records",
+                        "file_keys": file_keys}
         else:
             logger.info("No records found in event")
             return {"status": "success", "message": "No records found in event"}

lambdas/id_sync/src/record_processor.py

@@ -1,7 +1,11 @@
 '''
     record Processor
 '''
+from common.aws_lambda_sqs_event_record import AwsLambdaSqsEventRecord
+from common.clients import logger
 
 
-def process_record(record, _):
+def process_record(event_record, _):
+    record = AwsLambdaSqsEventRecord(event_record) if isinstance(event_record, dict) else event_record
+    logger.info("Processing record: %s", record)
     return f"hello world {record}"