@azure/arm-mysql 4.0.0 → 5.0.0-alpha.20211221.3

package/src/mySQLManagementClient.ts

@@ -3,118 +3,180 @@
  * 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 { TokenCredential } from "@azure/core-auth";
-import * as msRestAzure from "@azure/ms-rest-azure-js";
-import * as Models from "./models";
-import * as Mappers from "./models/mappers";
+import * as coreClient from "@azure/core-client";
+import * as coreAuth from "@azure/core-auth";
+import { PollerLike, PollOperationState, LroEngine } from "@azure/core-lro";
+import { LroImpl } from "./lroImpl";
+import {
+  ServersImpl,
+  ReplicasImpl,
+  FirewallRulesImpl,
+  VirtualNetworkRulesImpl,
+  DatabasesImpl,
+  ConfigurationsImpl,
+  ServerParametersImpl,
+  LogFilesImpl,
+  ServerAdministratorsImpl,
+  RecoverableServersImpl,
+  ServerBasedPerformanceTierImpl,
+  LocationBasedPerformanceTierImpl,
+  CheckNameAvailabilityImpl,
+  OperationsImpl,
+  ServerSecurityAlertPoliciesImpl,
+  QueryTextsImpl,
+  TopQueryStatisticsImpl,
+  WaitStatisticsImpl,
+  AdvisorsImpl,
+  RecommendedActionsImpl,
+  LocationBasedRecommendedActionSessionsOperationStatusImpl,
+  LocationBasedRecommendedActionSessionsResultImpl,
+  PrivateEndpointConnectionsImpl,
+  PrivateLinkResourcesImpl,
+  ServerKeysImpl
+} from "./operations";
+import {
+  Servers,
+  Replicas,
+  FirewallRules,
+  VirtualNetworkRules,
+  Databases,
+  Configurations,
+  ServerParameters,
+  LogFiles,
+  ServerAdministrators,
+  RecoverableServers,
+  ServerBasedPerformanceTier,
+  LocationBasedPerformanceTier,
+  CheckNameAvailability,
+  Operations,
+  ServerSecurityAlertPolicies,
+  QueryTexts,
+  TopQueryStatistics,
+  WaitStatistics,
+  Advisors,
+  RecommendedActions,
+  LocationBasedRecommendedActionSessionsOperationStatus,
+  LocationBasedRecommendedActionSessionsResult,
+  PrivateEndpointConnections,
+  PrivateLinkResources,
+  ServerKeys
+} from "./operationsInterfaces";
 import * as Parameters from "./models/parameters";
-import * as operations from "./operations";
-import { MySQLManagementClientContext } from "./mySQLManagementClientContext";
-
+import * as Mappers from "./models/mappers";
+import {
+  MySQLManagementClientOptionalParams,
+  ResetQueryPerformanceInsightDataOptionalParams,
+  ResetQueryPerformanceInsightDataResponse,
+  CreateRecommendedActionSessionOptionalParams
+} from "./models";
 
