[server]: Use a GSI to fetch data in return_all_data

server/src/functions/return_all_data/lambda_function.py

@@ -1,26 +1,32 @@
-import boto3
 import json
 import os
+import boto3
 
 dynamodb = boto3.resource('dynamodb')
 table = dynamodb.Table(os.environ['TABLE_NAME'])
 
 def lambda_handler(event, context):
     try:
+        # Check if objectType is present in query string parameters
+        if 'queryStringParameters' in event and event['queryStringParameters'] and 'objectType' in event['queryStringParameters']:
+            objectType = event['queryStringParameters']['objectType']
+
+            # Query using objectType as the key (assumes GSI is created on objectType)
+            response = table.query(
+                IndexName='objectTypeIndex',  # Name of GSI
+                KeyConditionExpression=boto3.dynamodb.conditions.Key('objectType').eq(objectType)
+            )
+            items = response.get('Items', [])
+        else:
+            # Fallback to scanning the table (not recommended for large tables)
             items = []
             response = table.scan()
-
             items.extend(response.get('Items', []))
 
-        # continue to scan while there are more pages
             while 'LastEvaluatedKey' in response:
                 response = table.scan(ExclusiveStartKey=response['LastEvaluatedKey'])
                 items.extend(response.get('Items', []))
 
-        if 'queryStringParameters' in event and event['queryStringParameters'] and 'objectType' in event['queryStringParameters']:
-            objectType = event['queryStringParameters']['objectType']
-            items = [item for item in items if item['objectType'] == objectType]
-
         return {
             'statusCode': 200,
             'body': json.dumps(items)