@azure/arm-datamigration 2.0.0 → 3.0.0-beta.1

package/src/operations/resourceSkus.ts

@@ -1,100 +1,112 @@
 /*
- * Copyright (c) Microsoft Corporation. All rights reserved.
- * Licensed under the MIT License. See License.txt in the project root for
- * license information.
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT License.
  *
  * Code generated by Microsoft (R) AutoRest Code Generator.
- * Changes may cause incorrect behavior and will be lost if the code is
- * regenerated.
+ * Changes may cause incorrect behavior and will be lost if the code is regenerated.
  */
 
-import * as msRest from "@azure/ms-rest-js";
-import * as Models from "../models";
-import * as Mappers from "../models/resourceSkusMappers";
+import { PagedAsyncIterableIterator } from "@azure/core-paging";
+import { ResourceSkus } from "../operationsInterfaces";
+import * as coreClient from "@azure/core-client";
+import * as Mappers from "../models/mappers";
 import * as Parameters from "../models/parameters";
-import { DataMigrationServiceClientContext } from "../dataMigrationServiceClientContext";
+import { DataMigrationManagementClient } from "../dataMigrationManagementClient";
+import {
+  ResourceSku,
+  ResourceSkusListSkusNextOptionalParams,
+  ResourceSkusListSkusOptionalParams,
+  ResourceSkusListSkusResponse,
+  ResourceSkusListSkusNextResponse
+} from "../models";
 
-/** Class representing a ResourceSkus. */
-export class ResourceSkus {
-  private readonly client: DataMigrationServiceClientContext;
+/// <reference lib="esnext.asynciterable" />
+/** Class containing ResourceSkus operations. */
+export class ResourceSkusImpl implements ResourceSkus {
+  private readonly client: DataMigrationManagementClient;
 
   /**
-   * Create a ResourceSkus.
-   * @param {DataMigrationServiceClientContext} client Reference to the service client.
+   * Initialize a new instance of the class ResourceSkus class.
+   * @param client Reference to the service client
    */
-  constructor(client: DataMigrationServiceClientContext) {
+  constructor(client: DataMigrationManagementClient) {
     this.client = client;
   }
 
   /**
    * The skus action returns the list of SKUs that DMS supports.
-   * @summary Get supported SKUs
-   * @param [options] The optional parameters
-   * @returns Promise<Models.ResourceSkusListSkusResponse>
+   * @param options The options parameters.
    */
-  listSkus(options?: msRest.RequestOptionsBase): Promise<Models.ResourceSkusListSkusResponse>;
-  /**
-   * @param callback The callback
-   */
-  listSkus(callback: msRest.ServiceCallback<Models.ResourceSkusResult>): void;
-  /**
-   * @param options The optional parameters
-   * @param callback The callback
-   */
-  listSkus(options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.ResourceSkusResult>): void;
-  listSkus(options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.ResourceSkusResult>, callback?: msRest.ServiceCallback<Models.ResourceSkusResult>): Promise<Models.ResourceSkusListSkusResponse> {
-    return this.client.sendOperationRequest(
-      {
-        options
+  public listSkus(
+    options?: ResourceSkusListSkusOptionalParams
+  ): PagedAsyncIterableIterator<ResourceSku> {
+    const iter = this.listSkusPagingAll(options);
+    return {
+      next() {
+        return iter.next();
+      },
+      [Symbol.asyncIterator]() {
+        return this;
       },
-      listSkusOperationSpec,
-      callback) as Promise<Models.ResourceSkusListSkusResponse>;
+      byPage: () => {
+        return this.listSkusPagingPage(options);
+      }
+    };
+  }
+
+  private async *listSkusPagingPage(
+    options?: ResourceSkusListSkusOptionalParams
+  ): AsyncIterableIterator<ResourceSku[]> {
+    let result = await this._listSkus(options);
+    yield result.value || [];
+    let continuationToken = result.nextLink;
+    while (continuationToken) {
+      result = await this._listSkusNext(continuationToken, options);
+      continuationToken = result.nextLink;
+      yield result.value || [];
+    }
+  }
+
+  private async *listSkusPagingAll(
+    options?: ResourceSkusListSkusOptionalParams
+  ): AsyncIterableIterator<ResourceSku> {
+    for await (const page of this.listSkusPagingPage(options)) {
+      yield* page;
+    }
   }
 
   /**
    * The skus action returns the list of SKUs that DMS supports.
-   * @summary Get supported SKUs
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.ResourceSkusListSkusNextResponse>
+   * @param options The options parameters.
    */
-  listSkusNext(nextPageLink: string, options?: msRest.RequestOptionsBase): Promise<Models.ResourceSkusListSkusNextResponse>;
-  /**
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param callback The callback
-   */
-  listSkusNext(nextPageLink: string, callback: msRest.ServiceCallback<Models.ResourceSkusResult>): void;
+  private _listSkus(
+    options?: ResourceSkusListSkusOptionalParams
+  ): Promise<ResourceSkusListSkusResponse> {
+    return this.client.sendOperationRequest({ options }, listSkusOperationSpec);
+  }
+
   /**
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param options The optional parameters
-   * @param callback The callback
+   * ListSkusNext
+   * @param nextLink The nextLink from the previous successful call to the ListSkus method.
+   * @param options The options parameters.
    */
-  listSkusNext(nextPageLink: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.ResourceSkusResult>): void;
-  listSkusNext(nextPageLink: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.ResourceSkusResult>, callback?: msRest.ServiceCallback<Models.ResourceSkusResult>): Promise<Models.ResourceSkusListSkusNextResponse> {
+  private _listSkusNext(
+    nextLink: string,
+    options?: ResourceSkusListSkusNextOptionalParams
+  ): Promise<ResourceSkusListSkusNextResponse> {
     return this.client.sendOperationRequest(
-      {
-        nextPageLink,
-        options
-      },
-      listSkusNextOperationSpec,
-      callback) as Promise<Models.ResourceSkusListSkusNextResponse>;
+      { nextLink, options },
+      listSkusNextOperationSpec
+    );
   }
 }
-
 // Operation Specifications
-const serializer = new msRest.Serializer(Mappers);
-const listSkusOperationSpec: msRest.OperationSpec = {
+const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
+
+const listSkusOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/providers/Microsoft.DataMigration/skus",
   httpMethod: "GET",
-  path: "subscriptions/{subscriptionId}/providers/Microsoft.DataMigration/skus",
-  urlParameters: [
-    Parameters.subscriptionId
-  ],
-  queryParameters: [
-    Parameters.apiVersion
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
   responses: {
     200: {
       bodyMapper: Mappers.ResourceSkusResult
@@ -103,19 +115,14 @@ const listSkusOperationSpec: msRest.OperationSpec = {
       bodyMapper: Mappers.ApiError
     }
   },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [Parameters.$host, Parameters.subscriptionId],
+  headerParameters: [Parameters.accept],
   serializer
 };
-
-const listSkusNextOperationSpec: msRest.OperationSpec = {
-  httpMethod: "GET",
-  baseUrl: "https://management.azure.com",
+const listSkusNextOperationSpec: coreClient.OperationSpec = {
   path: "{nextLink}",
-  urlParameters: [
-    Parameters.nextPageLink
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
+  httpMethod: "GET",
   responses: {
     200: {
       bodyMapper: Mappers.ResourceSkusResult
@@ -124,5 +131,12 @@ const listSkusNextOperationSpec: msRest.OperationSpec = {
       bodyMapper: Mappers.ApiError
     }
   },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.nextLink
+  ],
+  headerParameters: [Parameters.accept],
   serializer
 };