@azure/arm-desktopvirtualization 1.0.0-beta.3 → 1.0.0-beta.4

package/src/operations/privateEndpointConnections.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 { PrivateEndpointConnections } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -16,16 +17,16 @@ import {
   PrivateEndpointConnectionWithSystemData,
   PrivateEndpointConnectionsListByHostPoolNextOptionalParams,
   PrivateEndpointConnectionsListByHostPoolOptionalParams,
+  PrivateEndpointConnectionsListByHostPoolResponse,
   PrivateEndpointConnectionsListByWorkspaceNextOptionalParams,
   PrivateEndpointConnectionsListByWorkspaceOptionalParams,
-  PrivateEndpointConnectionsListByHostPoolResponse,
+  PrivateEndpointConnectionsListByWorkspaceResponse,
   PrivateEndpointConnectionsGetByHostPoolOptionalParams,
   PrivateEndpointConnectionsGetByHostPoolResponse,
   PrivateEndpointConnectionsDeleteByHostPoolOptionalParams,
   PrivateEndpointConnection,
   PrivateEndpointConnectionsUpdateByHostPoolOptionalParams,
   PrivateEndpointConnectionsUpdateByHostPoolResponse,
-  PrivateEndpointConnectionsListByWorkspaceResponse,
   PrivateEndpointConnectionsGetByWorkspaceOptionalParams,
   PrivateEndpointConnectionsGetByWorkspaceResponse,
   PrivateEndpointConnectionsDeleteByWorkspaceOptionalParams,
@@ -72,11 +73,15 @@ export class PrivateEndpointConnectionsImpl
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
         return this.listByHostPoolPagingPage(
           resourceGroupName,
           hostPoolName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -85,15 +90,22 @@ export class PrivateEndpointConnectionsImpl
   private async *listByHostPoolPagingPage(
     resourceGroupName: string,
     hostPoolName: string,
-    options?: PrivateEndpointConnectionsListByHostPoolOptionalParams
+    options?: PrivateEndpointConnectionsListByHostPoolOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<PrivateEndpointConnectionWithSystemData[]> {
-    let result = await this._listByHostPool(
-      resourceGroupName,
-      hostPoolName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: PrivateEndpointConnectionsListByHostPoolResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listByHostPool(
+        resourceGroupName,
+        hostPoolName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listByHostPoolNext(
         resourceGroupName,
@@ -102,7 +114,9 @@ export class PrivateEndpointConnectionsImpl
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -143,11 +157,15 @@ export class PrivateEndpointConnectionsImpl
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
         return this.listByWorkspacePagingPage(
           resourceGroupName,
           workspaceName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -156,15 +174,22 @@ export class PrivateEndpointConnectionsImpl
   private async *listByWorkspacePagingPage(
     resourceGroupName: string,
     workspaceName: string,
-    options?: PrivateEndpointConnectionsListByWorkspaceOptionalParams
+    options?: PrivateEndpointConnectionsListByWorkspaceOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<PrivateEndpointConnectionWithSystemData[]> {
-    let result = await this._listByWorkspace(
-      resourceGroupName,
-      workspaceName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: PrivateEndpointConnectionsListByWorkspaceResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listByWorkspace(
+        resourceGroupName,
+        workspaceName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listByWorkspaceNext(
         resourceGroupName,
@@ -173,7 +198,9 @@ export class PrivateEndpointConnectionsImpl
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 

package/src/operations/privateLinkResources.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 { PrivateLinkResources } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -16,9 +17,9 @@ import {
   PrivateLinkResource,
   PrivateLinkResourcesListByHostPoolNextOptionalParams,
   PrivateLinkResourcesListByHostPoolOptionalParams,
+  PrivateLinkResourcesListByHostPoolResponse,
   PrivateLinkResourcesListByWorkspaceNextOptionalParams,
   PrivateLinkResourcesListByWorkspaceOptionalParams,
-  PrivateLinkResourcesListByHostPoolResponse,
   PrivateLinkResourcesListByWorkspaceResponse,
   PrivateLinkResourcesListByHostPoolNextResponse,
   PrivateLinkResourcesListByWorkspaceNextResponse
@@ -60,11 +61,15 @@ export class PrivateLinkResourcesImpl implements PrivateLinkResources {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
         return this.listByHostPoolPagingPage(
           resourceGroupName,
           hostPoolName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -73,15 +78,22 @@ export class PrivateLinkResourcesImpl implements PrivateLinkResources {
   private async *listByHostPoolPagingPage(
     resourceGroupName: string,
     hostPoolName: string,
-    options?: PrivateLinkResourcesListByHostPoolOptionalParams
+    options?: PrivateLinkResourcesListByHostPoolOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<PrivateLinkResource[]> {
-    let result = await this._listByHostPool(
-      resourceGroupName,
-      hostPoolName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: PrivateLinkResourcesListByHostPoolResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listByHostPool(
+        resourceGroupName,
+        hostPoolName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listByHostPoolNext(
         resourceGroupName,
@@ -90,7 +102,9 @@ export class PrivateLinkResourcesImpl implements PrivateLinkResources {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -131,11 +145,15 @@ export class PrivateLinkResourcesImpl implements PrivateLinkResources {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
         return this.listByWorkspacePagingPage(
           resourceGroupName,
           workspaceName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -144,15 +162,22 @@ export class PrivateLinkResourcesImpl implements PrivateLinkResources {
   private async *listByWorkspacePagingPage(
     resourceGroupName: string,
     workspaceName: string,
-    options?: PrivateLinkResourcesListByWorkspaceOptionalParams
+    options?: PrivateLinkResourcesListByWorkspaceOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<PrivateLinkResource[]> {
-    let result = await this._listByWorkspace(
-      resourceGroupName,
-      workspaceName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: PrivateLinkResourcesListByWorkspaceResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listByWorkspace(
+        resourceGroupName,
+        workspaceName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listByWorkspaceNext(
         resourceGroupName,
@@ -161,7 +186,9 @@ export class PrivateLinkResourcesImpl implements PrivateLinkResources {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 

package/src/operations/scalingPlans.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 { ScalingPlans } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -16,10 +17,13 @@ import {
   ScalingPlan,
   ScalingPlansListByResourceGroupNextOptionalParams,
   ScalingPlansListByResourceGroupOptionalParams,
+  ScalingPlansListByResourceGroupResponse,
   ScalingPlansListBySubscriptionNextOptionalParams,
   ScalingPlansListBySubscriptionOptionalParams,
+  ScalingPlansListBySubscriptionResponse,
   ScalingPlansListByHostPoolNextOptionalParams,
   ScalingPlansListByHostPoolOptionalParams,
+  ScalingPlansListByHostPoolResponse,
   ScalingPlansGetOptionalParams,
   ScalingPlansGetResponse,
   ScalingPlansCreateOptionalParams,
@@ -27,9 +31,6 @@ import {
   ScalingPlansDeleteOptionalParams,
   ScalingPlansUpdateOptionalParams,
   ScalingPlansUpdateResponse,
-  ScalingPlansListByResourceGroupResponse,
-  ScalingPlansListBySubscriptionResponse,
-  ScalingPlansListByHostPoolResponse,
   ScalingPlansListByResourceGroupNextResponse,
   ScalingPlansListBySubscriptionNextResponse,
   ScalingPlansListByHostPoolNextResponse
@@ -65,19 +66,33 @@ export class ScalingPlansImpl implements ScalingPlans {
       [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?: ScalingPlansListByResourceGroupOptionalParams
+    options?: ScalingPlansListByResourceGroupOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<ScalingPlan[]> {
-    let result = await this._listByResourceGroup(resourceGroupName, options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: ScalingPlansListByResourceGroupResponse;
+    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,
@@ -85,7 +100,9 @@ export class ScalingPlansImpl implements ScalingPlans {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -116,22 +133,34 @@ export class ScalingPlansImpl implements ScalingPlans {
       [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?: ScalingPlansListBySubscriptionOptionalParams
+    options?: ScalingPlansListBySubscriptionOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<ScalingPlan[]> {
-    let result = await this._listBySubscription(options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: ScalingPlansListBySubscriptionResponse;
+    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;
     }
   }
 
@@ -166,11 +195,15 @@ export class ScalingPlansImpl implements ScalingPlans {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
         return this.listByHostPoolPagingPage(
           resourceGroupName,
           hostPoolName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -179,15 +212,22 @@ export class ScalingPlansImpl implements ScalingPlans {
   private async *listByHostPoolPagingPage(
     resourceGroupName: string,
     hostPoolName: string,
-    options?: ScalingPlansListByHostPoolOptionalParams
+    options?: ScalingPlansListByHostPoolOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<ScalingPlan[]> {
-    let result = await this._listByHostPool(
-      resourceGroupName,
-      hostPoolName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: ScalingPlansListByHostPoolResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listByHostPool(
+        resourceGroupName,
+        hostPoolName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listByHostPoolNext(
         resourceGroupName,
@@ -196,7 +236,9 @@ export class ScalingPlansImpl implements ScalingPlans {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 

package/src/operations/sessionHosts.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 { SessionHosts } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -16,12 +17,12 @@ import {
   SessionHost,
   SessionHostsListNextOptionalParams,
   SessionHostsListOptionalParams,
+  SessionHostsListResponse,
   SessionHostsGetOptionalParams,
   SessionHostsGetResponse,
   SessionHostsDeleteOptionalParams,
   SessionHostsUpdateOptionalParams,
   SessionHostsUpdateResponse,
-  SessionHostsListResponse,
   SessionHostsListNextResponse
 } from "../models";
 
@@ -57,8 +58,16 @@ export class SessionHostsImpl implements SessionHosts {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
-        return this.listPagingPage(resourceGroupName, hostPoolName, options);
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
+        return this.listPagingPage(
+          resourceGroupName,
+          hostPoolName,
+          options,
+          settings
+        );
       }
     };
   }
@@ -66,11 +75,18 @@ export class SessionHostsImpl implements SessionHosts {
   private async *listPagingPage(
     resourceGroupName: string,
     hostPoolName: string,
-    options?: SessionHostsListOptionalParams
+    options?: SessionHostsListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<SessionHost[]> {
-    let result = await this._list(resourceGroupName, hostPoolName, options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: SessionHostsListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(resourceGroupName, hostPoolName, options);
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(
         resourceGroupName,
@@ -79,7 +95,9 @@ export class SessionHostsImpl implements SessionHosts {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 

package/src/operations/startMenuItems.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 { StartMenuItems } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -56,11 +57,15 @@ export class StartMenuItemsImpl implements StartMenuItems {
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
         return this.listPagingPage(
           resourceGroupName,
           applicationGroupName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -69,15 +74,22 @@ export class StartMenuItemsImpl implements StartMenuItems {
   private async *listPagingPage(
     resourceGroupName: string,
     applicationGroupName: string,
-    options?: StartMenuItemsListOptionalParams
+    options?: StartMenuItemsListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<StartMenuItem[]> {
-    let result = await this._list(
-      resourceGroupName,
-      applicationGroupName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: StartMenuItemsListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(
+        resourceGroupName,
+        applicationGroupName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(
         resourceGroupName,
@@ -86,7 +98,9 @@ export class StartMenuItemsImpl implements StartMenuItems {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }