@azure/arm-databricks 3.0.0-beta.3 → 3.0.0

package/src/operations/workspaces.ts

@@ -6,20 +6,27 @@
  * 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 { Workspaces } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
 import * as Parameters from "../models/parameters";
 import { AzureDatabricksManagementClient } from "../azureDatabricksManagementClient";
-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 {
   Workspace,
   WorkspacesListByResourceGroupNextOptionalParams,
   WorkspacesListByResourceGroupOptionalParams,
+  WorkspacesListByResourceGroupResponse,
   WorkspacesListBySubscriptionNextOptionalParams,
   WorkspacesListBySubscriptionOptionalParams,
+  WorkspacesListBySubscriptionResponse,
   WorkspacesGetOptionalParams,
   WorkspacesGetResponse,
   WorkspacesDeleteOptionalParams,
@@ -28,8 +35,6 @@ import {
   WorkspaceUpdate,
   WorkspacesUpdateOptionalParams,
   WorkspacesUpdateResponse,
-  WorkspacesListByResourceGroupResponse,
-  WorkspacesListBySubscriptionResponse,
   WorkspacesListByResourceGroupNextResponse,
   WorkspacesListBySubscriptionNextResponse
 } from "../models";
@@ -64,19 +69,33 @@ export class WorkspacesImpl implements Workspaces {
       [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?: WorkspacesListByResourceGroupOptionalParams
+    options?: WorkspacesListByResourceGroupOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<Workspace[]> {
-    let result = await this._listByResourceGroup(resourceGroupName, options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: WorkspacesListByResourceGroupResponse;
+    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,
@@ -84,7 +103,9 @@ export class WorkspacesImpl implements Workspaces {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -115,22 +136,34 @@ export class WorkspacesImpl implements Workspaces {
       [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?: WorkspacesListBySubscriptionOptionalParams
+    options?: WorkspacesListBySubscriptionOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<Workspace[]> {
-    let result = await this._listBySubscription(options);
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: WorkspacesListBySubscriptionResponse;
+    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;
     }
   }
 
@@ -169,14 +202,14 @@ export class WorkspacesImpl implements Workspaces {
     resourceGroupName: string,
     workspaceName: string,
     options?: WorkspacesDeleteOptionalParams
-  ): 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
     ) => {
@@ -209,13 +242,13 @@ export class WorkspacesImpl implements Workspaces {
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      { resourceGroupName, workspaceName, options },
-      deleteOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: { resourceGroupName, workspaceName, options },
+      spec: deleteOperationSpec
+    });
+    const poller = await createHttpPoller<void, OperationState<void>>(lro, {
+      restoreFrom: options?.resumeFrom,
       intervalInMs: options?.updateIntervalInMs
     });
     await poller.poll();
@@ -254,8 +287,8 @@ export class WorkspacesImpl implements Workspaces {
     parameters: Workspace,
     options?: WorkspacesCreateOrUpdateOptionalParams
   ): Promise<
-    PollerLike<
-      PollOperationState<WorkspacesCreateOrUpdateResponse>,
+    SimplePollerLike<
+      OperationState<WorkspacesCreateOrUpdateResponse>,
       WorkspacesCreateOrUpdateResponse
     >
   > {
@@ -265,7 +298,7 @@ export class WorkspacesImpl implements Workspaces {
     ): Promise<WorkspacesCreateOrUpdateResponse> => {
       return this.client.sendOperationRequest(args, spec);
     };
-    const sendOperation = async (
+    const sendOperationFn = async (
       args: coreClient.OperationArguments,
       spec: coreClient.OperationSpec
     ) => {
@@ -298,13 +331,16 @@ export class WorkspacesImpl implements Workspaces {
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      { resourceGroupName, workspaceName, parameters, options },
-      createOrUpdateOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: { resourceGroupName, workspaceName, parameters, options },
+      spec: createOrUpdateOperationSpec
+    });
+    const poller = await createHttpPoller<
+      WorkspacesCreateOrUpdateResponse,
+      OperationState<WorkspacesCreateOrUpdateResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
       intervalInMs: options?.updateIntervalInMs
     });
     await poller.poll();
@@ -346,8 +382,8 @@ export class WorkspacesImpl implements Workspaces {
     parameters: WorkspaceUpdate,
     options?: WorkspacesUpdateOptionalParams
   ): Promise<
-    PollerLike<
-      PollOperationState<WorkspacesUpdateResponse>,
+    SimplePollerLike<
+      OperationState<WorkspacesUpdateResponse>,
       WorkspacesUpdateResponse
     >
   > {
@@ -357,7 +393,7 @@ export class WorkspacesImpl implements Workspaces {
     ): Promise<WorkspacesUpdateResponse> => {
       return this.client.sendOperationRequest(args, spec);
     };
-    const sendOperation = async (
+    const sendOperationFn = async (
       args: coreClient.OperationArguments,
       spec: coreClient.OperationSpec
     ) => {
@@ -390,13 +426,16 @@ export class WorkspacesImpl implements Workspaces {
       };
     };
 
-    const lro = new LroImpl(
-      sendOperation,
-      { resourceGroupName, workspaceName, parameters, options },
-      updateOperationSpec
-    );
-    const poller = new LroEngine(lro, {
-      resumeFrom: options?.resumeFrom,
+    const lro = createLroSpec({
+      sendOperationFn,
+      args: { resourceGroupName, workspaceName, parameters, options },
+      spec: updateOperationSpec
+    });
+    const poller = await createHttpPoller<
+      WorkspacesUpdateResponse,
+      OperationState<WorkspacesUpdateResponse>
+    >(lro, {
+      restoreFrom: options?.resumeFrom,
       intervalInMs: options?.updateIntervalInMs
     });
     await poller.poll();
@@ -648,7 +687,6 @@ const listByResourceGroupNextOperationSpec: coreClient.OperationSpec = {
       bodyMapper: Mappers.ErrorResponse
     }
   },
-  queryParameters: [Parameters.apiVersion],
   urlParameters: [
     Parameters.$host,
     Parameters.resourceGroupName,
@@ -669,7 +707,6 @@ const listBySubscriptionNextOperationSpec: coreClient.OperationSpec = {
       bodyMapper: Mappers.ErrorResponse
     }
   },
-  queryParameters: [Parameters.apiVersion],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,

package/src/operationsInterfaces/accessConnectors.ts

@@ -0,0 +1,139 @@
+/*
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT License.
+ *
+ * Code generated by Microsoft (R) AutoRest Code Generator.
+ * Changes may cause incorrect behavior and will be lost if the code is regenerated.
+ */
+
+import { PagedAsyncIterableIterator } from "@azure/core-paging";
+import { SimplePollerLike, OperationState } from "@azure/core-lro";
+import {
+  AccessConnector,
+  AccessConnectorsListByResourceGroupOptionalParams,
+  AccessConnectorsListBySubscriptionOptionalParams,
+  AccessConnectorsGetOptionalParams,
+  AccessConnectorsGetResponse,
+  AccessConnectorsDeleteOptionalParams,
+  AccessConnectorsCreateOrUpdateOptionalParams,
+  AccessConnectorsCreateOrUpdateResponse,
+  AccessConnectorUpdate,
+  AccessConnectorsUpdateOptionalParams,
+  AccessConnectorsUpdateResponse
+} from "../models";
+
+/// <reference lib="esnext.asynciterable" />
+/** Interface representing a AccessConnectors. */
+export interface AccessConnectors {
+  /**
+   * Gets all the azure databricks accessConnectors within a resource group.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param options The options parameters.
+   */
+  listByResourceGroup(
+    resourceGroupName: string,
+    options?: AccessConnectorsListByResourceGroupOptionalParams
+  ): PagedAsyncIterableIterator<AccessConnector>;
+  /**
+   * Gets all the azure databricks accessConnectors within a subscription.
+   * @param options The options parameters.
+   */
+  listBySubscription(
+    options?: AccessConnectorsListBySubscriptionOptionalParams
+  ): PagedAsyncIterableIterator<AccessConnector>;
+  /**
+   * Gets an azure databricks accessConnector.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param connectorName The name of the azure databricks accessConnector.
+   * @param options The options parameters.
+   */
+  get(
+    resourceGroupName: string,
+    connectorName: string,
+    options?: AccessConnectorsGetOptionalParams
+  ): Promise<AccessConnectorsGetResponse>;
+  /**
+   * Deletes the azure databricks accessConnector.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param connectorName The name of the azure databricks accessConnector.
+   * @param options The options parameters.
+   */
+  beginDelete(
+    resourceGroupName: string,
+    connectorName: string,
+    options?: AccessConnectorsDeleteOptionalParams
+  ): Promise<SimplePollerLike<OperationState<void>, void>>;
+  /**
+   * Deletes the azure databricks accessConnector.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param connectorName The name of the azure databricks accessConnector.
+   * @param options The options parameters.
+   */
+  beginDeleteAndWait(
+    resourceGroupName: string,
+    connectorName: string,
+    options?: AccessConnectorsDeleteOptionalParams
+  ): Promise<void>;
+  /**
+   * Creates or updates azure databricks accessConnector.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param connectorName The name of the azure databricks accessConnector.
+   * @param parameters Parameters supplied to the create or update an azure databricks accessConnector.
+   * @param options The options parameters.
+   */
+  beginCreateOrUpdate(
+    resourceGroupName: string,
+    connectorName: string,
+    parameters: AccessConnector,
+    options?: AccessConnectorsCreateOrUpdateOptionalParams
+  ): Promise<
+    SimplePollerLike<
+      OperationState<AccessConnectorsCreateOrUpdateResponse>,
+      AccessConnectorsCreateOrUpdateResponse
+    >
+  >;
+  /**
+   * Creates or updates azure databricks accessConnector.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param connectorName The name of the azure databricks accessConnector.
+   * @param parameters Parameters supplied to the create or update an azure databricks accessConnector.
+   * @param options The options parameters.
+   */
+  beginCreateOrUpdateAndWait(
+    resourceGroupName: string,
+    connectorName: string,
+    parameters: AccessConnector,
+    options?: AccessConnectorsCreateOrUpdateOptionalParams
+  ): Promise<AccessConnectorsCreateOrUpdateResponse>;
+  /**
+   * Updates an azure databricks accessConnector.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param connectorName The name of the azure databricks accessConnector.
+   * @param parameters The update to the azure databricks accessConnector.
+   * @param options The options parameters.
+   */
+  beginUpdate(
+    resourceGroupName: string,
+    connectorName: string,
+    parameters: AccessConnectorUpdate,
+    options?: AccessConnectorsUpdateOptionalParams
+  ): Promise<
+    SimplePollerLike<
+      OperationState<AccessConnectorsUpdateResponse>,
+      AccessConnectorsUpdateResponse
+    >
+  >;
+  /**
+   * Updates an azure databricks accessConnector.
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param connectorName The name of the azure databricks accessConnector.
+   * @param parameters The update to the azure databricks accessConnector.
+   * @param options The options parameters.
+   */
+  beginUpdateAndWait(
+    resourceGroupName: string,
+    connectorName: string,
+    parameters: AccessConnectorUpdate,
+    options?: AccessConnectorsUpdateOptionalParams
+  ): Promise<AccessConnectorsUpdateResponse>;
+}

package/src/operationsInterfaces/index.ts

@@ -12,3 +12,4 @@ export * from "./privateLinkResources";
 export * from "./privateEndpointConnections";
 export * from "./outboundNetworkDependenciesEndpoints";
 export * from "./vNetPeering";
+export * from "./accessConnectors";

package/src/operationsInterfaces/outboundNetworkDependenciesEndpoints.ts

@@ -16,7 +16,7 @@ export interface OutboundNetworkDependenciesEndpoints {
   /**
    * Gets the list of endpoints that VNET Injected Workspace calls Azure Databricks Control Plane. You
    * must configure outbound access with these endpoints. For more information, see
-   * https://docs.microsoft.com/en-us/azure/databricks/administration-guide/cloud-configurations/azure/udr
+   * https://docs.microsoft.com/azure/databricks/administration-guide/cloud-configurations/azure/udr
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param workspaceName The name of the workspace.
    * @param options The options parameters.

package/src/operationsInterfaces/privateEndpointConnections.ts

@@ -7,7 +7,7 @@
  */
 
 import { PagedAsyncIterableIterator } from "@azure/core-paging";
-import { PollerLike, PollOperationState } from "@azure/core-lro";
+import { SimplePollerLike, OperationState } from "@azure/core-lro";
 import {
   PrivateEndpointConnection,
   PrivateEndpointConnectionsListOptionalParams,
@@ -60,8 +60,8 @@ export interface PrivateEndpointConnections {
     privateEndpointConnection: PrivateEndpointConnection,
     options?: PrivateEndpointConnectionsCreateOptionalParams
   ): Promise<
-    PollerLike<
-      PollOperationState<PrivateEndpointConnectionsCreateResponse>,
+    SimplePollerLike<
+      OperationState<PrivateEndpointConnectionsCreateResponse>,
       PrivateEndpointConnectionsCreateResponse
     >
   >;
@@ -92,7 +92,7 @@ export interface PrivateEndpointConnections {
     workspaceName: string,
     privateEndpointConnectionName: string,
     options?: PrivateEndpointConnectionsDeleteOptionalParams
-  ): Promise<PollerLike<PollOperationState<void>, void>>;
+  ): Promise<SimplePollerLike<OperationState<void>, void>>;
   /**
    * Remove private endpoint connection with the specified name
    * @param resourceGroupName The name of the resource group. The name is case insensitive.

package/src/operationsInterfaces/vNetPeering.ts

@@ -7,7 +7,7 @@
  */
 
 import { PagedAsyncIterableIterator } from "@azure/core-paging";
-import { PollerLike, PollOperationState } from "@azure/core-lro";
+import { SimplePollerLike, OperationState } from "@azure/core-lro";
 import {
   VirtualNetworkPeering,
   VNetPeeringListByWorkspaceOptionalParams,
@@ -57,7 +57,7 @@ export interface VNetPeering {
     workspaceName: string,
     peeringName: string,
     options?: VNetPeeringDeleteOptionalParams
-  ): Promise<PollerLike<PollOperationState<void>, void>>;
+  ): Promise<SimplePollerLike<OperationState<void>, void>>;
   /**
    * Deletes the workspace vNetPeering.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
@@ -86,8 +86,8 @@ export interface VNetPeering {
     virtualNetworkPeeringParameters: VirtualNetworkPeering,
     options?: VNetPeeringCreateOrUpdateOptionalParams
   ): Promise<
-    PollerLike<
-      PollOperationState<VNetPeeringCreateOrUpdateResponse>,
+    SimplePollerLike<
+      OperationState<VNetPeeringCreateOrUpdateResponse>,
       VNetPeeringCreateOrUpdateResponse
     >
   >;

package/src/operationsInterfaces/workspaces.ts

@@ -7,7 +7,7 @@
  */
 
 import { PagedAsyncIterableIterator } from "@azure/core-paging";
-import { PollerLike, PollOperationState } from "@azure/core-lro";
+import { SimplePollerLike, OperationState } from "@azure/core-lro";
 import {
   Workspace,
   WorkspacesListByResourceGroupOptionalParams,
@@ -62,7 +62,7 @@ export interface Workspaces {
     resourceGroupName: string,
     workspaceName: string,
     options?: WorkspacesDeleteOptionalParams
-  ): Promise<PollerLike<PollOperationState<void>, void>>;
+  ): Promise<SimplePollerLike<OperationState<void>, void>>;
   /**
    * Deletes the workspace.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
@@ -87,8 +87,8 @@ export interface Workspaces {
     parameters: Workspace,
     options?: WorkspacesCreateOrUpdateOptionalParams
   ): Promise<
-    PollerLike<
-      PollOperationState<WorkspacesCreateOrUpdateResponse>,
+    SimplePollerLike<
+      OperationState<WorkspacesCreateOrUpdateResponse>,
       WorkspacesCreateOrUpdateResponse
     >
   >;
@@ -118,8 +118,8 @@ export interface Workspaces {
     parameters: WorkspaceUpdate,
     options?: WorkspacesUpdateOptionalParams
   ): Promise<
-    PollerLike<
-      PollOperationState<WorkspacesUpdateResponse>,
+    SimplePollerLike<
+      OperationState<WorkspacesUpdateResponse>,
       WorkspacesUpdateResponse
     >
   >;

package/src/pagingHelper.ts

@@ -0,0 +1,39 @@
+/*
+ * Copyright (c) Microsoft Corporation.
+ * Licensed under the MIT License.
+ *
+ * Code generated by Microsoft (R) AutoRest Code Generator.
+ * Changes may cause incorrect behavior and will be lost if the code is regenerated.
+ */
+
+export interface PageInfo {
+  continuationToken?: string;
+}
+
+const pageMap = new WeakMap<object, PageInfo>();
+
+/**
+ * Given the last `.value` produced by the `byPage` iterator,
+ * returns a continuation token that can be used to begin paging from
+ * that point later.
+ * @param page An object from accessing `value` on the IteratorResult from a `byPage` iterator.
+ * @returns The continuation token that can be passed into byPage() during future calls.
+ */
+export function getContinuationToken(page: unknown): string | undefined {
+  if (typeof page !== "object" || page === null) {
+    return undefined;
+  }
+  return pageMap.get(page)?.continuationToken;
+}
+
+export function setContinuationToken(
+  page: unknown,
+  continuationToken: string | undefined
+): void {
+  if (typeof page !== "object" || page === null || !continuationToken) {
+    return;
+  }
+  const pageInfo = pageMap.get(page) ?? {};
+  pageInfo.continuationToken = continuationToken;
+  pageMap.set(page, pageInfo);
+}