@azure/arm-compute 16.5.0 → 17.0.0-beta.2

package/src/operations/disks.ts

@@ -3,625 +3,871 @@
  * 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 msRestAzure from "@azure/ms-rest-azure-js";
-import * as Models from "../models";
-import * as Mappers from "../models/disksMappers";
+import { PagedAsyncIterableIterator } from "@azure/core-paging";
+import { Disks } from "../operationsInterfaces";
+import * as coreClient from "@azure/core-client";
+import * as Mappers from "../models/mappers";
 import * as Parameters from "../models/parameters";
 import { ComputeManagementClientContext } from "../computeManagementClientContext";
+import { PollerLike, PollOperationState, LroEngine } from "@azure/core-lro";
+import { LroImpl } from "../lroImpl";
+import {
+  Disk,
+  DisksListByResourceGroupNextOptionalParams,
+  DisksListByResourceGroupOptionalParams,
+  DisksListNextOptionalParams,
+  DisksListOptionalParams,
+  DisksCreateOrUpdateOptionalParams,
+  DisksCreateOrUpdateResponse,
+  DiskUpdate,
+  DisksUpdateOptionalParams,
+  DisksUpdateResponse,
+  DisksGetOptionalParams,
+  DisksGetResponse,
+  DisksDeleteOptionalParams,
+  DisksListByResourceGroupResponse,
+  DisksListResponse,
+  GrantAccessData,
+  DisksGrantAccessOptionalParams,
+  DisksGrantAccessResponse,
+  DisksRevokeAccessOptionalParams,
+  DisksListByResourceGroupNextResponse,
+  DisksListNextResponse
+} from "../models";
 
