chore(lib-dynamodb): codegen sync

aws · Aug 1, 2024 · 671f12c · 671f12c
1 parent db8eef8
commit 671f12c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 72 deletions.
diff --git a/lib/lib-dynamodb/src/pagination/QueryPaginator.ts b/lib/lib-dynamodb/src/pagination/QueryPaginator.ts
@@ -1,4 +1,5 @@
 // smithy-typescript generated code
+import { createPaginator } from "@smithy/core";
 import { Paginator } from "@smithy/types";
 
 import { QueryCommand, QueryCommandInput, QueryCommandOutput } from "../commands/QueryCommand";
@@ -9,45 +10,15 @@ import { DynamoDBDocumentPaginationConfiguration } from "./Interfaces";
  * @public
  */
 export { Paginator };
-/**
- * @internal
- */
-const makePagedClientRequest = async (
-  client: DynamoDBDocumentClient,
-  input: QueryCommandInput,
-  ...args: any
-): Promise<QueryCommandOutput> => {
-  // @ts-ignore
-  return await client.send(new QueryCommand(input), ...args);
-};
 /**
  * @public
- *
- * @param QueryCommandInput - {@link QueryCommandInput}
- * @returns {@link QueryCommandOutput}
- *
  */
-export async function* paginateQuery(
+export const paginateQuery: (
   config: DynamoDBDocumentPaginationConfiguration,
   input: QueryCommandInput,
   ...additionalArguments: any
-): Paginator<QueryCommandOutput> {
-  // ToDo: replace with actual type instead of typeof input.ExclusiveStartKey
-  let token: typeof input.ExclusiveStartKey | undefined = config.startingToken || undefined;
-  let hasNext = true;
-  let page: QueryCommandOutput;
-  while (hasNext) {
-    input.ExclusiveStartKey = token;
-    input["Limit"] = config.pageSize;
-    if (config.client instanceof DynamoDBDocumentClient) {
-      page = await makePagedClientRequest(config.client, input, ...additionalArguments);
-    } else {
-      throw new Error("Invalid client, expected DynamoDBDocument | DynamoDBDocumentClient");
-    }
-    yield page;
-    token = page.LastEvaluatedKey;
-    hasNext = !!token;
-  }
-  // @ts-ignore
-  return undefined;
-}
+) => Paginator<QueryCommandOutput> = createPaginator<
+  DynamoDBDocumentPaginationConfiguration,
+  QueryCommandInput,
+  QueryCommandOutput
+>(DynamoDBDocumentClient, QueryCommand, "ExclusiveStartKey", "LastEvaluatedKey", "Limit");
diff --git a/lib/lib-dynamodb/src/pagination/ScanPaginator.ts b/lib/lib-dynamodb/src/pagination/ScanPaginator.ts
@@ -1,4 +1,5 @@
 // smithy-typescript generated code
+import { createPaginator } from "@smithy/core";
 import { Paginator } from "@smithy/types";
 
 import { ScanCommand, ScanCommandInput, ScanCommandOutput } from "../commands/ScanCommand";
@@ -9,45 +10,15 @@ import { DynamoDBDocumentPaginationConfiguration } from "./Interfaces";
  * @public
  */
 export { Paginator };
-/**
- * @internal
- */
-const makePagedClientRequest = async (
-  client: DynamoDBDocumentClient,
-  input: ScanCommandInput,
-  ...args: any
-): Promise<ScanCommandOutput> => {
-  // @ts-ignore
-  return await client.send(new ScanCommand(input), ...args);
-};
 /**
  * @public
- *
- * @param ScanCommandInput - {@link ScanCommandInput}
- * @returns {@link ScanCommandOutput}
- *
  */
-export async function* paginateScan(
+export const paginateScan: (
   config: DynamoDBDocumentPaginationConfiguration,
   input: ScanCommandInput,
   ...additionalArguments: any
-): Paginator<ScanCommandOutput> {
-  // ToDo: replace with actual type instead of typeof input.ExclusiveStartKey
-  let token: typeof input.ExclusiveStartKey | undefined = config.startingToken || undefined;
-  let hasNext = true;
-  let page: ScanCommandOutput;
-  while (hasNext) {
-    input.ExclusiveStartKey = token;
-    input["Limit"] = config.pageSize;
-    if (config.client instanceof DynamoDBDocumentClient) {
-      page = await makePagedClientRequest(config.client, input, ...additionalArguments);
-    } else {
-      throw new Error("Invalid client, expected DynamoDBDocument | DynamoDBDocumentClient");
-    }
-    yield page;
-    token = page.LastEvaluatedKey;
-    hasNext = !!token;
-  }
-  // @ts-ignore
-  return undefined;
-}
+) => Paginator<ScanCommandOutput> = createPaginator<
+  DynamoDBDocumentPaginationConfiguration,
+  ScanCommandInput,
+  ScanCommandOutput
+>(DynamoDBDocumentClient, ScanCommand, "ExclusiveStartKey", "LastEvaluatedKey", "Limit");