feat(client-marketplace-entitlement-service): Add paginators to GetEntitlements.

Author: awstools

clients/client-marketplace-entitlement-service/src/index.ts

@@ -27,6 +27,7 @@ export { ClientInputEndpointParameters } from "./endpoint/EndpointParameters";
 export { RuntimeExtension } from "./runtimeExtensions";
 export { MarketplaceEntitlementServiceExtensionConfiguration } from "./extensionConfiguration";
 export * from "./commands";
+export * from "./pagination";
 export * from "./models";
 
 import "@aws-sdk/util-endpoints";

clients/client-marketplace-entitlement-service/src/pagination/GetEntitlementsPaginator.ts

@@ -0,0 +1,50 @@
+// smithy-typescript generated code
+import { Paginator } from "@smithy/types";
+
+import {
+  GetEntitlementsCommand,
+  GetEntitlementsCommandInput,
+  GetEntitlementsCommandOutput,
+} from "../commands/GetEntitlementsCommand";
+import { MarketplaceEntitlementServiceClient } from "../MarketplaceEntitlementServiceClient";
+import { MarketplaceEntitlementServicePaginationConfiguration } from "./Interfaces";
+
+/**
+ * @internal
+ */
+const makePagedClientRequest = async (
+  client: MarketplaceEntitlementServiceClient,
+  input: GetEntitlementsCommandInput,
+  ...args: any
+): Promise<GetEntitlementsCommandOutput> => {
+  // @ts-ignore
+  return await client.send(new GetEntitlementsCommand(input), ...args);
+};
+/**
+ * @public
+ */
+export async function* paginateGetEntitlements(
+  config: MarketplaceEntitlementServicePaginationConfiguration,
+  input: GetEntitlementsCommandInput,
+  ...additionalArguments: any
+): Paginator<GetEntitlementsCommandOutput> {
+  // ToDo: replace with actual type instead of typeof input.NextToken
+  let token: typeof input.NextToken | undefined = config.startingToken || undefined;
+  let hasNext = true;
+  let page: GetEntitlementsCommandOutput;
+  while (hasNext) {
+    input.NextToken = token;
+    input["MaxResults"] = config.pageSize;
+    if (config.client instanceof MarketplaceEntitlementServiceClient) {
+      page = await makePagedClientRequest(config.client, input, ...additionalArguments);
+    } else {
+      throw new Error("Invalid client, expected MarketplaceEntitlementService | MarketplaceEntitlementServiceClient");
+    }
+    yield page;
+    const prevToken = token;
+    token = page.NextToken;
+    hasNext = !!(token && (!config.stopOnSameToken || token !== prevToken));
+  }
+  // @ts-ignore
+  return undefined;
+}

clients/client-marketplace-entitlement-service/src/pagination/Interfaces.ts

@@ -0,0 +1,11 @@
+// smithy-typescript generated code
+import { PaginationConfiguration } from "@smithy/types";
+
+import { MarketplaceEntitlementServiceClient } from "../MarketplaceEntitlementServiceClient";
+
+/**
+ * @public
+ */
+export interface MarketplaceEntitlementServicePaginationConfiguration extends PaginationConfiguration {
+  client: MarketplaceEntitlementServiceClient;
+}

clients/client-marketplace-entitlement-service/src/pagination/index.ts

@@ -0,0 +1,3 @@
+export * from "./GetEntitlementsPaginator";
+// smithy-typescript generated code
+export * from "./Interfaces";

codegen/sdk-codegen/aws-models/marketplace-entitlement-service.json

@@ -857,7 +857,12 @@
         }
       ],
       "traits": {
-        "smithy.api#documentation": "<p>GetEntitlements retrieves entitlement values for a given product. The results can be\n      filtered based on customer identifier or product dimensions.</p>"
+        "smithy.api#documentation": "<p>GetEntitlements retrieves entitlement values for a given product. The results can be\n      filtered based on customer identifier or product dimensions.</p>",
+        "smithy.api#paginated": {
+          "inputToken": "NextToken",
+          "outputToken": "NextToken",
+          "pageSize": "MaxResults"
+        }
       }
     },
     "com.amazonaws.marketplaceentitlementservice#GetEntitlementsRequest": {
@@ -883,7 +888,7 @@
           }
         },
         "MaxResults": {
-          "target": "com.amazonaws.marketplaceentitlementservice#Integer",
+          "target": "com.amazonaws.marketplaceentitlementservice#PageSizeInteger",
           "traits": {
             "smithy.api#documentation": "<p>The maximum number of items to retrieve from the GetEntitlements operation. For\n   pagination, use the NextToken field in subsequent calls to GetEntitlements.</p>"
           }
@@ -948,6 +953,15 @@
         "smithy.api#pattern": "^\\S+$"
       }
     },
+    "com.amazonaws.marketplaceentitlementservice#PageSizeInteger": {
+      "type": "integer",
+      "traits": {
+        "smithy.api#range": {
+          "min": 1,
+          "max": 25
+        }
+      }
+    },
     "com.amazonaws.marketplaceentitlementservice#ProductCode": {
       "type": "string",
       "traits": {