-/** Class representing a Disks. */
-export class Disks {
+/// <reference lib="esnext.asynciterable" />
+/** Class containing Disks operations. */
+export class DisksImpl implements Disks {
   private readonly client: ComputeManagementClientContext;
 
   /**
-   * Create a Disks.
-   * @param {ComputeManagementClientContext} client Reference to the service client.
+   * Initialize a new instance of the class Disks class.
+   * @param client Reference to the service client
    */
   constructor(client: ComputeManagementClientContext) {
     this.client = client;
   }
 
   /**
-   * Creates or updates a disk.
+   * Lists all the disks under a resource group.
    * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param disk Disk object supplied in the body of the Put disk operation.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.DisksCreateOrUpdateResponse>
+   * @param options The options parameters.
    */
-  createOrUpdate(resourceGroupName: string, diskName: string, disk: Models.Disk, options?: msRest.RequestOptionsBase): Promise<Models.DisksCreateOrUpdateResponse> {
-    return this.beginCreateOrUpdate(resourceGroupName,diskName,disk,options)
-      .then(lroPoller => lroPoller.pollUntilFinished()) as Promise<Models.DisksCreateOrUpdateResponse>;
+  public listByResourceGroup(
+    resourceGroupName: string,
+    options?: DisksListByResourceGroupOptionalParams
+  ): PagedAsyncIterableIterator<Disk> {
+    const iter = this.listByResourceGroupPagingAll(resourceGroupName, options);
+    return {
+      next() {
+        return iter.next();
+      },
+      [Symbol.asyncIterator]() {
+        return this;
+      },
+      byPage: () => {
+        return this.listByResourceGroupPagingPage(resourceGroupName, options);
+      }
+    };
   }
 
-  /**
-   * Updates (patches) a disk.
-   * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param disk Disk object supplied in the body of the Patch disk operation.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.DisksUpdateResponse>
-   */
-  update(resourceGroupName: string, diskName: string, disk: Models.DiskUpdate, options?: msRest.RequestOptionsBase): Promise<Models.DisksUpdateResponse> {
-    return this.beginUpdate(resourceGroupName,diskName,disk,options)
-      .then(lroPoller => lroPoller.pollUntilFinished()) as Promise<Models.DisksUpdateResponse>;
+  private async *listByResourceGroupPagingPage(
+    resourceGroupName: string,
+    options?: DisksListByResourceGroupOptionalParams
+  ): AsyncIterableIterator<Disk[]> {
+    let result = await this._listByResourceGroup(resourceGroupName, options);
+    yield result.value || [];
+    let continuationToken = result.nextLink;
+    while (continuationToken) {
+      result = await this._listByResourceGroupNext(
+        resourceGroupName,
+        continuationToken,
+        options
+      );
+      continuationToken = result.nextLink;
+      yield result.value || [];
+    }
+  }
+
+  private async *listByResourceGroupPagingAll(
+    resourceGroupName: string,
+    options?: DisksListByResourceGroupOptionalParams
+  ): AsyncIterableIterator<Disk> {
+    for await (const page of this.listByResourceGroupPagingPage(
+      resourceGroupName,
+      options
+    )) {
+      yield* page;
+    }
   }
 
   /**
-   * Gets information about a disk.
-   * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.DisksGetResponse>
-   */
-  get(resourceGroupName: string, diskName: string, options?: msRest.RequestOptionsBase): Promise<Models.DisksGetResponse>;
-  /**
-   * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param callback The callback
-   */
-  get(resourceGroupName: string, diskName: string, callback: msRest.ServiceCallback<Models.Disk>): void;
-  /**
-   * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param options The optional parameters
-   * @param callback The callback
+   * Lists all the disks under a subscription.
+   * @param options The options parameters.
    */
-  get(resourceGroupName: string, diskName: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.Disk>): void;
-  get(resourceGroupName: string, diskName: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.Disk>, callback?: msRest.ServiceCallback<Models.Disk>): Promise<Models.DisksGetResponse> {
-    return this.client.sendOperationRequest(
-      {
-        resourceGroupName,
-        diskName,
-        options
+  public list(
+    options?: DisksListOptionalParams
+  ): PagedAsyncIterableIterator<Disk> {
+    const iter = this.listPagingAll(options);
+    return {
+      next() {
+        return iter.next();
       },
-      getOperationSpec,
-      callback) as Promise<Models.DisksGetResponse>;
+      [Symbol.asyncIterator]() {
+        return this;
+      },
+      byPage: () => {
+        return this.listPagingPage(options);
+      }
+    };
+  }
+
+  private async *listPagingPage(
+    options?: DisksListOptionalParams
+  ): AsyncIterableIterator<Disk[]> {
+    let result = await this._list(options);
+    yield result.value || [];
+    let continuationToken = result.nextLink;
+    while (continuationToken) {
+      result = await this._listNext(continuationToken, options);
+      continuationToken = result.nextLink;
+      yield result.value || [];
+    }
+  }
+
+  private async *listPagingAll(
+    options?: DisksListOptionalParams
+  ): AsyncIterableIterator<Disk> {
+    for await (const page of this.listPagingPage(options)) {
+      yield* page;
+    }
   }
 
   /**
-   * Deletes a disk.
+   * Creates or updates a disk.
    * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param [options] The optional parameters
-   * @returns Promise<msRest.RestResponse>
+   * @param diskName The name of the managed disk that is being created. The name can't be changed after
+   *                 the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name
+   *                 length is 80 characters.
+   * @param disk Disk object supplied in the body of the Put disk operation.
+   * @param options The options parameters.
    */
-  deleteMethod(resourceGroupName: string, diskName: string, options?: msRest.RequestOptionsBase): Promise<msRest.RestResponse> {
-    return this.beginDeleteMethod(resourceGroupName,diskName,options)
-      .then(lroPoller => lroPoller.pollUntilFinished());
+  async beginCreateOrUpdate(
+    resourceGroupName: string,
+    diskName: string,
+    disk: Disk,
+    options?: DisksCreateOrUpdateOptionalParams
+  ): Promise<
+    PollerLike<
+      PollOperationState<DisksCreateOrUpdateResponse>,
+      DisksCreateOrUpdateResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<DisksCreateOrUpdateResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ) => {
+      let currentRawResponse:
+        | coreClient.FullOperationResponse
+        | undefined = undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback
+        }
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON()
+        }
+      };
+    };
+
+    const lro = new LroImpl(
+      sendOperation,
+      { resourceGroupName, diskName, disk, options },
+      createOrUpdateOperationSpec
+    );
+    return new LroEngine(lro, {
+      resumeFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs
+    });
   }
 
   /**
-   * Lists all the disks under a resource group.
+   * Creates or updates a disk.
    * @param resourceGroupName The name of the resource group.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.DisksListByResourceGroupResponse>
+   * @param diskName The name of the managed disk that is being created. The name can't be changed after
+   *                 the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name
+   *                 length is 80 characters.
+   * @param disk Disk object supplied in the body of the Put disk operation.
+   * @param options The options parameters.
    */
-  listByResourceGroup(resourceGroupName: string, options?: msRest.RequestOptionsBase): Promise<Models.DisksListByResourceGroupResponse>;
+  async beginCreateOrUpdateAndWait(
+    resourceGroupName: string,
+    diskName: string,
+    disk: Disk,
+    options?: DisksCreateOrUpdateOptionalParams
+  ): Promise<DisksCreateOrUpdateResponse> {
+    const poller = await this.beginCreateOrUpdate(
+      resourceGroupName,
+      diskName,
+      disk,
+      options
+    );
+    return poller.pollUntilDone();
+  }
+
   /**
+   * Updates (patches) a disk.
    * @param resourceGroupName The name of the resource group.
-   * @param callback The callback
+   * @param diskName The name of the managed disk that is being created. The name can't be changed after
+   *                 the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name
+   *                 length is 80 characters.
+   * @param disk Disk object supplied in the body of the Patch disk operation.
+   * @param options The options parameters.
    */
-  listByResourceGroup(resourceGroupName: string, callback: msRest.ServiceCallback<Models.DiskList>): void;
+  async beginUpdate(
+    resourceGroupName: string,
+    diskName: string,
+    disk: DiskUpdate,
+    options?: DisksUpdateOptionalParams
+  ): Promise<
+    PollerLike<PollOperationState<DisksUpdateResponse>, DisksUpdateResponse>
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<DisksUpdateResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ) => {
+      let currentRawResponse:
+        | coreClient.FullOperationResponse
+        | undefined = undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback
+        }
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON()
+        }
+      };
+    };
+
+    const lro = new LroImpl(
+      sendOperation,
+      { resourceGroupName, diskName, disk, options },
+      updateOperationSpec
+    );
+    return new LroEngine(lro, {
+      resumeFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs
+    });
+  }
+
   /**
+   * Updates (patches) a disk.
    * @param resourceGroupName The name of the resource group.
-   * @param options The optional parameters
-   * @param callback The callback
+   * @param diskName The name of the managed disk that is being created. The name can't be changed after
+   *                 the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name
+   *                 length is 80 characters.
+   * @param disk Disk object supplied in the body of the Patch disk operation.
+   * @param options The options parameters.
    */
-  listByResourceGroup(resourceGroupName: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.DiskList>): void;
-  listByResourceGroup(resourceGroupName: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.DiskList>, callback?: msRest.ServiceCallback<Models.DiskList>): Promise<Models.DisksListByResourceGroupResponse> {
-    return this.client.sendOperationRequest(
-      {
-        resourceGroupName,
-        options
-      },
-      listByResourceGroupOperationSpec,
-      callback) as Promise<Models.DisksListByResourceGroupResponse>;
+  async beginUpdateAndWait(
+    resourceGroupName: string,
+    diskName: string,
+    disk: DiskUpdate,
+    options?: DisksUpdateOptionalParams
+  ): Promise<DisksUpdateResponse> {
+    const poller = await this.beginUpdate(
+      resourceGroupName,
+      diskName,
+      disk,
+      options
+    );
+    return poller.pollUntilDone();
   }
 
   /**
-   * Lists all the disks under a subscription.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.DisksListResponse>
-   */
-  list(options?: msRest.RequestOptionsBase): Promise<Models.DisksListResponse>;
-  /**
-   * @param callback The callback
-   */
-  list(callback: msRest.ServiceCallback<Models.DiskList>): void;
-  /**
-   * @param options The optional parameters
-   * @param callback The callback
+   * Gets information about a disk.
+   * @param resourceGroupName The name of the resource group.
+   * @param diskName The name of the managed disk that is being created. The name can't be changed after
+   *                 the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name
+   *                 length is 80 characters.
+   * @param options The options parameters.
    */
-  list(options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.DiskList>): void;
-  list(options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.DiskList>, callback?: msRest.ServiceCallback<Models.DiskList>): Promise<Models.DisksListResponse> {
+  get(
+    resourceGroupName: string,
+    diskName: string,
+    options?: DisksGetOptionalParams
+  ): Promise<DisksGetResponse> {
     return this.client.sendOperationRequest(
-      {
-        options
-      },
-      listOperationSpec,
-      callback) as Promise<Models.DisksListResponse>;
+      { resourceGroupName, diskName, options },
+      getOperationSpec
+    );
   }
 
   /**
-   * Grants access to a disk.
+   * Deletes a disk.
    * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param grantAccessData Access data object supplied in the body of the get disk access operation.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.DisksGrantAccessResponse>
+   * @param diskName The name of the managed disk that is being created. The name can't be changed after
+   *                 the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name
+   *                 length is 80 characters.
+   * @param options The options parameters.
    */
-  grantAccess(resourceGroupName: string, diskName: string, grantAccessData: Models.GrantAccessData, options?: msRest.RequestOptionsBase): Promise<Models.DisksGrantAccessResponse> {
-    return this.beginGrantAccess(resourceGroupName,diskName,grantAccessData,options)
-      .then(lroPoller => lroPoller.pollUntilFinished()) as Promise<Models.DisksGrantAccessResponse>;
+  async beginDelete(
+    resourceGroupName: string,
+    diskName: string,
+    options?: DisksDeleteOptionalParams
+  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<void> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ) => {
+      let currentRawResponse:
+        | coreClient.FullOperationResponse
+        | undefined = undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback
+        }
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON()
+        }
+      };
+    };
+
+    const lro = new LroImpl(
+      sendOperation,
+      { resourceGroupName, diskName, options },
+      deleteOperationSpec
+    );
+    return new LroEngine(lro, {
+      resumeFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs
+    });
   }
 
   /**
-   * Revokes access to a disk.
+   * Deletes a disk.
    * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param [options] The optional parameters
-   * @returns Promise<msRest.RestResponse>
+   * @param diskName The name of the managed disk that is being created. The name can't be changed after
+   *                 the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name
+   *                 length is 80 characters.
+   * @param options The options parameters.
    */
