@azure/arm-elasticsan 1.0.0-beta.1 → 1.0.0-beta.3

package/src/operations/volumeGroups.ts

@@ -6,14 +6,19 @@
  * Changes may cause incorrect behavior and will be lost if the code is regenerated.
  */
 
-import { PagedAsyncIterableIterator } from "@azure/core-paging";
+import { PagedAsyncIterableIterator, PageSettings } from "@azure/core-paging";
+import { setContinuationToken } from "../pagingHelper";
 import { VolumeGroups } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
 import * as Parameters from "../models/parameters";
 import { ElasticSanManagement } from "../elasticSanManagement";
-import { PollerLike, PollOperationState, LroEngine } from "@azure/core-lro";
-import { LroImpl } from "../lroImpl";
+import {
+  SimplePollerLike,
+  OperationState,
+  createHttpPoller
+} from "@azure/core-lro";
+import { createLroSpec } from "../lroImpl";
 import {
   VolumeGroup,
   VolumeGroupsListByElasticSanNextOptionalParams,
@@ -66,11 +71,15 @@ export class VolumeGroupsImpl implements VolumeGroups {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
         return this.listByElasticSanPagingPage(
           resourceGroupName,
           elasticSanName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -79,15 +88,22 @@ export class VolumeGroupsImpl implements VolumeGroups {
   private async *listByElasticSanPagingPage(
     resourceGroupName: string,
     elasticSanName: string,
-    options?: VolumeGroupsListByElasticSanOptionalParams
+    options?: VolumeGroupsListByElasticSanOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<VolumeGroup[]> {
-    let result = await this._listByElasticSan(
-      resourceGroupName,
-      elasticSanName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: VolumeGroupsListByElasticSanResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listByElasticSan(
+        resourceGroupName,
+        elasticSanName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listByElasticSanNext(
         resourceGroupName,
@@ -96,7 +112,9 @@ export class VolumeGroupsImpl implements VolumeGroups {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -146,8 +164,8 @@ export class VolumeGroupsImpl implements VolumeGroups {
     parameters: VolumeGroup,
     options?: VolumeGroupsCreateOptionalParams
   ): Promise<
-    PollerLike<
-      PollOperationState<VolumeGroupsCreateResponse>,
+    SimplePollerLike<
+      OperationState<VolumeGroupsCreateResponse>,
       VolumeGroupsCreateResponse
     >
   > {
@@ -157,7 +175,7 @@ export class VolumeGroupsImpl implements VolumeGroups {
     ): Promise<VolumeGroupsCreateResponse> => {
       return this.client.sendOperationRequest(args, spec);
     };
-    const sendOperation = async (
+    const sendOperationFn = async (
       args: coreClient.OperationArguments,
       spec: coreClient.OperationSpec
     ) => {
@@ -190,21 +208,24 @@ export class VolumeGroupsImpl implements VolumeGroups {
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      {
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
         resourceGroupName,
         elasticSanName,
         volumeGroupName,
         parameters,
         options
       },
-      createOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
+      spec: createOperationSpec
+    });
+    const poller = await createHttpPoller<
+      VolumeGroupsCreateResponse,
+      OperationState<VolumeGroupsCreateResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
       intervalInMs: options?.updateIntervalInMs,
-      lroResourceLocationConfig: "azure-async-operation"
+      resourceLocationConfig: "location"
     });
     await poller.poll();
     return poller;
@@ -250,8 +271,8 @@ export class VolumeGroupsImpl implements VolumeGroups {
     parameters: VolumeGroupUpdate,
     options?: VolumeGroupsUpdateOptionalParams
   ): Promise<
-    PollerLike<
-      PollOperationState<VolumeGroupsUpdateResponse>,
+    SimplePollerLike<
+      OperationState<VolumeGroupsUpdateResponse>,
       VolumeGroupsUpdateResponse
     >
   > {
@@ -261,7 +282,7 @@ export class VolumeGroupsImpl implements VolumeGroups {
     ): Promise<VolumeGroupsUpdateResponse> => {
       return this.client.sendOperationRequest(args, spec);
     };
-    const sendOperation = async (
+    const sendOperationFn = async (
       args: coreClient.OperationArguments,
       spec: coreClient.OperationSpec
     ) => {
@@ -294,21 +315,24 @@ export class VolumeGroupsImpl implements VolumeGroups {
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      {
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: {
         resourceGroupName,
         elasticSanName,
         volumeGroupName,
         parameters,
         options
       },
-      updateOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
+      spec: updateOperationSpec
+    });
+    const poller = await createHttpPoller<
+      VolumeGroupsUpdateResponse,
+      OperationState<VolumeGroupsUpdateResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
       intervalInMs: options?.updateIntervalInMs,
-      lroResourceLocationConfig: "azure-async-operation"
+      resourceLocationConfig: "location"
     });
     await poller.poll();
     return poller;
@@ -351,14 +375,14 @@ export class VolumeGroupsImpl implements VolumeGroups {
     elasticSanName: string,
     volumeGroupName: string,
     options?: VolumeGroupsDeleteOptionalParams
-  ): Promise<PollerLike<PollOperationState<void>, void>> {
+  ): Promise<SimplePollerLike<OperationState<void>, void>> {
     const directSendOperation = async (
       args: coreClient.OperationArguments,
       spec: coreClient.OperationSpec
     ): Promise<void> => {
       return this.client.sendOperationRequest(args, spec);
     };
-    const sendOperation = async (
+    const sendOperationFn = async (
       args: coreClient.OperationArguments,
       spec: coreClient.OperationSpec
     ) => {
@@ -391,15 +415,15 @@ export class VolumeGroupsImpl implements VolumeGroups {
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      { resourceGroupName, elasticSanName, volumeGroupName, options },
-      deleteOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: { resourceGroupName, elasticSanName, volumeGroupName, options },
+      spec: deleteOperationSpec
+    });
+    const poller = await createHttpPoller<void, OperationState<void>>(lro, {
+      restoreFrom: options?.resumeFrom,
       intervalInMs: options?.updateIntervalInMs,
-      lroResourceLocationConfig: "azure-async-operation"
+      resourceLocationConfig: "location"
     });
     await poller.poll();
     return poller;
@@ -477,7 +501,7 @@ const listByElasticSanOperationSpec: coreClient.OperationSpec = {
       bodyMapper: Mappers.VolumeGroupList
     },
     default: {
-      bodyMapper: Mappers.ErrorModel
+      bodyMapper: Mappers.ErrorResponse
     }
   },
   queryParameters: [Parameters.apiVersion],
@@ -508,7 +532,7 @@ const createOperationSpec: coreClient.OperationSpec = {
       bodyMapper: Mappers.VolumeGroup
     },
     default: {
-      bodyMapper: Mappers.ErrorModel
+      bodyMapper: Mappers.ErrorResponse
     }
   },
   requestBody: Parameters.parameters2,
@@ -542,7 +566,7 @@ const updateOperationSpec: coreClient.OperationSpec = {
       bodyMapper: Mappers.VolumeGroup
     },
     default: {
-      bodyMapper: Mappers.ErrorModel
+      bodyMapper: Mappers.ErrorResponse
     }
   },
   requestBody: Parameters.parameters3,
@@ -568,7 +592,7 @@ const deleteOperationSpec: coreClient.OperationSpec = {
     202: {},
     204: {},
     default: {
-      bodyMapper: Mappers.ErrorModel
+      bodyMapper: Mappers.ErrorResponse
     }
   },
   queryParameters: [Parameters.apiVersion],
@@ -591,7 +615,7 @@ const getOperationSpec: coreClient.OperationSpec = {
       bodyMapper: Mappers.VolumeGroup
     },
     default: {
-      bodyMapper: Mappers.ErrorModel
+      bodyMapper: Mappers.ErrorResponse
     }
   },
   queryParameters: [Parameters.apiVersion],
@@ -613,10 +637,9 @@ const listByElasticSanNextOperationSpec: coreClient.OperationSpec = {
       bodyMapper: Mappers.VolumeGroupList
     },
     default: {
-      bodyMapper: Mappers.ErrorModel
+      bodyMapper: Mappers.ErrorResponse
     }
   },
-  queryParameters: [Parameters.apiVersion],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,