Enhanced DynamoDB client that automatically handles LastEvaluatedKeys

enhanced-dynamodb-client.js

'use strict';

const AWS = require("aws-sdk");

class EnhancedDynamoDBClient extends AWS.DynamoDB.DocumentClient {
    constructor(options) {
        super(options);
    }

    scanAll(params) {
        const doScan = (items) => {
            return this.scan(params).promise()
                .then(result => {
                    items.push(...result.Items);

                    if (!result.LastEvaluatedKey) {
                        return items;
                    }
                    else {
                        console.log(result.LastEvaluatedKey)
                        params.ExclusiveStartKey = result.LastEvaluatedKey;
                        return doScan(items);
                    }
                });
        };

        return doScan([]);
    }

    queryAll(params) {
        const doQuery = (items) => {
            return this.query(params).promise()
                .then(result => {
                    items.push(...result.Items);

                    if (!result.LastEvaluatedKey) {
                        return items;
                    }
                    else {
                        params.ExclusiveStartKey = result.LastEvaluatedKey;
                        return doQuery(items);
                    }
                });
        };

        return doQuery([]);
    }
}