-  revokeAccess(resourceGroupName: string, diskName: string, options?: msRest.RequestOptionsBase): Promise<msRest.RestResponse> {
-    return this.beginRevokeAccess(resourceGroupName,diskName,options)
-      .then(lroPoller => lroPoller.pollUntilFinished());
+  async beginDeleteAndWait(
+    resourceGroupName: string,
+    diskName: string,
+    options?: DisksDeleteOptionalParams
+  ): Promise<void> {
+    const poller = await this.beginDelete(resourceGroupName, diskName, options);
+    return poller.pollUntilDone();
   }
 
   /**
-   * Creates or updates a disk.
+   * Lists all the disks under a resource group.
    * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param disk Disk object supplied in the body of the Put disk operation.
-   * @param [options] The optional parameters
-   * @returns Promise<msRestAzure.LROPoller>
+   * @param options The options parameters.
    */
-  beginCreateOrUpdate(resourceGroupName: string, diskName: string, disk: Models.Disk, options?: msRest.RequestOptionsBase): Promise<msRestAzure.LROPoller> {
-    return this.client.sendLRORequest(
-      {
-        resourceGroupName,
-        diskName,
-        disk,
-        options
-      },
-      beginCreateOrUpdateOperationSpec,
-      options);
+  private _listByResourceGroup(
+    resourceGroupName: string,
+    options?: DisksListByResourceGroupOptionalParams
+  ): Promise<DisksListByResourceGroupResponse> {
+    return this.client.sendOperationRequest(
+      { resourceGroupName, options },
+      listByResourceGroupOperationSpec
+    );
   }
 
   /**
-   * Updates (patches) a disk.
-   * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param disk Disk object supplied in the body of the Patch disk operation.
-   * @param [options] The optional parameters
-   * @returns Promise<msRestAzure.LROPoller>
+   * Lists all the disks under a subscription.
+   * @param options The options parameters.
    */
-  beginUpdate(resourceGroupName: string, diskName: string, disk: Models.DiskUpdate, options?: msRest.RequestOptionsBase): Promise<msRestAzure.LROPoller> {
-    return this.client.sendLRORequest(
-      {
-        resourceGroupName,
-        diskName,
-        disk,
-        options
-      },
-      beginUpdateOperationSpec,
-      options);
+  private _list(options?: DisksListOptionalParams): Promise<DisksListResponse> {
+    return this.client.sendOperationRequest({ options }, listOperationSpec);
   }
 
   /**
-   * Deletes a disk.
+   * Grants access to a disk.
    * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param [options] The optional parameters
-   * @returns Promise<msRestAzure.LROPoller>
+   * @param diskName The name of the managed disk that is being created. The name can't be changed after
+   *                 the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name
+   *                 length is 80 characters.
+   * @param grantAccessData Access data object supplied in the body of the get disk access operation.
+   * @param options The options parameters.
    */
-  beginDeleteMethod(resourceGroupName: string, diskName: string, options?: msRest.RequestOptionsBase): Promise<msRestAzure.LROPoller> {
-    return this.client.sendLRORequest(
-      {
-        resourceGroupName,
-        diskName,
-        options
-      },
-      beginDeleteMethodOperationSpec,
-      options);
+  async beginGrantAccess(
+    resourceGroupName: string,
+    diskName: string,
+    grantAccessData: GrantAccessData,
+    options?: DisksGrantAccessOptionalParams
+  ): Promise<
+    PollerLike<
+      PollOperationState<DisksGrantAccessResponse>,
+      DisksGrantAccessResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<DisksGrantAccessResponse> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ) => {
+      let currentRawResponse:
+        | coreClient.FullOperationResponse
+        | undefined = undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback
+        }
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON()
+        }
+      };
+    };
+
+    const lro = new LroImpl(
+      sendOperation,
+      { resourceGroupName, diskName, grantAccessData, options },
+      grantAccessOperationSpec
+    );
+    return new LroEngine(lro, {
+      resumeFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      lroResourceLocationConfig: "location"
+    });
   }
 
   /**
    * Grants access to a disk.
    * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
+   * @param diskName The name of the managed disk that is being created. The name can't be changed after
+   *                 the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name
+   *                 length is 80 characters.
    * @param grantAccessData Access data object supplied in the body of the get disk access operation.
-   * @param [options] The optional parameters
-   * @returns Promise<msRestAzure.LROPoller>
+   * @param options The options parameters.
    */
