From b5c43ca50b1109dfefc06419ef554967d9837bdd Mon Sep 17 00:00:00 2001 From: Simon Li Date: Wed, 20 Mar 2024 10:54:06 +0000 Subject: [PATCH] Paginate list_requests DynamoDb.scan --- egress_backend/lambda/egress_api/list_requests.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/egress_backend/lambda/egress_api/list_requests.py b/egress_backend/lambda/egress_api/list_requests.py index 37aad4e..c4b3250 100644 --- a/egress_backend/lambda/egress_api/list_requests.py +++ b/egress_backend/lambda/egress_api/list_requests.py @@ -21,6 +21,10 @@ def list_requests(): ddb_table = ddb.Table(table) response = ddb_table.scan() + data = response["Items"] + while response.get("LastEvaluatedKey"): + response = ddb_table.scan(ExclusiveStartKey=response["LastEvaluatedKey"]) + data.extend(response["Items"]) - logger.debug("Succesful database scan of all egress requests") - return response["Items"] + logger.debug("Successful database scan of all egress requests") + return data