@azure/arm-streamanalytics 3.0.0 → 4.0.0

package/src/operations/privateEndpoints.ts

@@ -3,70 +3,140 @@
  * 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.
+ * Changes may cause incorrect behavior and will be lost if the code is regenerated.
  */
 
-import * as msRest from "@azure/ms-rest-js";
-import * as msRestAzure from "@azure/ms-rest-azure-js";
-import * as Models from "../models";
-import * as Mappers from "../models/privateEndpointsMappers";
+import { PagedAsyncIterableIterator } from "@azure/core-paging";
+import { PrivateEndpoints } from "../operationsInterfaces";
+import * as coreClient from "@azure/core-client";
+import * as Mappers from "../models/mappers";
 import * as Parameters from "../models/parameters";
-import { StreamAnalyticsManagementClientContext } from "../streamAnalyticsManagementClientContext";
+import { StreamAnalyticsManagementClient } from "../streamAnalyticsManagementClient";
+import { PollerLike, PollOperationState, LroEngine } from "@azure/core-lro";
+import { LroImpl } from "../lroImpl";
+import {
+  PrivateEndpoint,
+  PrivateEndpointsListByClusterNextOptionalParams,
+  PrivateEndpointsListByClusterOptionalParams,
+  PrivateEndpointsCreateOrUpdateOptionalParams,
+  PrivateEndpointsCreateOrUpdateResponse,
+  PrivateEndpointsGetOptionalParams,
+  PrivateEndpointsGetResponse,
+  PrivateEndpointsDeleteOptionalParams,
+  PrivateEndpointsListByClusterResponse,
+  PrivateEndpointsListByClusterNextResponse
+} from "../models";
 