-  beginGrantAccess(resourceGroupName: string, diskName: string, grantAccessData: Models.GrantAccessData, options?: msRest.RequestOptionsBase): Promise<msRestAzure.LROPoller> {
-    return this.client.sendLRORequest(
-      {
-        resourceGroupName,
-        diskName,
-        grantAccessData,
-        options
-      },
-      beginGrantAccessOperationSpec,
-      options);
+  async beginGrantAccessAndWait(
+    resourceGroupName: string,
+    diskName: string,
+    grantAccessData: GrantAccessData,
+    options?: DisksGrantAccessOptionalParams
+  ): Promise<DisksGrantAccessResponse> {
+    const poller = await this.beginGrantAccess(
+      resourceGroupName,
+      diskName,
+      grantAccessData,
+      options
+    );
+    return poller.pollUntilDone();
   }
 
   /**
    * Revokes access to a disk.
    * @param resourceGroupName The name of the resource group.
-   * @param diskName The name of the managed disk that is being created. The name can't be changed
-   * after the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The
-   * maximum name length is 80 characters.
-   * @param [options] The optional parameters
-   * @returns Promise<msRestAzure.LROPoller>
+   * @param diskName The name of the managed disk that is being created. The name can't be changed after
+   *                 the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name
+   *                 length is 80 characters.
+   * @param options The options parameters.
    */
-  beginRevokeAccess(resourceGroupName: string, diskName: string, options?: msRest.RequestOptionsBase): Promise<msRestAzure.LROPoller> {
-    return this.client.sendLRORequest(
-      {
-        resourceGroupName,
-        diskName,
-        options
-      },
-      beginRevokeAccessOperationSpec,
-      options);
+  async beginRevokeAccess(
+    resourceGroupName: string,
+    diskName: string,
+    options?: DisksRevokeAccessOptionalParams
+  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<void> => {
+      return this.client.sendOperationRequest(args, spec);
+    };
+    const sendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ) => {
+      let currentRawResponse:
+        | coreClient.FullOperationResponse
+        | undefined = undefined;
+      const providedCallback = args.options?.onResponse;
+      const callback: coreClient.RawResponseCallback = (
+        rawResponse: coreClient.FullOperationResponse,
+        flatResponse: unknown
+      ) => {
+        currentRawResponse = rawResponse;
+        providedCallback?.(rawResponse, flatResponse);
+      };
+      const updatedArgs = {
+        ...args,
+        options: {
+          ...args.options,
+          onResponse: callback
+        }
+      };
+      const flatResponse = await directSendOperation(updatedArgs, spec);
+      return {
+        flatResponse,
+        rawResponse: {
+          statusCode: currentRawResponse!.status,
+          body: currentRawResponse!.parsedBody,
+          headers: currentRawResponse!.headers.toJSON()
+        }
+      };
+    };
+
+    const lro = new LroImpl(
+      sendOperation,
+      { resourceGroupName, diskName, options },
+      revokeAccessOperationSpec
+    );
+    return new LroEngine(lro, {
+      resumeFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs,
+      lroResourceLocationConfig: "location"
+    });
   }
 
   /**
-   * Lists all the disks under a resource group.
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.DisksListByResourceGroupNextResponse>
-   */
-  listByResourceGroupNext(nextPageLink: string, options?: msRest.RequestOptionsBase): Promise<Models.DisksListByResourceGroupNextResponse>;
-  /**
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param callback The callback
+   * Revokes access to a disk.
+   * @param resourceGroupName The name of the resource group.
+   * @param diskName The name of the managed disk that is being created. The name can't be changed after
+   *                 the disk is created. Supported characters for the name are a-z, A-Z, 0-9, _ and -. The maximum name
+   *                 length is 80 characters.
+   * @param options The options parameters.
    */
-  listByResourceGroupNext(nextPageLink: string, callback: msRest.ServiceCallback<Models.DiskList>): void;
+  async beginRevokeAccessAndWait(
+    resourceGroupName: string,
+    diskName: string,
+    options?: DisksRevokeAccessOptionalParams
+  ): Promise<void> {
+    const poller = await this.beginRevokeAccess(
+      resourceGroupName,
+      diskName,
+      options
+    );
+    return poller.pollUntilDone();
+  }
+
   /**
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param options The optional parameters
-   * @param callback The callback
+   * ListByResourceGroupNext
+   * @param resourceGroupName The name of the resource group.
+   * @param nextLink The nextLink from the previous successful call to the ListByResourceGroup method.
+   * @param options The options parameters.
    */
-  listByResourceGroupNext(nextPageLink: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.DiskList>): void;
-  listByResourceGroupNext(nextPageLink: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.DiskList>, callback?: msRest.ServiceCallback<Models.DiskList>): Promise<Models.DisksListByResourceGroupNextResponse> {
+  private _listByResourceGroupNext(
+    resourceGroupName: string,
+    nextLink: string,
+    options?: DisksListByResourceGroupNextOptionalParams
+  ): Promise<DisksListByResourceGroupNextResponse> {
     return this.client.sendOperationRequest(
-      {
-        nextPageLink,
-        options
-      },
-      listByResourceGroupNextOperationSpec,
-      callback) as Promise<Models.DisksListByResourceGroupNextResponse>;
+      { resourceGroupName, nextLink, options },
+      listByResourceGroupNextOperationSpec
+    );
   }
 
   /**
-   * Lists all the disks under a subscription.
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.DisksListNextResponse>
+   * ListNext
+   * @param nextLink The nextLink from the previous successful call to the List method.
+   * @param options The options parameters.
    */
-  listNext(nextPageLink: string, options?: msRest.RequestOptionsBase): Promise<Models.DisksListNextResponse>;
-  /**
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param callback The callback
-   */
-  listNext(nextPageLink: string, callback: msRest.ServiceCallback<Models.DiskList>): void;
-  /**
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param options The optional parameters
-   * @param callback The callback
-   */
-  listNext(nextPageLink: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.DiskList>): void;
-  listNext(nextPageLink: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.DiskList>, callback?: msRest.ServiceCallback<Models.DiskList>): Promise<Models.DisksListNextResponse> {
+  private _listNext(
+    nextLink: string,
+    options?: DisksListNextOptionalParams
+  ): Promise<DisksListNextResponse> {
     return this.client.sendOperationRequest(
-      {
-        nextPageLink,
-        options
-      },
-      listNextOperationSpec,
-      callback) as Promise<Models.DisksListNextResponse>;
+      { nextLink, options },
+      listNextOperationSpec
+    );
   }
 }
