@azure/arm-synapse 8.1.0-beta.1 → 9.0.0-alpha.20230210.2

package/src/operations/sqlPoolBlobAuditingPolicies.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 { SqlPoolBlobAuditingPolicies } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -16,11 +17,11 @@ import {
   SqlPoolBlobAuditingPolicy,
   SqlPoolBlobAuditingPoliciesListBySqlPoolNextOptionalParams,
   SqlPoolBlobAuditingPoliciesListBySqlPoolOptionalParams,
+  SqlPoolBlobAuditingPoliciesListBySqlPoolResponse,
   SqlPoolBlobAuditingPoliciesGetOptionalParams,
   SqlPoolBlobAuditingPoliciesGetResponse,
   SqlPoolBlobAuditingPoliciesCreateOrUpdateOptionalParams,
   SqlPoolBlobAuditingPoliciesCreateOrUpdateResponse,
-  SqlPoolBlobAuditingPoliciesListBySqlPoolResponse,
   SqlPoolBlobAuditingPoliciesListBySqlPoolNextResponse
 } from "../models";
 
@@ -64,12 +65,16 @@ export class SqlPoolBlobAuditingPoliciesImpl
       [Symbol.asyncIterator]() {
         return this;
       },
-      byPage: () => {
+      byPage: (settings?: PageSettings) => {
+        if (settings?.maxPageSize) {
+          throw new Error("maxPageSize is not supported by this operation.");
+        }
         return this.listBySqlPoolPagingPage(
           resourceGroupName,
           workspaceName,
           sqlPoolName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -79,16 +84,23 @@ export class SqlPoolBlobAuditingPoliciesImpl
     resourceGroupName: string,
     workspaceName: string,
     sqlPoolName: string,
-    options?: SqlPoolBlobAuditingPoliciesListBySqlPoolOptionalParams
+    options?: SqlPoolBlobAuditingPoliciesListBySqlPoolOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<SqlPoolBlobAuditingPolicy[]> {
-    let result = await this._listBySqlPool(
-      resourceGroupName,
-      workspaceName,
-      sqlPoolName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: SqlPoolBlobAuditingPoliciesListBySqlPoolResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._listBySqlPool(
+        resourceGroupName,
+        workspaceName,
+        sqlPoolName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listBySqlPoolNext(
         resourceGroupName,
@@ -98,7 +110,9 @@ export class SqlPoolBlobAuditingPoliciesImpl
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -280,7 +294,6 @@ const listBySqlPoolNextOperationSpec: coreClient.OperationSpec = {
     },
     default: {}
   },
-  queryParameters: [Parameters.apiVersion],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,

package/src/operations/sqlPoolGeoBackupPolicies.ts

@@ -6,7 +6,7 @@
  * 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 { SqlPoolGeoBackupPolicies } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -62,12 +62,16 @@ export class SqlPoolGeoBackupPoliciesImpl implements SqlPoolGeoBackupPolicies {
       [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,
           workspaceName,
           sqlPoolName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -77,9 +81,11 @@ export class SqlPoolGeoBackupPoliciesImpl implements SqlPoolGeoBackupPolicies {
     resourceGroupName: string,
     workspaceName: string,
     sqlPoolName: string,
-    options?: SqlPoolGeoBackupPoliciesListOptionalParams
+    options?: SqlPoolGeoBackupPoliciesListOptionalParams,
+    _settings?: PageSettings
   ): AsyncIterableIterator<GeoBackupPolicy[]> {
-    let result = await this._list(
+    let result: SqlPoolGeoBackupPoliciesListResponse;
+    result = await this._list(
       resourceGroupName,
       workspaceName,
       sqlPoolName,

package/src/operations/sqlPoolMetadataSyncConfigs.ts

@@ -89,7 +89,9 @@ const getOperationSpec: coreClient.OperationSpec = {
     200: {
       bodyMapper: Mappers.MetadataSyncConfig
     },
-    404: {},
+    404: {
+      isError: true
+    },
     default: {
       bodyMapper: Mappers.ErrorResponse
     }
@@ -113,7 +115,9 @@ const createOperationSpec: coreClient.OperationSpec = {
     200: {
       bodyMapper: Mappers.MetadataSyncConfig
     },
-    404: {},
+    404: {
+      isError: true
+    },
     default: {
       bodyMapper: Mappers.ErrorResponse
     }

package/src/operations/sqlPoolOperationResults.ts

@@ -11,6 +11,8 @@ import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
 import * as Parameters from "../models/parameters";
 import { SynapseManagementClient } from "../synapseManagementClient";
+import { PollerLike, PollOperationState, LroEngine } from "@azure/core-lro";
+import { LroImpl } from "../lroImpl";
 import {
   SqlPoolOperationResultsGetLocationHeaderResultOptionalParams,
   SqlPoolOperationResultsGetLocationHeaderResultResponse
@@ -36,17 +38,95 @@ export class SqlPoolOperationResultsImpl implements SqlPoolOperationResults {
    * @param operationId Operation ID
    * @param options The options parameters.
    */
-  getLocationHeaderResult(
+  async beginGetLocationHeaderResult(
     resourceGroupName: string,
     workspaceName: string,
     sqlPoolName: string,
     operationId: string,
     options?: SqlPoolOperationResultsGetLocationHeaderResultOptionalParams
-  ): Promise<SqlPoolOperationResultsGetLocationHeaderResultResponse> {
-    return this.client.sendOperationRequest(
+  ): Promise<
+    PollerLike<
+      PollOperationState<
+        SqlPoolOperationResultsGetLocationHeaderResultResponse
+      >,
+      SqlPoolOperationResultsGetLocationHeaderResultResponse
+    >
+  > {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<SqlPoolOperationResultsGetLocationHeaderResultResponse> => {
+      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, workspaceName, sqlPoolName, operationId, options },
       getLocationHeaderResultOperationSpec
     );
+    const poller = new LroEngine(lro, {
+      resumeFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs
+    });
+    await poller.poll();
+    return poller;
+  }
+
+  /**
+   * Get the status of a SQL pool operation
+   * @param resourceGroupName The name of the resource group. The name is case insensitive.
+   * @param workspaceName The name of the workspace.
+   * @param sqlPoolName SQL pool name
+   * @param operationId Operation ID
+   * @param options The options parameters.
+   */
+  async beginGetLocationHeaderResultAndWait(
+    resourceGroupName: string,
+    workspaceName: string,
+    sqlPoolName: string,
+    operationId: string,
+    options?: SqlPoolOperationResultsGetLocationHeaderResultOptionalParams
+  ): Promise<SqlPoolOperationResultsGetLocationHeaderResultResponse> {
+    const poller = await this.beginGetLocationHeaderResult(
+      resourceGroupName,
+      workspaceName,
+      sqlPoolName,
+      operationId,
+      options
+    );
+    return poller.pollUntilDone();
   }
 }
 // Operation Specifications
@@ -58,14 +138,16 @@ const getLocationHeaderResultOperationSpec: coreClient.OperationSpec = {
   httpMethod: "GET",
   responses: {
     200: {
-      bodyMapper: {
-        type: { name: "Dictionary", value: { type: { name: "any" } } }
-      }
+      bodyMapper: Mappers.SqlPool
+    },
+    201: {
+      bodyMapper: Mappers.SqlPool
     },
     202: {
-      bodyMapper: {
-        type: { name: "Dictionary", value: { type: { name: "any" } } }
-      }
+      bodyMapper: Mappers.SqlPool
+    },
+    204: {
+      bodyMapper: Mappers.SqlPool
     },
     default: {
       bodyMapper: Mappers.ErrorResponse

package/src/operations/sqlPoolOperations.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 { SqlPoolOperations } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -59,12 +60,16 @@ export class SqlPoolOperationsImpl implements SqlPoolOperations {
       [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,
           workspaceName,
           sqlPoolName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -74,16 +79,23 @@ export class SqlPoolOperationsImpl implements SqlPoolOperations {
     resourceGroupName: string,
     workspaceName: string,
     sqlPoolName: string,
-    options?: SqlPoolOperationsListOptionalParams
+    options?: SqlPoolOperationsListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<SqlPoolOperation[]> {
-    let result = await this._list(
-      resourceGroupName,
-      workspaceName,
-      sqlPoolName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: SqlPoolOperationsListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(
+        resourceGroupName,
+        workspaceName,
+        sqlPoolName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(
         resourceGroupName,
@@ -93,7 +105,9 @@ export class SqlPoolOperationsImpl implements SqlPoolOperations {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -186,7 +200,6 @@ const listNextOperationSpec: coreClient.OperationSpec = {
     },
     default: {}
   },
-  queryParameters: [Parameters.apiVersion],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,

package/src/operations/sqlPoolReplicationLinks.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 { SqlPoolReplicationLinks } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -61,12 +62,16 @@ export class SqlPoolReplicationLinksImpl implements SqlPoolReplicationLinks {
       [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,
           workspaceName,
           sqlPoolName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -76,16 +81,23 @@ export class SqlPoolReplicationLinksImpl implements SqlPoolReplicationLinks {
     resourceGroupName: string,
     workspaceName: string,
     sqlPoolName: string,
-    options?: SqlPoolReplicationLinksListOptionalParams
+    options?: SqlPoolReplicationLinksListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<ReplicationLink[]> {
-    let result = await this._list(
-      resourceGroupName,
-      workspaceName,
-      sqlPoolName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: SqlPoolReplicationLinksListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(
+        resourceGroupName,
+        workspaceName,
+        sqlPoolName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(
         resourceGroupName,
@@ -95,7 +107,9 @@ export class SqlPoolReplicationLinksImpl implements SqlPoolReplicationLinks {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -237,7 +251,6 @@ const listNextOperationSpec: coreClient.OperationSpec = {
       bodyMapper: Mappers.ErrorResponse
     }
   },
-  queryParameters: [Parameters.apiVersion],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,

package/src/operations/sqlPoolRestorePoints.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 { SqlPoolRestorePoints } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -67,12 +68,16 @@ export class SqlPoolRestorePointsImpl implements SqlPoolRestorePoints {
       [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,
           workspaceName,
           sqlPoolName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -82,16 +87,23 @@ export class SqlPoolRestorePointsImpl implements SqlPoolRestorePoints {
     resourceGroupName: string,
     workspaceName: string,
     sqlPoolName: string,
-    options?: SqlPoolRestorePointsListOptionalParams
+    options?: SqlPoolRestorePointsListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<RestorePoint[]> {
-    let result = await this._list(
-      resourceGroupName,
-      workspaceName,
-      sqlPoolName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: SqlPoolRestorePointsListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(
+        resourceGroupName,
+        workspaceName,
+        sqlPoolName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(
         resourceGroupName,
@@ -101,7 +113,9 @@ export class SqlPoolRestorePointsImpl implements SqlPoolRestorePoints {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -420,7 +434,6 @@ const listNextOperationSpec: coreClient.OperationSpec = {
       bodyMapper: Mappers.ErrorResponse
     }
   },
-  queryParameters: [Parameters.apiVersion],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,

package/src/operations/sqlPoolSchemas.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 { SqlPoolSchemas } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -61,12 +62,16 @@ export class SqlPoolSchemasImpl implements SqlPoolSchemas {
       [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,
           workspaceName,
           sqlPoolName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -76,16 +81,23 @@ export class SqlPoolSchemasImpl implements SqlPoolSchemas {
     resourceGroupName: string,
     workspaceName: string,
     sqlPoolName: string,
-    options?: SqlPoolSchemasListOptionalParams
+    options?: SqlPoolSchemasListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<SqlPoolSchema[]> {
-    let result = await this._list(
-      resourceGroupName,
-      workspaceName,
-      sqlPoolName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: SqlPoolSchemasListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(
+        resourceGroupName,
+        workspaceName,
+        sqlPoolName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(
         resourceGroupName,
@@ -95,7 +107,9 @@ export class SqlPoolSchemasImpl implements SqlPoolSchemas {
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -231,7 +245,6 @@ const listNextOperationSpec: coreClient.OperationSpec = {
     },
     default: {}
   },
-  queryParameters: [Parameters.apiVersion, Parameters.filter],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,

package/src/operations/sqlPoolSecurityAlertPolicies.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 { SqlPoolSecurityAlertPolicies } from "../operationsInterfaces";
 import * as coreClient from "@azure/core-client";
 import * as Mappers from "../models/mappers";
@@ -65,12 +66,16 @@ export class SqlPoolSecurityAlertPoliciesImpl
       [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,
           workspaceName,
           sqlPoolName,
-          options
+          options,
+          settings
         );
       }
     };
@@ -80,16 +85,23 @@ export class SqlPoolSecurityAlertPoliciesImpl
     resourceGroupName: string,
     workspaceName: string,
     sqlPoolName: string,
-    options?: SqlPoolSecurityAlertPoliciesListOptionalParams
+    options?: SqlPoolSecurityAlertPoliciesListOptionalParams,
+    settings?: PageSettings
   ): AsyncIterableIterator<SqlPoolSecurityAlertPolicy[]> {
-    let result = await this._list(
-      resourceGroupName,
-      workspaceName,
-      sqlPoolName,
-      options
-    );
-    yield result.value || [];
-    let continuationToken = result.nextLink;
+    let result: SqlPoolSecurityAlertPoliciesListResponse;
+    let continuationToken = settings?.continuationToken;
+    if (!continuationToken) {
+      result = await this._list(
+        resourceGroupName,
+        workspaceName,
+        sqlPoolName,
+        options
+      );
+      let page = result.value || [];
+      continuationToken = result.nextLink;
+      setContinuationToken(page, continuationToken);
+      yield page;
+    }
     while (continuationToken) {
       result = await this._listNext(
         resourceGroupName,
@@ -99,7 +111,9 @@ export class SqlPoolSecurityAlertPoliciesImpl
         options
       );
       continuationToken = result.nextLink;
-      yield result.value || [];
+      let page = result.value || [];
+      setContinuationToken(page, continuationToken);
+      yield page;
     }
   }
 
@@ -298,7 +312,6 @@ const listNextOperationSpec: coreClient.OperationSpec = {
     },
     default: {}
   },
-  queryParameters: [Parameters.apiVersion],
   urlParameters: [
     Parameters.$host,
     Parameters.subscriptionId,