-/** Class representing a PrivateEndpoints. */
-export class PrivateEndpoints {
-  private readonly client: StreamAnalyticsManagementClientContext;
+/// <reference lib="esnext.asynciterable" />
+/** Class containing PrivateEndpoints operations. */
+export class PrivateEndpointsImpl implements PrivateEndpoints {
+  private readonly client: StreamAnalyticsManagementClient;
 
   /**
-   * Create a PrivateEndpoints.
-   * @param {StreamAnalyticsManagementClientContext} client Reference to the service client.
+   * Initialize a new instance of the class PrivateEndpoints class.
+   * @param client Reference to the service client
    */
-  constructor(client: StreamAnalyticsManagementClientContext) {
+  constructor(client: StreamAnalyticsManagementClient) {
     this.client = client;
   }
 
   /**
-   * Creates a Stream Analytics Private Endpoint or replaces an already existing Private Endpoint.
-   * @param privateEndpoint The definition of the private endpoint that will be used to create a new
-   * cluster or replace the existing one.
+   * Lists the private endpoints in the cluster.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param clusterName The name of the cluster.
-   * @param privateEndpointName The name of the private endpoint.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.PrivateEndpointsCreateOrUpdateResponse>
+   * @param options The options parameters.
    */
-  createOrUpdate(privateEndpoint: Models.PrivateEndpoint, resourceGroupName: string, clusterName: string, privateEndpointName: string, options?: Models.PrivateEndpointsCreateOrUpdateOptionalParams): Promise<Models.PrivateEndpointsCreateOrUpdateResponse>;
+  public listByCluster(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: PrivateEndpointsListByClusterOptionalParams
+  ): PagedAsyncIterableIterator<PrivateEndpoint> {
+    const iter = this.listByClusterPagingAll(
+      resourceGroupName,
+      clusterName,
+      options
+    );
+    return {
+      next() {
+        return iter.next();
+      },
+      [Symbol.asyncIterator]() {
+        return this;
+      },
+      byPage: () => {
+        return this.listByClusterPagingPage(
+          resourceGroupName,
+          clusterName,
+          options
+        );
+      }
+    };
+  }
+
+  private async *listByClusterPagingPage(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: PrivateEndpointsListByClusterOptionalParams
+  ): AsyncIterableIterator<PrivateEndpoint[]> {
+    let result = await this._listByCluster(
+      resourceGroupName,
+      clusterName,
+      options
+    );
+    yield result.value || [];
+    let continuationToken = result.nextLink;
+    while (continuationToken) {
+      result = await this._listByClusterNext(
+        resourceGroupName,
+        clusterName,
+        continuationToken,
+        options
+      );
+      continuationToken = result.nextLink;
+      yield result.value || [];
+    }
+  }
+
+  private async *listByClusterPagingAll(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: PrivateEndpointsListByClusterOptionalParams
+  ): AsyncIterableIterator<PrivateEndpoint> {
+    for await (const page of this.listByClusterPagingPage(
+      resourceGroupName,
+      clusterName,
+      options
+    )) {
+      yield* page;
+    }
+  }
+
   /**
-   * @param privateEndpoint The definition of the private endpoint that will be used to create a new
-   * cluster or replace the existing one.
+   * Creates a Stream Analytics Private Endpoint or replaces an already existing Private Endpoint.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param clusterName The name of the cluster.
    * @param privateEndpointName The name of the private endpoint.
-   * @param callback The callback
-   */
-  createOrUpdate(privateEndpoint: Models.PrivateEndpoint, resourceGroupName: string, clusterName: string, privateEndpointName: string, callback: msRest.ServiceCallback<Models.PrivateEndpoint>): void;
-  /**
    * @param privateEndpoint The definition of the private endpoint that will be used to create a new
-   * cluster or replace the existing one.
-   * @param resourceGroupName The name of the resource group. The name is case insensitive.
-   * @param clusterName The name of the cluster.
-   * @param privateEndpointName The name of the private endpoint.
-   * @param options The optional parameters
-   * @param callback The callback
+   *                        cluster or replace the existing one.
+   * @param options The options parameters.
    */
-  createOrUpdate(privateEndpoint: Models.PrivateEndpoint, resourceGroupName: string, clusterName: string, privateEndpointName: string, options: Models.PrivateEndpointsCreateOrUpdateOptionalParams, callback: msRest.ServiceCallback<Models.PrivateEndpoint>): void;
-  createOrUpdate(privateEndpoint: Models.PrivateEndpoint, resourceGroupName: string, clusterName: string, privateEndpointName: string, options?: Models.PrivateEndpointsCreateOrUpdateOptionalParams | msRest.ServiceCallback<Models.PrivateEndpoint>, callback?: msRest.ServiceCallback<Models.PrivateEndpoint>): Promise<Models.PrivateEndpointsCreateOrUpdateResponse> {
+  createOrUpdate(
+    resourceGroupName: string,
+    clusterName: string,
+    privateEndpointName: string,
+    privateEndpoint: PrivateEndpoint,
+    options?: PrivateEndpointsCreateOrUpdateOptionalParams
+  ): Promise<PrivateEndpointsCreateOrUpdateResponse> {
     return this.client.sendOperationRequest(
       {
-        privateEndpoint,
         resourceGroupName,
         clusterName,
         privateEndpointName,
+        privateEndpoint,
         options
       },
-      createOrUpdateOperationSpec,
-      callback) as Promise<Models.PrivateEndpointsCreateOrUpdateResponse>;
+      createOrUpdateOperationSpec
+    );
   }
 
   /**
@@ -74,35 +144,81 @@ export class PrivateEndpoints {
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param clusterName The name of the cluster.
    * @param privateEndpointName The name of the private endpoint.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.PrivateEndpointsGetResponse>
+   * @param options The options parameters.
    */
-  get(resourceGroupName: string, clusterName: string, privateEndpointName: string, options?: msRest.RequestOptionsBase): Promise<Models.PrivateEndpointsGetResponse>;
-  /**
-   * @param resourceGroupName The name of the resource group. The name is case insensitive.
-   * @param clusterName The name of the cluster.
-   * @param privateEndpointName The name of the private endpoint.
-   * @param callback The callback
-   */
-  get(resourceGroupName: string, clusterName: string, privateEndpointName: string, callback: msRest.ServiceCallback<Models.PrivateEndpoint>): void;
+  get(
+    resourceGroupName: string,
+    clusterName: string,
+    privateEndpointName: string,
+    options?: PrivateEndpointsGetOptionalParams
+  ): Promise<PrivateEndpointsGetResponse> {
+    return this.client.sendOperationRequest(
+      { resourceGroupName, clusterName, privateEndpointName, options },
+      getOperationSpec
+    );
+  }
+
   /**
+   * Delete the specified private endpoint.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param clusterName The name of the cluster.
    * @param privateEndpointName The name of the private endpoint.
-   * @param options The optional parameters
-   * @param callback The callback
+   * @param options The options parameters.
    */
-  get(resourceGroupName: string, clusterName: string, privateEndpointName: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.PrivateEndpoint>): void;
-  get(resourceGroupName: string, clusterName: string, privateEndpointName: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.PrivateEndpoint>, callback?: msRest.ServiceCallback<Models.PrivateEndpoint>): Promise<Models.PrivateEndpointsGetResponse> {
-    return this.client.sendOperationRequest(
-      {
-        resourceGroupName,
-        clusterName,
-        privateEndpointName,
-        options
-      },
-      getOperationSpec,
-      callback) as Promise<Models.PrivateEndpointsGetResponse>;
+  async beginDelete(
+    resourceGroupName: string,
+    clusterName: string,
+    privateEndpointName: string,
+    options?: PrivateEndpointsDeleteOptionalParams
+  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<void> => {
+      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, clusterName, privateEndpointName, options },
+      deleteOperationSpec
+    );
+    return new LroEngine(lro, {
+      resumeFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs
+    });
   }
 
   /**
@@ -110,121 +226,66 @@ export class PrivateEndpoints {
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param clusterName The name of the cluster.
    * @param privateEndpointName The name of the private endpoint.
-   * @param [options] The optional parameters
-   * @returns Promise<msRest.RestResponse>
+   * @param options The options parameters.
    */
-  deleteMethod(resourceGroupName: string, clusterName: string, privateEndpointName: string, options?: msRest.RequestOptionsBase): Promise<msRest.RestResponse> {
-    return this.beginDeleteMethod(resourceGroupName,clusterName,privateEndpointName,options)
-      .then(lroPoller => lroPoller.pollUntilFinished());
+  async beginDeleteAndWait(
+    resourceGroupName: string,
+    clusterName: string,
+    privateEndpointName: string,
+    options?: PrivateEndpointsDeleteOptionalParams
+  ): Promise<void> {
+    const poller = await this.beginDelete(
+      resourceGroupName,
+      clusterName,
+      privateEndpointName,
+      options
+    );
+    return poller.pollUntilDone();
   }
 
   /**
    * Lists the private endpoints in the cluster.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param clusterName The name of the cluster.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.PrivateEndpointsListByClusterResponse>
+   * @param options The options parameters.
    */
-  listByCluster(resourceGroupName: string, clusterName: string, options?: msRest.RequestOptionsBase): Promise<Models.PrivateEndpointsListByClusterResponse>;
-  /**
-   * @param resourceGroupName The name of the resource group. The name is case insensitive.
-   * @param clusterName The name of the cluster.
-   * @param callback The callback
-   */
-  listByCluster(resourceGroupName: string, clusterName: string, callback: msRest.ServiceCallback<Models.PrivateEndpointListResult>): void;
-  /**
-   * @param resourceGroupName The name of the resource group. The name is case insensitive.
-   * @param clusterName The name of the cluster.
-   * @param options The optional parameters
-   * @param callback The callback
-   */
-  listByCluster(resourceGroupName: string, clusterName: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.PrivateEndpointListResult>): void;
-  listByCluster(resourceGroupName: string, clusterName: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.PrivateEndpointListResult>, callback?: msRest.ServiceCallback<Models.PrivateEndpointListResult>): Promise<Models.PrivateEndpointsListByClusterResponse> {
+  private _listByCluster(
+    resourceGroupName: string,
+    clusterName: string,
+    options?: PrivateEndpointsListByClusterOptionalParams
+  ): Promise<PrivateEndpointsListByClusterResponse> {
     return this.client.sendOperationRequest(
-      {
-        resourceGroupName,
-        clusterName,
-        options
-      },
-      listByClusterOperationSpec,
-      callback) as Promise<Models.PrivateEndpointsListByClusterResponse>;
+      { resourceGroupName, clusterName, options },
+      listByClusterOperationSpec
+    );
   }
 
   /**
-   * Delete the specified private endpoint.
+   * ListByClusterNext
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param clusterName The name of the cluster.
-   * @param privateEndpointName The name of the private endpoint.
-   * @param [options] The optional parameters
-   * @returns Promise<msRestAzure.LROPoller>
+   * @param nextLink The nextLink from the previous successful call to the ListByCluster method.
+   * @param options The options parameters.
    */
-  beginDeleteMethod(resourceGroupName: string, clusterName: string, privateEndpointName: string, options?: msRest.RequestOptionsBase): Promise<msRestAzure.LROPoller> {
-    return this.client.sendLRORequest(
-      {
-        resourceGroupName,
-        clusterName,
-        privateEndpointName,
-        options
-      },
-      beginDeleteMethodOperationSpec,
-      options);
-  }
-
-  /**
-   * Lists the private endpoints in the cluster.
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.PrivateEndpointsListByClusterNextResponse>
-   */
-  listByClusterNext(nextPageLink: string, options?: msRest.RequestOptionsBase): Promise<Models.PrivateEndpointsListByClusterNextResponse>;
-  /**
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param callback The callback
-   */
-  listByClusterNext(nextPageLink: string, callback: msRest.ServiceCallback<Models.PrivateEndpointListResult>): void;
-  /**
-   * @param nextPageLink The NextLink from the previous successful call to List operation.
-   * @param options The optional parameters
-   * @param callback The callback
-   */
-  listByClusterNext(nextPageLink: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.PrivateEndpointListResult>): void;
-  listByClusterNext(nextPageLink: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.PrivateEndpointListResult>, callback?: msRest.ServiceCallback<Models.PrivateEndpointListResult>): Promise<Models.PrivateEndpointsListByClusterNextResponse> {
+  private _listByClusterNext(
+    resourceGroupName: string,
+    clusterName: string,
+    nextLink: string,
+    options?: PrivateEndpointsListByClusterNextOptionalParams
+  ): Promise<PrivateEndpointsListByClusterNextResponse> {
     return this.client.sendOperationRequest(
-      {
-        nextPageLink,
-        options
-      },
-      listByClusterNextOperationSpec,
-      callback) as Promise<Models.PrivateEndpointsListByClusterNextResponse>;
+      { resourceGroupName, clusterName, nextLink, options },
+      listByClusterNextOperationSpec
+    );
   }
 }
-
 // Operation Specifications
-const serializer = new msRest.Serializer(Mappers);
-const createOrUpdateOperationSpec: msRest.OperationSpec = {
+const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
+
+const createOrUpdateOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StreamAnalytics/clusters/{clusterName}/privateEndpoints/{privateEndpointName}",
   httpMethod: "PUT",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StreamAnalytics/clusters/{clusterName}/privateEndpoints/{privateEndpointName}",
-  urlParameters: [
-    Parameters.subscriptionId,
-    Parameters.resourceGroupName,
-    Parameters.clusterName,
-    Parameters.privateEndpointName
-  ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.ifMatch,
-    Parameters.ifNoneMatch,
-    Parameters.acceptLanguage
-  ],
-  requestBody: {
-    parameterPath: "privateEndpoint",
-    mapper: {
-      ...Mappers.PrivateEndpoint,
-      required: true
-    }
-  },
   responses: {
     200: {
       bodyMapper: Mappers.PrivateEndpoint
@@ -236,24 +297,28 @@ const createOrUpdateOperationSpec: msRest.OperationSpec = {
       bodyMapper: Mappers.ErrorModel
     }
   },
-  serializer
-};
-
-const getOperationSpec: msRest.OperationSpec = {
-  httpMethod: "GET",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StreamAnalytics/clusters/{clusterName}/privateEndpoints/{privateEndpointName}",
+  requestBody: Parameters.privateEndpoint,
+  queryParameters: [Parameters.apiVersion],
   urlParameters: [
+    Parameters.$host,
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.clusterName,
     Parameters.privateEndpointName
   ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
   headerParameters: [
-    Parameters.acceptLanguage
+    Parameters.accept,
+    Parameters.contentType,
+    Parameters.ifMatch,
+    Parameters.ifNoneMatch
   ],
+  mediaType: "json",
+  serializer
+};
+const getOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StreamAnalytics/clusters/{clusterName}/privateEndpoints/{privateEndpointName}",
+  httpMethod: "GET",
   responses: {
     200: {
       bodyMapper: Mappers.PrivateEndpoint
@@ -262,73 +327,66 @@ const getOperationSpec: msRest.OperationSpec = {
       bodyMapper: Mappers.ErrorModel
     }
   },
-  serializer
-};
-
-const listByClusterOperationSpec: msRest.OperationSpec = {
-  httpMethod: "GET",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StreamAnalytics/clusters/{clusterName}/privateEndpoints",
+  queryParameters: [Parameters.apiVersion],
   urlParameters: [
+    Parameters.$host,
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
-    Parameters.clusterName
-  ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
+    Parameters.clusterName,
+    Parameters.privateEndpointName
   ],
+  headerParameters: [Parameters.accept],
+  serializer
+};
+const deleteOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StreamAnalytics/clusters/{clusterName}/privateEndpoints/{privateEndpointName}",
+  httpMethod: "DELETE",
   responses: {
-    200: {
-      bodyMapper: Mappers.PrivateEndpointListResult
-    },
+    200: {},
+    201: {},
+    202: {},
+    204: {},
     default: {
       bodyMapper: Mappers.ErrorModel
     }
   },
-  serializer
-};
-
-const beginDeleteMethodOperationSpec: msRest.OperationSpec = {
-  httpMethod: "DELETE",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StreamAnalytics/clusters/{clusterName}/privateEndpoints/{privateEndpointName}",
+  queryParameters: [Parameters.apiVersion],
   urlParameters: [
+    Parameters.$host,
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.clusterName,
     Parameters.privateEndpointName
   ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
+  headerParameters: [Parameters.accept],
+  serializer
+};
+const listByClusterOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.StreamAnalytics/clusters/{clusterName}/privateEndpoints",
+  httpMethod: "GET",
   responses: {
-    200: {},
-    202: {},
-    204: {},
+    200: {
+      bodyMapper: Mappers.PrivateEndpointListResult
+    },
     default: {
       bodyMapper: Mappers.ErrorModel
     }
   },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.clusterName
+  ],
+  headerParameters: [Parameters.accept],
   serializer
 };
-
-const listByClusterNextOperationSpec: msRest.OperationSpec = {
-  httpMethod: "GET",
-  baseUrl: "https://management.azure.com",
+const listByClusterNextOperationSpec: coreClient.OperationSpec = {
   path: "{nextLink}",
-  urlParameters: [
-    Parameters.nextPageLink
-  ],
-  queryParameters: [
-    Parameters.apiVersion1
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
+  httpMethod: "GET",
   responses: {
     200: {
       bodyMapper: Mappers.PrivateEndpointListResult
@@ -337,5 +395,14 @@ const listByClusterNextOperationSpec: msRest.OperationSpec = {
       bodyMapper: Mappers.ErrorModel
     }
   },
+  queryParameters: [Parameters.apiVersion],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.nextLink,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.clusterName
+  ],
+  headerParameters: [Parameters.accept],
   serializer
 };