aindreas/fyp
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[server]: Make return_all_data accept a list of objectTypes

Browse Source
This commit is contained in:
Aindréas Ó hAodha
2025-03-08 15:59:50 +00:00
parent 33079b1eaf
commit 65fa4401bb
1 changed files with 21 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
server/src/functions/return_all_data/lambda_function.py
View File

@ -1,24 +1,37 @@
import json import json
import os import os
import boto3 import boto3
from boto3.dynamodb.conditions import Key, Attr
dynamodb = boto3.resource('dynamodb') dynamodb = boto3.resource('dynamodb')
table = dynamodb.Table(os.environ['TABLE_NAME']) table = dynamodb.Table(os.environ['TABLE_NAME'])
def lambda_handler(event, context): def lambda_handler(event, context):
try: try:
# Check if objectType is present in query string parameters # Check if objectType is present in query string parameters
if 'queryStringParameters' in event and event['queryStringParameters'] and 'objectType' in event['queryStringParameters']: if 'queryStringParameters' in event and event['queryStringParameters'] and 'objectType' in event[
'queryStringParameters']:
# Get objectType values and split by comma if multiple values are present
objectType = event['queryStringParameters']['objectType'] objectType = event['queryStringParameters']['objectType']
object_types = objectType.split(',')
# Query using objectType as the key (assumes GSI is created on objectType) # Fetch items matching any of the object types using a scan with FilterExpression
response = table.query( items = []
IndexName='objectTypeIndex', # Name of GSI response = table.scan(
KeyConditionExpression=boto3.dynamodb.conditions.Key('objectType').eq(objectType) FilterExpression=Attr('objectType').is_in(object_types)
) )
items = response.get('Items', []) items.extend(response.get('Items', []))
# Handle pagination
while 'LastEvaluatedKey' in response:
response = table.scan(
FilterExpression=Attr('objectType').is_in(object_types),
ExclusiveStartKey=response['LastEvaluatedKey']
)
items.extend(response.get('Items', []))
else: else:
# Fallback to scanning the table (not recommended for large tables) # Fallback to scanning the entire table
items = [] items = []
response = table.scan() response = table.scan()
items.extend(response.get('Items', [])) items.extend(response.get('Items', []))
@ -31,7 +44,7 @@ def lambda_handler(event, context):
'statusCode': 200, 'statusCode': 200,
'body': json.dumps(items) 'body': json.dumps(items)
} }
except Exception as e: except Exception as e:
return { return {
'statusCode': 500, 'statusCode': 500,