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

[server]: Add pagination handling to return_all_data

Browse Source
This commit is contained in:
Aindréas Ó hAodha
2025-02-28 23:28:39 +00:00
parent 47e6cb5dc5
commit 4357c4b518
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -7,9 +7,15 @@ table = dynamodb.Table(os.environ['TABLE_NAME'])
def lambda_handler(event, context): def lambda_handler(event, context):
try: try:
items = []
response = table.scan() response = table.scan()
items = response.get('Items', []) 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']: if 'queryStringParameters' in event and event['queryStringParameters'] and 'objectType' in event['queryStringParameters']:
objectType = event['queryStringParameters']['objectType'] objectType = event['queryStringParameters']['objectType']