-
 // Operation Specifications
-const serializer = new msRest.Serializer(Mappers);
-const getOperationSpec: msRest.OperationSpec = {
-  httpMethod: "GET",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}",
-  urlParameters: [
-    Parameters.subscriptionId,
-    Parameters.resourceGroupName,
-    Parameters.diskName
-  ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
+const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
+
+const createOrUpdateOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}",
+  httpMethod: "PUT",
   responses: {
     200: {
       bodyMapper: Mappers.Disk
     },
-    default: {
-      bodyMapper: Mappers.CloudError
-    }
-  },
-  serializer
-};
-
-const listByResourceGroupOperationSpec: msRest.OperationSpec = {
-  httpMethod: "GET",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks",
-  urlParameters: [
-    Parameters.subscriptionId,
-    Parameters.resourceGroupName
-  ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
-  responses: {
-    200: {
-      bodyMapper: Mappers.DiskList
+    201: {
+      bodyMapper: Mappers.Disk
     },
-    default: {
-      bodyMapper: Mappers.CloudError
-    }
-  },
-  serializer
-};
-
-const listOperationSpec: msRest.OperationSpec = {
-  httpMethod: "GET",
-  path: "subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks",
-  urlParameters: [
-    Parameters.subscriptionId
-  ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
-  responses: {
-    200: {
-      bodyMapper: Mappers.DiskList
+    202: {
+      bodyMapper: Mappers.Disk
     },
-    default: {
-      bodyMapper: Mappers.CloudError
+    204: {
+      bodyMapper: Mappers.Disk
     }
   },
-  serializer
-};
-
-const beginCreateOrUpdateOperationSpec: msRest.OperationSpec = {
-  httpMethod: "PUT",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}",
+  requestBody: Parameters.disk,
+  queryParameters: [Parameters.apiVersion1],
   urlParameters: [
-    Parameters.subscriptionId,
+    Parameters.$host,
     Parameters.resourceGroupName,
+    Parameters.subscriptionId,
     Parameters.diskName
   ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
-  requestBody: {
-    parameterPath: "disk",
-    mapper: {
-      ...Mappers.Disk,
-      required: true
-    }
-  },
+  headerParameters: [Parameters.accept, Parameters.contentType],
+  mediaType: "json",
+  serializer
+};
+const updateOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}",
+  httpMethod: "PATCH",
   responses: {
     200: {
       bodyMapper: Mappers.Disk
     },
+    201: {
+      bodyMapper: Mappers.Disk
+    },
     202: {
       bodyMapper: Mappers.Disk
     },
-    default: {
-      bodyMapper: Mappers.CloudError
+    204: {
+      bodyMapper: Mappers.Disk
     }
   },
-  serializer
-};
-
-const beginUpdateOperationSpec: msRest.OperationSpec = {
-  httpMethod: "PATCH",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}",
+  requestBody: Parameters.disk1,
+  queryParameters: [Parameters.apiVersion1],
   urlParameters: [
-    Parameters.subscriptionId,
+    Parameters.$host,
     Parameters.resourceGroupName,
+    Parameters.subscriptionId,
     Parameters.diskName
   ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
-  requestBody: {
-    parameterPath: "disk",
-    mapper: {
-      ...Mappers.DiskUpdate,
-      required: true
-    }
-  },
+  headerParameters: [Parameters.accept, Parameters.contentType],
+  mediaType: "json",
+  serializer
+};
+const getOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}",
+  httpMethod: "GET",
   responses: {
     200: {
       bodyMapper: Mappers.Disk
-    },
-    202: {
-      bodyMapper: Mappers.Disk
-    },
-    default: {
-      bodyMapper: Mappers.CloudError
     }
   },
+  queryParameters: [Parameters.apiVersion1],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.resourceGroupName,
+    Parameters.subscriptionId,
+    Parameters.diskName
+  ],
+  headerParameters: [Parameters.accept],
   serializer
 };
-
-const beginDeleteMethodOperationSpec: msRest.OperationSpec = {
+const deleteOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}",
   httpMethod: "DELETE",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}",