-class MySQLManagementClient extends MySQLManagementClientContext {
-  // Operation groups
-  servers: operations.Servers;
-  replicas: operations.Replicas;
-  firewallRules: operations.FirewallRules;
-  virtualNetworkRules: operations.VirtualNetworkRules;
-  databases: operations.Databases;
-  configurations: operations.Configurations;
-  serverParameters: operations.ServerParameters;
-  logFiles: operations.LogFiles;
-  serverAdministrators: operations.ServerAdministrators;
-  recoverableServers: operations.RecoverableServers;
-  serverBasedPerformanceTier: operations.ServerBasedPerformanceTier;
-  locationBasedPerformanceTier: operations.LocationBasedPerformanceTier;
-  checkNameAvailability: operations.CheckNameAvailability;
-  operations: operations.Operations;
-  serverSecurityAlertPolicies: operations.ServerSecurityAlertPolicies;
-  queryTexts: operations.QueryTexts;
-  topQueryStatistics: operations.TopQueryStatistics;
-  waitStatistics: operations.WaitStatistics;
-  advisors: operations.Advisors;
-  recommendedActions: operations.RecommendedActions;
-  locationBasedRecommendedActionSessionsOperationStatus: operations.LocationBasedRecommendedActionSessionsOperationStatus;
-  locationBasedRecommendedActionSessionsResult: operations.LocationBasedRecommendedActionSessionsResult;
-  privateEndpointConnections: operations.PrivateEndpointConnections;
-  privateLinkResources: operations.PrivateLinkResources;
-  serverKeys: operations.ServerKeys;
+export class MySQLManagementClient extends coreClient.ServiceClient {
+  $host: string;
+  subscriptionId: string;
 
   /**
    * Initializes a new instance of the MySQLManagementClient class.
-   * @param credentials Credentials needed for the client to connect to Azure. Credentials
-   * implementing the TokenCredential interface from the @azure/identity package are recommended. For
-   * more information about these credentials, see
-   * {@link https://www.npmjs.com/package/@azure/identity}. Credentials implementing the
-   * ServiceClientCredentials interface from the older packages @azure/ms-rest-nodeauth and
-   * @azure/ms-rest-browserauth are also supported.
+   * @param credentials Subscription credentials which uniquely identify client subscription.
    * @param subscriptionId The ID of the target subscription.
-   * @param [options] The parameter options
+   * @param options The parameter options
    */
-  constructor(credentials: msRest.ServiceClientCredentials | TokenCredential, subscriptionId: string, options?: Models.MySQLManagementClientOptions) {
-    super(credentials, subscriptionId, options);
-    this.servers = new operations.Servers(this);
-    this.replicas = new operations.Replicas(this);
-    this.firewallRules = new operations.FirewallRules(this);
-    this.virtualNetworkRules = new operations.VirtualNetworkRules(this);
-    this.databases = new operations.Databases(this);
-    this.configurations = new operations.Configurations(this);
-    this.serverParameters = new operations.ServerParameters(this);
-    this.logFiles = new operations.LogFiles(this);
-    this.serverAdministrators = new operations.ServerAdministrators(this);
-    this.recoverableServers = new operations.RecoverableServers(this);
-    this.serverBasedPerformanceTier = new operations.ServerBasedPerformanceTier(this);
-    this.locationBasedPerformanceTier = new operations.LocationBasedPerformanceTier(this);
-    this.checkNameAvailability = new operations.CheckNameAvailability(this);
-    this.operations = new operations.Operations(this);
-    this.serverSecurityAlertPolicies = new operations.ServerSecurityAlertPolicies(this);
-    this.queryTexts = new operations.QueryTexts(this);
-    this.topQueryStatistics = new operations.TopQueryStatistics(this);
-    this.waitStatistics = new operations.WaitStatistics(this);
-    this.advisors = new operations.Advisors(this);
-    this.recommendedActions = new operations.RecommendedActions(this);
-    this.locationBasedRecommendedActionSessionsOperationStatus = new operations.LocationBasedRecommendedActionSessionsOperationStatus(this);
-    this.locationBasedRecommendedActionSessionsResult = new operations.LocationBasedRecommendedActionSessionsResult(this);
-    this.privateEndpointConnections = new operations.PrivateEndpointConnections(this);
-    this.privateLinkResources = new operations.PrivateLinkResources(this);
-    this.serverKeys = new operations.ServerKeys(this);
+  constructor(
+    credentials: coreAuth.TokenCredential,
+    subscriptionId: string,
+    options?: MySQLManagementClientOptionalParams
+  ) {
+    if (credentials === undefined) {
+      throw new Error("'credentials' cannot be null");
+    }
+    if (subscriptionId === undefined) {
+      throw new Error("'subscriptionId' cannot be null");
+    }
+
+    // Initializing default values for options
+    if (!options) {
+      options = {};
+    }
+    const defaults: MySQLManagementClientOptionalParams = {
+      requestContentType: "application/json; charset=utf-8",
+      credential: credentials
+    };
+
+    const packageDetails = `azsdk-js-arm-mysql/5.0.0`;
+    const userAgentPrefix =
+      options.userAgentOptions && options.userAgentOptions.userAgentPrefix
+        ? `${options.userAgentOptions.userAgentPrefix} ${packageDetails}`
+        : `${packageDetails}`;
+
+    if (!options.credentialScopes) {
+      options.credentialScopes = ["https://management.azure.com/.default"];
+    }
+    const optionsWithDefaults = {
+      ...defaults,
+      ...options,
+      userAgentOptions: {
+        userAgentPrefix
+      },
+      baseUri: options.endpoint || "https://management.azure.com"
+    };
+    super(optionsWithDefaults);
+    // Parameter assignments
+    this.subscriptionId = subscriptionId;
+
+    // Assigning values to Constant parameters
+    this.$host = options.$host || "https://management.azure.com";
+    this.servers = new ServersImpl(this);
+    this.replicas = new ReplicasImpl(this);
+    this.firewallRules = new FirewallRulesImpl(this);
+    this.virtualNetworkRules = new VirtualNetworkRulesImpl(this);
+    this.databases = new DatabasesImpl(this);
+    this.configurations = new ConfigurationsImpl(this);
+    this.serverParameters = new ServerParametersImpl(this);
+    this.logFiles = new LogFilesImpl(this);
+    this.serverAdministrators = new ServerAdministratorsImpl(this);
+    this.recoverableServers = new RecoverableServersImpl(this);
+    this.serverBasedPerformanceTier = new ServerBasedPerformanceTierImpl(this);
+    this.locationBasedPerformanceTier = new LocationBasedPerformanceTierImpl(
+      this
+    );
+    this.checkNameAvailability = new CheckNameAvailabilityImpl(this);
+    this.operations = new OperationsImpl(this);
+    this.serverSecurityAlertPolicies = new ServerSecurityAlertPoliciesImpl(
+      this
+    );
+    this.queryTexts = new QueryTextsImpl(this);
+    this.topQueryStatistics = new TopQueryStatisticsImpl(this);
+    this.waitStatistics = new WaitStatisticsImpl(this);
+    this.advisors = new AdvisorsImpl(this);
+    this.recommendedActions = new RecommendedActionsImpl(this);
+    this.locationBasedRecommendedActionSessionsOperationStatus = new LocationBasedRecommendedActionSessionsOperationStatusImpl(
+      this
+    );
+    this.locationBasedRecommendedActionSessionsResult = new LocationBasedRecommendedActionSessionsResultImpl(
+      this
+    );
+    this.privateEndpointConnections = new PrivateEndpointConnectionsImpl(this);
+    this.privateLinkResources = new PrivateLinkResourcesImpl(this);
+    this.serverKeys = new ServerKeysImpl(this);
   }
 
   /**
    * Reset data for Query Performance Insight.
    * @param resourceGroupName The name of the resource group. The name is case insensitive.
    * @param serverName The name of the server.
-   * @param [options] The optional parameters
-   * @returns Promise<Models.ResetQueryPerformanceInsightDataResponse>
+   * @param options The options parameters.
    */
-  resetQueryPerformanceInsightData(resourceGroupName: string, serverName: string, options?: msRest.RequestOptionsBase): Promise<Models.ResetQueryPerformanceInsightDataResponse>;
-  /**
-   * @param resourceGroupName The name of the resource group. The name is case insensitive.
-   * @param serverName The name of the server.
-   * @param callback The callback
-   */
-  resetQueryPerformanceInsightData(resourceGroupName: string, serverName: string, callback: msRest.ServiceCallback<Models.QueryPerformanceInsightResetDataResult>): void;
-  /**
-   * @param resourceGroupName The name of the resource group. The name is case insensitive.
-   * @param serverName The name of the server.
-   * @param options The optional parameters
-   * @param callback The callback
-   */
-  resetQueryPerformanceInsightData(resourceGroupName: string, serverName: string, options: msRest.RequestOptionsBase, callback: msRest.ServiceCallback<Models.QueryPerformanceInsightResetDataResult>): void;
-  resetQueryPerformanceInsightData(resourceGroupName: string, serverName: string, options?: msRest.RequestOptionsBase | msRest.ServiceCallback<Models.QueryPerformanceInsightResetDataResult>, callback?: msRest.ServiceCallback<Models.QueryPerformanceInsightResetDataResult>): Promise<Models.ResetQueryPerformanceInsightDataResponse> {
+  resetQueryPerformanceInsightData(
+    resourceGroupName: string,
+    serverName: string,
+    options?: ResetQueryPerformanceInsightDataOptionalParams
+  ): Promise<ResetQueryPerformanceInsightDataResponse> {
     return this.sendOperationRequest(
-      {
-        resourceGroupName,
-        serverName,
-        options
-      },
-      resetQueryPerformanceInsightDataOperationSpec,
-      callback) as Promise<Models.ResetQueryPerformanceInsightDataResponse>;
+      { resourceGroupName, serverName, options },
+      resetQueryPerformanceInsightDataOperationSpec
+    );
   }
 
   /**
@@ -123,12 +185,63 @@ class MySQLManagementClient extends MySQLManagementClientContext {
    * @param serverName The name of the server.
    * @param advisorName The advisor name for recommendation action.
    * @param databaseName The name of the database.
-   * @param [options] The optional parameters
-   * @returns Promise<msRest.RestResponse>
+   * @param options The options parameters.
    */
-  createRecommendedActionSession(resourceGroupName: string, serverName: string, advisorName: string, databaseName: string, options?: msRest.RequestOptionsBase): Promise<msRest.RestResponse> {
-    return this.beginCreateRecommendedActionSession(resourceGroupName,serverName,advisorName,databaseName,options)
-      .then(lroPoller => lroPoller.pollUntilFinished());
+  async beginCreateRecommendedActionSession(
+    resourceGroupName: string,
+    serverName: string,
+    advisorName: string,
+    databaseName: string,
+    options?: CreateRecommendedActionSessionOptionalParams
+  ): Promise<PollerLike<PollOperationState<void>, void>> {
+    const directSendOperation = async (
+      args: coreClient.OperationArguments,
+      spec: coreClient.OperationSpec
+    ): Promise<void> => {
+      return this.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, serverName, advisorName, databaseName, options },
+      createRecommendedActionSessionOperationSpec
+    );
+    return new LroEngine(lro, {
+      resumeFrom: options?.resumeFrom,
+      intervalInMs: options?.updateIntervalInMs
+    });
   }
 
   /**
@@ -137,39 +250,58 @@ class MySQLManagementClient extends MySQLManagementClientContext {
    * @param serverName The name of the server.
    * @param advisorName The advisor name for recommendation action.
    * @param databaseName The name of the database.
-   * @param [options] The optional parameters
-   * @returns Promise<msRestAzure.LROPoller>
+   * @param options The options parameters.
    */
-  beginCreateRecommendedActionSession(resourceGroupName: string, serverName: string, advisorName: string, databaseName: string, options?: msRest.RequestOptionsBase): Promise<msRestAzure.LROPoller> {
-    return this.sendLRORequest(
-      {
-        resourceGroupName,
-        serverName,
-        advisorName,
-        databaseName,
-        options
-      },
-      beginCreateRecommendedActionSessionOperationSpec,
-      options);
+  async beginCreateRecommendedActionSessionAndWait(
+    resourceGroupName: string,
+    serverName: string,
+    advisorName: string,
+    databaseName: string,
+    options?: CreateRecommendedActionSessionOptionalParams
+  ): Promise<void> {
+    const poller = await this.beginCreateRecommendedActionSession(
+      resourceGroupName,
+      serverName,
+      advisorName,
+      databaseName,
+      options
+    );
+    return poller.pollUntilDone();
   }
-}
 
+  servers: Servers;
+  replicas: Replicas;
+  firewallRules: FirewallRules;
+  virtualNetworkRules: VirtualNetworkRules;
+  databases: Databases;
+  configurations: Configurations;
+  serverParameters: ServerParameters;
+  logFiles: LogFiles;
+  serverAdministrators: ServerAdministrators;
+  recoverableServers: RecoverableServers;
+  serverBasedPerformanceTier: ServerBasedPerformanceTier;
+  locationBasedPerformanceTier: LocationBasedPerformanceTier;
+  checkNameAvailability: CheckNameAvailability;
+  operations: Operations;
+  serverSecurityAlertPolicies: ServerSecurityAlertPolicies;
+  queryTexts: QueryTexts;
+  topQueryStatistics: TopQueryStatistics;
+  waitStatistics: WaitStatistics;
+  advisors: Advisors;
+  recommendedActions: RecommendedActions;
+  locationBasedRecommendedActionSessionsOperationStatus: LocationBasedRecommendedActionSessionsOperationStatus;
+  locationBasedRecommendedActionSessionsResult: LocationBasedRecommendedActionSessionsResult;
+  privateEndpointConnections: PrivateEndpointConnections;
+  privateLinkResources: PrivateLinkResources;
+  serverKeys: ServerKeys;
+}
 // Operation Specifications
-const serializer = new msRest.Serializer(Mappers);
-const resetQueryPerformanceInsightDataOperationSpec: msRest.OperationSpec = {
+const serializer = coreClient.createSerializer(Mappers, /* isXml */ false);
+
+const resetQueryPerformanceInsightDataOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DBforMySQL/servers/{serverName}/resetQueryPerformanceInsightData",
   httpMethod: "POST",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DBforMySQL/servers/{serverName}/resetQueryPerformanceInsightData",
-  urlParameters: [
-    Parameters.subscriptionId,
-    Parameters.resourceGroupName,
-    Parameters.serverName
-  ],
-  queryParameters: [
-    Parameters.apiVersion2
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
   responses: {
     200: {
       bodyMapper: Mappers.QueryPerformanceInsightResetDataResult
@@ -178,39 +310,28 @@ const resetQueryPerformanceInsightDataOperationSpec: msRest.OperationSpec = {
       bodyMapper: Mappers.CloudError
     }
   },
+  queryParameters: [Parameters.apiVersion2],
+  urlParameters: [
+    Parameters.$host,
+    Parameters.subscriptionId,
+    Parameters.resourceGroupName,
+    Parameters.serverName
+  ],
+  headerParameters: [Parameters.accept],
   serializer
 };
-
-const beginCreateRecommendedActionSessionOperationSpec: msRest.OperationSpec = {
+const createRecommendedActionSessionOperationSpec: coreClient.OperationSpec = {
+  path:
+    "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DBforMySQL/servers/{serverName}/advisors/{advisorName}/createRecommendedActionSession",
   httpMethod: "POST",
-  path: "subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.DBforMySQL/servers/{serverName}/advisors/{advisorName}/createRecommendedActionSession",
+  responses: { 200: {}, 201: {}, 202: {}, 204: {} },
+  queryParameters: [Parameters.apiVersion2, Parameters.databaseName1],
   urlParameters: [
+    Parameters.$host,
     Parameters.subscriptionId,
     Parameters.resourceGroupName,
     Parameters.serverName,
     Parameters.advisorName
   ],
-  queryParameters: [
-    Parameters.apiVersion2,
-    Parameters.databaseName
-  ],
-  headerParameters: [
-    Parameters.acceptLanguage
-  ],
-  responses: {
-    200: {},
-    202: {},
-    default: {
-      bodyMapper: Mappers.CloudError
-    }
-  },
   serializer
 };
-
-export {
-  MySQLManagementClient,
-  MySQLManagementClientContext,
-  Models as MySQLManagementModels,
-  Mappers as MySQLManagementMappers
-};
-export * from "./operations";