@amazon-dax-sdk/client-dax/src/paginators/DaxPaginateQuery.js

Amazon DAX Client for JavaScript

/*
 * Copyright 2024 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"). You may not
 * use this file except in compliance with the License. A copy of the License
 * is located at
 *
 *    http://aws.amazon.com/apache2.0/
 *
 * or in the "license" file accompanying this file. This file is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

const Dax = require('../Dax');
const daxPaginateQuery = async function* (config, input) {
  let token = config.startingToken || undefined;
  let hasNext = true;
  let page;
  while(hasNext) {
    input.ExclusiveStartKey = token || input.ExclusiveStartKey;
    input['Limit'] = input['Limit'] ?? config.pageSize;
    if(config.client instanceof Dax) {
      page = await config.client.query(input);
    } else {
      throw new Error('Invalid client, expected instance of Dax');
    }
    yield page;
    const prevToken = token;
    token = page.LastEvaluatedKey;
    hasNext = !!(token && (!config.stopOnSameToken || token !== prevToken));
  }
  return undefined;
};

module.exports = daxPaginateQuery;