+  responses: { 200: {}, 201: {}, 202: {}, 204: {} },
+  queryParameters: [Parameters.apiVersion1],
   urlParameters: [
-    Parameters.subscriptionId,
+    Parameters.$host,
     Parameters.resourceGroupName,
+    Parameters.subscriptionId,
     Parameters.diskName
   ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
+  serializer
+};
+const listByResourceGroupOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks",
+  httpMethod: "GET",
   responses: {
-    200: {},
-    202: {},
-    204: {},
-    default: {
-      bodyMapper: Mappers.CloudError
+    200: {
+      bodyMapper: Mappers.DiskList
     }
   },
-  serializer
-};
-
-const beginGrantAccessOperationSpec: msRest.OperationSpec = {
-  httpMethod: "POST",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess",
+  queryParameters: [Parameters.apiVersion1],
   urlParameters: [
-    Parameters.subscriptionId,
+    Parameters.$host,
     Parameters.resourceGroupName,
-    Parameters.diskName
-  ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
+    Parameters.subscriptionId
   ],
-  requestBody: {
-    parameterPath: "grantAccessData",
-    mapper: {
-      ...Mappers.GrantAccessData,
-      required: true
+  headerParameters: [Parameters.accept],
+  serializer
+};
+const listOperationSpec: coreClient.OperationSpec = {
+  path: "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/disks",
+  httpMethod: "GET",
+  responses: {
+    200: {
+      bodyMapper: Mappers.DiskList
     }
   },
+  queryParameters: [Parameters.apiVersion1],
+  urlParameters: [Parameters.$host, Parameters.subscriptionId],
+  headerParameters: [Parameters.accept],
+  serializer
+};
+const grantAccessOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/beginGetAccess",
+  httpMethod: "POST",
   responses: {
     200: {
       bodyMapper: Mappers.AccessUri
     },
-    202: {},
-    default: {
-      bodyMapper: Mappers.CloudError
+    201: {
+      bodyMapper: Mappers.AccessUri
+    },
+    202: {
+      bodyMapper: Mappers.AccessUri
+    },
+    204: {
+      bodyMapper: Mappers.AccessUri
     }
   },
+  requestBody: Parameters.grantAccessData,
+  queryParameters: [Parameters.apiVersion1],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.resourceGroupName,
+    Parameters.subscriptionId,
+    Parameters.diskName
+  ],
+  headerParameters: [Parameters.accept, Parameters.contentType],
+  mediaType: "json",
   serializer
 };
-
-const beginRevokeAccessOperationSpec: msRest.OperationSpec = {
+const revokeAccessOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess",
   httpMethod: "POST",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/disks/{diskName}/endGetAccess",
+  responses: { 200: {}, 201: {}, 202: {}, 204: {} },
+  queryParameters: [Parameters.apiVersion1],
   urlParameters: [
-    Parameters.subscriptionId,
+    Parameters.$host,
     Parameters.resourceGroupName,
+    Parameters.subscriptionId,
     Parameters.diskName
   ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
-  responses: {
-    200: {},
-    202: {},
-    default: {
-      bodyMapper: Mappers.CloudError
-    }
-  },
   serializer
 };
-
-const listByResourceGroupNextOperationSpec: msRest.OperationSpec = {
-  httpMethod: "GET",
-  baseUrl: "https://management.azure.com",
+const listByResourceGroupNextOperationSpec: coreClient.OperationSpec = {
   path: "{nextLink}",
-  urlParameters: [
-    Parameters.nextPageLink
-  ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
+  httpMethod: "GET",
   responses: {
     200: {
       bodyMapper: Mappers.DiskList
-    },
-    default: {
-      bodyMapper: Mappers.CloudError
     }
   },
+  queryParameters: [Parameters.apiVersion1],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.resourceGroupName,
+    Parameters.subscriptionId,
+    Parameters.nextLink
+  ],
+  headerParameters: [Parameters.accept],
   serializer
 };
-
-const listNextOperationSpec: msRest.OperationSpec = {
-  httpMethod: "GET",
-  baseUrl: "https://management.azure.com",
+const listNextOperationSpec: coreClient.OperationSpec = {
   path: "{nextLink}",
-  urlParameters: [
-    Parameters.nextPageLink
-  ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
+  httpMethod: "GET",
   responses: {
     200: {
       bodyMapper: Mappers.DiskList
-    },
-    default: {
-      bodyMapper: Mappers.CloudError
     }
   },
+  queryParameters: [Parameters.apiVersion1],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.nextLink
+  ],
+  headerParameters: [Parameters.accept],
   serializer
 };