@azure/arm-datamigration 3.0.0-beta.2 → 3.0.0-beta.3

package/src/operations/files.ts

@@ -6,7 +6,8 @@
  * 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 { Files } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -71,12 +72,16 @@ export class FilesImpl implements Files {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
         return this.listPagingPage(
           groupName,
           serviceName,
           projectName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -86,11 +91,18 @@ export class FilesImpl implements Files {
     groupName: string,
     serviceName: string,
     projectName: string,
-    options?: FilesListOptionalParams
+    options?: FilesListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<ProjectFile[]> {
-    let result = await this._list(groupName, serviceName, projectName, options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: FilesListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(groupName, serviceName, projectName, options);
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(
         groupName,
@@ -100,7 +112,9 @@ export class FilesImpl implements Files {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 

package/src/operations/operations.ts

@@ -6,7 +6,8 @@
  * 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 { Operations } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -48,22 +49,34 @@ export class OperationsImpl implements Operations {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
-        return this.listPagingPage(options);
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listPagingPage(options, settings);
       }
     };
   }
 
   private async *listPagingPage(
-    options?: OperationsListOptionalParams
+    options?: OperationsListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<OperationsDefinition[]> {
-    let result = await this._list(options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: OperationsListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(options);
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(continuationToken, options);
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 

package/src/operations/projects.ts

@@ -6,7 +6,8 @@
  * 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 { Projects } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -60,8 +61,11 @@ export class ProjectsImpl implements Projects {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
-        return this.listPagingPage(groupName, serviceName, options);
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listPagingPage(groupName, serviceName, options, settings);
       }
     };
   }
@@ -69,11 +73,18 @@ export class ProjectsImpl implements Projects {
   private async *listPagingPage(
     groupName: string,
     serviceName: string,
-    options?: ProjectsListOptionalParams
+    options?: ProjectsListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<Project[]> {
-    let result = await this._list(groupName, serviceName, options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: ProjectsListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(groupName, serviceName, options);
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(
         groupName,
@@ -82,7 +93,9 @@ export class ProjectsImpl implements Projects {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 

package/src/operations/resourceSkus.ts

@@ -6,7 +6,8 @@
  * 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 { ResourceSkus } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -48,22 +49,34 @@ export class ResourceSkusImpl implements ResourceSkus {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
-        return this.listSkusPagingPage(options);
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listSkusPagingPage(options, settings);
       }
     };
   }
 
   private async *listSkusPagingPage(
-    options?: ResourceSkusListSkusOptionalParams
+    options?: ResourceSkusListSkusOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<ResourceSku[]> {
-    let result = await this._listSkus(options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: ResourceSkusListSkusResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listSkus(options);
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listSkusNext(continuationToken, options);
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 

package/src/operations/serviceTasks.ts

@@ -6,7 +6,8 @@
  * 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 { ServiceTasks } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -63,8 +64,11 @@ export class ServiceTasksImpl implements ServiceTasks {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
-        return this.listPagingPage(groupName, serviceName, options);
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listPagingPage(groupName, serviceName, options, settings);
       }
     };
   }
@@ -72,11 +76,18 @@ export class ServiceTasksImpl implements ServiceTasks {
   private async *listPagingPage(
     groupName: string,
     serviceName: string,
-    options?: ServiceTasksListOptionalParams
+    options?: ServiceTasksListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<ProjectTask[]> {
-    let result = await this._list(groupName, serviceName, options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: ServiceTasksListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(groupName, serviceName, options);
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(
         groupName,
@@ -85,7 +96,9 @@ export class ServiceTasksImpl implements ServiceTasks {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 

package/src/operations/services.ts

@@ -6,7 +6,8 @@
  * 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 { Services } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -18,11 +19,14 @@ import {
   AvailableServiceSku,
   ServicesListSkusNextOptionalParams,
   ServicesListSkusOptionalParams,
+  ServicesListSkusResponse,
   DataMigrationService,
   ServicesListByResourceGroupNextOptionalParams,
   ServicesListByResourceGroupOptionalParams,
+  ServicesListByResourceGroupResponse,
   ServicesListNextOptionalParams,
   ServicesListOptionalParams,
+  ServicesListResponse,
   ServicesCreateOrUpdateOptionalParams,
   ServicesCreateOrUpdateResponse,
   ServicesGetOptionalParams,
@@ -34,12 +38,9 @@ import {
   ServicesCheckStatusResponse,
   ServicesStartOptionalParams,
   ServicesStopOptionalParams,
-  ServicesListSkusResponse,
   NameAvailabilityRequest,
   ServicesCheckChildrenNameAvailabilityOptionalParams,
   ServicesCheckChildrenNameAvailabilityResponse,
-  ServicesListByResourceGroupResponse,
-  ServicesListResponse,
   ServicesCheckNameAvailabilityOptionalParams,
   ServicesCheckNameAvailabilityResponse,
   ServicesListSkusNextResponse,
@@ -80,8 +81,16 @@ export class ServicesImpl implements Services {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
-        return this.listSkusPagingPage(groupName, serviceName, options);
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listSkusPagingPage(
+          groupName,
+          serviceName,
+          options,
+          settings
+        );
       }
     };
   }
@@ -89,11 +98,18 @@ export class ServicesImpl implements Services {
   private async *listSkusPagingPage(
     groupName: string,
     serviceName: string,
-    options?: ServicesListSkusOptionalParams
+    options?: ServicesListSkusOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<AvailableServiceSku[]> {
-    let result = await this._listSkus(groupName, serviceName, options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: ServicesListSkusResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listSkus(groupName, serviceName, options);
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listSkusNext(
         groupName,
@@ -102,7 +118,9 @@ export class ServicesImpl implements Services {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -138,19 +156,29 @@ export class ServicesImpl implements Services {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
-        return this.listByResourceGroupPagingPage(groupName, options);
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listByResourceGroupPagingPage(groupName, options, settings);
       }
     };
   }
 
   private async *listByResourceGroupPagingPage(
     groupName: string,
-    options?: ServicesListByResourceGroupOptionalParams
+    options?: ServicesListByResourceGroupOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<DataMigrationService[]> {
-    let result = await this._listByResourceGroup(groupName, options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: ServicesListByResourceGroupResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listByResourceGroup(groupName, options);
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listByResourceGroupNext(
         groupName,
@@ -158,7 +186,9 @@ export class ServicesImpl implements Services {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -190,22 +220,34 @@ export class ServicesImpl implements Services {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
-        return this.listPagingPage(options);
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listPagingPage(options, settings);
       }
     };
   }
 
   private async *listPagingPage(
-    options?: ServicesListOptionalParams
+    options?: ServicesListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<DataMigrationService[]> {
-    let result = await this._list(options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: ServicesListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(options);
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(continuationToken, options);
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 

package/src/operations/sqlMigrationServices.ts

@@ -6,7 +6,8 @@
  * 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 { SqlMigrationServices } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -18,11 +19,14 @@ import {
   SqlMigrationService,
   SqlMigrationServicesListByResourceGroupNextOptionalParams,
   SqlMigrationServicesListByResourceGroupOptionalParams,
+  SqlMigrationServicesListByResourceGroupResponse,
   DatabaseMigration,
   SqlMigrationServicesListMigrationsNextOptionalParams,
   SqlMigrationServicesListMigrationsOptionalParams,
+  SqlMigrationServicesListMigrationsResponse,
   SqlMigrationServicesListBySubscriptionNextOptionalParams,
   SqlMigrationServicesListBySubscriptionOptionalParams,
+  SqlMigrationServicesListBySubscriptionResponse,
   SqlMigrationServicesGetOptionalParams,
   SqlMigrationServicesGetResponse,
   SqlMigrationServicesCreateOrUpdateOptionalParams,
@@ -31,7 +35,6 @@ import {
   SqlMigrationServiceUpdate,
   SqlMigrationServicesUpdateOptionalParams,
   SqlMigrationServicesUpdateResponse,
-  SqlMigrationServicesListByResourceGroupResponse,
   SqlMigrationServicesListAuthKeysOptionalParams,
   SqlMigrationServicesListAuthKeysResponse,
   RegenAuthKeys,
@@ -40,10 +43,8 @@ import {
   DeleteNode,
   SqlMigrationServicesDeleteNodeOptionalParams,
   SqlMigrationServicesDeleteNodeResponse,
-  SqlMigrationServicesListMigrationsResponse,
   SqlMigrationServicesListMonitoringDataOptionalParams,
   SqlMigrationServicesListMonitoringDataResponse,
-  SqlMigrationServicesListBySubscriptionResponse,
   SqlMigrationServicesListByResourceGroupNextResponse,
   SqlMigrationServicesListMigrationsNextResponse,
   SqlMigrationServicesListBySubscriptionNextResponse
@@ -80,19 +81,33 @@ export class SqlMigrationServicesImpl implements SqlMigrationServices {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
-        return this.listByResourceGroupPagingPage(resourceGroupName, options);
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listByResourceGroupPagingPage(
+          resourceGroupName,
+          options,
+          settings
+        );
       }
     };
   }
 
   private async *listByResourceGroupPagingPage(
     resourceGroupName: string,
-    options?: SqlMigrationServicesListByResourceGroupOptionalParams
+    options?: SqlMigrationServicesListByResourceGroupOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<SqlMigrationService[]> {
-    let result = await this._listByResourceGroup(resourceGroupName, options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: SqlMigrationServicesListByResourceGroupResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listByResourceGroup(resourceGroupName, options);
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listByResourceGroupNext(
         resourceGroupName,
@@ -100,7 +115,9 @@ export class SqlMigrationServicesImpl implements SqlMigrationServices {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -140,11 +157,15 @@ export class SqlMigrationServicesImpl implements SqlMigrationServices {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
         return this.listMigrationsPagingPage(
           resourceGroupName,
           sqlMigrationServiceName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -153,15 +174,22 @@ export class SqlMigrationServicesImpl implements SqlMigrationServices {
   private async *listMigrationsPagingPage(
     resourceGroupName: string,
     sqlMigrationServiceName: string,
-    options?: SqlMigrationServicesListMigrationsOptionalParams
+    options?: SqlMigrationServicesListMigrationsOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<DatabaseMigration[]> {
-    let result = await this._listMigrations(
-      resourceGroupName,
-      sqlMigrationServiceName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: SqlMigrationServicesListMigrationsResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listMigrations(
+        resourceGroupName,
+        sqlMigrationServiceName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listMigrationsNext(
         resourceGroupName,
@@ -170,7 +198,9 @@ export class SqlMigrationServicesImpl implements SqlMigrationServices {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -203,22 +233,34 @@ export class SqlMigrationServicesImpl implements SqlMigrationServices {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
-        return this.listBySubscriptionPagingPage(options);
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listBySubscriptionPagingPage(options, settings);
       }
     };
   }
 
   private async *listBySubscriptionPagingPage(
-    options?: SqlMigrationServicesListBySubscriptionOptionalParams
+    options?: SqlMigrationServicesListBySubscriptionOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<SqlMigrationService[]> {
-    let result = await this._listBySubscription(options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: SqlMigrationServicesListBySubscriptionResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listBySubscription(options);
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listBySubscriptionNext(continuationToken, options);
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }