/* * 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, PageSettings } from "@azure/core-paging"; import { setContinuationToken } from "../pagingHelper"; import { Servers } from "../operationsInterfaces"; import * as coreClient from "@azure/core-client"; import * as Mappers from "../models/mappers"; import * as Parameters from "../models/parameters"; import { SqlManagementClient } from "../sqlManagementClient"; import { SimplePollerLike, OperationState, createHttpPoller } from "@azure/core-lro"; import { createLroSpec } from "../lroImpl"; import { Server, ServersListNextOptionalParams, ServersListOptionalParams, ServersListResponse, ServersListByResourceGroupNextOptionalParams, ServersListByResourceGroupOptionalParams, ServersListByResourceGroupResponse, CheckNameAvailabilityRequest, ServersCheckNameAvailabilityOptionalParams, ServersCheckNameAvailabilityResponse, ServersGetOptionalParams, ServersGetResponse, ServersCreateOrUpdateOptionalParams, ServersCreateOrUpdateResponse, ServersDeleteOptionalParams, ServerUpdate, ServersUpdateOptionalParams, ServersUpdateResponse, ImportNewDatabaseDefinition, ServersImportDatabaseOptionalParams, ServersImportDatabaseResponse, ServersRefreshStatusOptionalParams, ServersRefreshStatusResponse, ServersListNextResponse, ServersListByResourceGroupNextResponse } from "../models"; /// /** Class containing Servers operations. */ export class ServersImpl implements Servers { private readonly client: SqlManagementClient; /** * Initialize a new instance of the class Servers class. * @param client Reference to the service client */ constructor(client: SqlManagementClient) { this.client = client; } /** * Gets a list of all servers in the subscription. * @param options The options parameters. */ public list( options?: ServersListOptionalParams ): PagedAsyncIterableIterator { const iter = this.listPagingAll(options); return { next() { return iter.next(); }, [Symbol.asyncIterator]() { return this; }, byPage: (settings?: PageSettings) => { if (settings?.maxPageSize) { throw new Error("maxPageSize is not supported by this operation."); } return this.listPagingPage(options, settings); } }; } private async *listPagingPage( options?: ServersListOptionalParams, settings?: PageSettings ): AsyncIterableIterator { let result: ServersListResponse; let continuationToken = settings?.continuationToken; if (!continuationToken) { result = await this._list(options); let page = result.value || []; continuationToken = result.nextLink; setContinuationToken(page, continuationToken); yield page; } while (continuationToken) { result = await this._listNext(continuationToken, options); continuationToken = result.nextLink; let page = result.value || []; setContinuationToken(page, continuationToken); yield page; } } private async *listPagingAll( options?: ServersListOptionalParams ): AsyncIterableIterator { for await (const page of this.listPagingPage(options)) { yield* page; } } /** * Gets a list of servers in a resource groups. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param options The options parameters. */ public listByResourceGroup( resourceGroupName: string, options?: ServersListByResourceGroupOptionalParams ): PagedAsyncIterableIterator { const iter = this.listByResourceGroupPagingAll(resourceGroupName, options); return { next() { return iter.next(); }, [Symbol.asyncIterator]() { return this; }, 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?: ServersListByResourceGroupOptionalParams, settings?: PageSettings ): AsyncIterableIterator { let result: ServersListByResourceGroupResponse; 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, continuationToken, options ); continuationToken = result.nextLink; let page = result.value || []; setContinuationToken(page, continuationToken); yield page; } } private async *listByResourceGroupPagingAll( resourceGroupName: string, options?: ServersListByResourceGroupOptionalParams ): AsyncIterableIterator { for await (const page of this.listByResourceGroupPagingPage( resourceGroupName, options )) { yield* page; } } /** * Determines whether a resource can be created with the specified name. * @param parameters The name availability request parameters. * @param options The options parameters. */ checkNameAvailability( parameters: CheckNameAvailabilityRequest, options?: ServersCheckNameAvailabilityOptionalParams ): Promise { return this.client.sendOperationRequest( { parameters, options }, checkNameAvailabilityOperationSpec ); } /** * Gets a list of all servers in the subscription. * @param options The options parameters. */ private _list( options?: ServersListOptionalParams ): Promise { return this.client.sendOperationRequest({ options }, listOperationSpec); } /** * Gets a list of servers in a resource groups. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param options The options parameters. */ private _listByResourceGroup( resourceGroupName: string, options?: ServersListByResourceGroupOptionalParams ): Promise { return this.client.sendOperationRequest( { resourceGroupName, options }, listByResourceGroupOperationSpec ); } /** * Gets a server. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param serverName The name of the server. * @param options The options parameters. */ get( resourceGroupName: string, serverName: string, options?: ServersGetOptionalParams ): Promise { return this.client.sendOperationRequest( { resourceGroupName, serverName, options }, getOperationSpec ); } /** * Creates or updates a server. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param serverName The name of the server. * @param parameters The requested server resource state. * @param options The options parameters. */ async beginCreateOrUpdate( resourceGroupName: string, serverName: string, parameters: Server, options?: ServersCreateOrUpdateOptionalParams ): Promise< SimplePollerLike< OperationState, ServersCreateOrUpdateResponse > > { const directSendOperation = async ( args: coreClient.OperationArguments, spec: coreClient.OperationSpec ): Promise => { return this.client.sendOperationRequest(args, spec); }; const sendOperationFn = 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 = createLroSpec({ sendOperationFn, args: { resourceGroupName, serverName, parameters, options }, spec: createOrUpdateOperationSpec }); const poller = await createHttpPoller< ServersCreateOrUpdateResponse, OperationState >(lro, { restoreFrom: options?.resumeFrom, intervalInMs: options?.updateIntervalInMs }); await poller.poll(); return poller; } /** * Creates or updates a server. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param serverName The name of the server. * @param parameters The requested server resource state. * @param options The options parameters. */ async beginCreateOrUpdateAndWait( resourceGroupName: string, serverName: string, parameters: Server, options?: ServersCreateOrUpdateOptionalParams ): Promise { const poller = await this.beginCreateOrUpdate( resourceGroupName, serverName, parameters, options ); return poller.pollUntilDone(); } /** * Deletes a server. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param serverName The name of the server. * @param options The options parameters. */ async beginDelete( resourceGroupName: string, serverName: string, options?: ServersDeleteOptionalParams ): Promise, void>> { const directSendOperation = async ( args: coreClient.OperationArguments, spec: coreClient.OperationSpec ): Promise => { return this.client.sendOperationRequest(args, spec); }; const sendOperationFn = 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 = createLroSpec({ sendOperationFn, args: { resourceGroupName, serverName, options }, spec: deleteOperationSpec }); const poller = await createHttpPoller>(lro, { restoreFrom: options?.resumeFrom, intervalInMs: options?.updateIntervalInMs }); await poller.poll(); return poller; } /** * Deletes a server. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param serverName The name of the server. * @param options The options parameters. */ async beginDeleteAndWait( resourceGroupName: string, serverName: string, options?: ServersDeleteOptionalParams ): Promise { const poller = await this.beginDelete( resourceGroupName, serverName, options ); return poller.pollUntilDone(); } /** * Updates a server. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param serverName The name of the server. * @param parameters The requested server resource state. * @param options The options parameters. */ async beginUpdate( resourceGroupName: string, serverName: string, parameters: ServerUpdate, options?: ServersUpdateOptionalParams ): Promise< SimplePollerLike< OperationState, ServersUpdateResponse > > { const directSendOperation = async ( args: coreClient.OperationArguments, spec: coreClient.OperationSpec ): Promise => { return this.client.sendOperationRequest(args, spec); }; const sendOperationFn = 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 = createLroSpec({ sendOperationFn, args: { resourceGroupName, serverName, parameters, options }, spec: updateOperationSpec }); const poller = await createHttpPoller< ServersUpdateResponse, OperationState >(lro, { restoreFrom: options?.resumeFrom, intervalInMs: options?.updateIntervalInMs }); await poller.poll(); return poller; } /** * Updates a server. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param serverName The name of the server. * @param parameters The requested server resource state. * @param options The options parameters. */ async beginUpdateAndWait( resourceGroupName: string, serverName: string, parameters: ServerUpdate, options?: ServersUpdateOptionalParams ): Promise { const poller = await this.beginUpdate( resourceGroupName, serverName, parameters, options ); return poller.pollUntilDone(); } /** * Imports a bacpac into a new database. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param serverName The name of the server. * @param parameters The database import request parameters. * @param options The options parameters. */ async beginImportDatabase( resourceGroupName: string, serverName: string, parameters: ImportNewDatabaseDefinition, options?: ServersImportDatabaseOptionalParams ): Promise< SimplePollerLike< OperationState, ServersImportDatabaseResponse > > { const directSendOperation = async ( args: coreClient.OperationArguments, spec: coreClient.OperationSpec ): Promise => { return this.client.sendOperationRequest(args, spec); }; const sendOperationFn = 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 = createLroSpec({ sendOperationFn, args: { resourceGroupName, serverName, parameters, options }, spec: importDatabaseOperationSpec }); const poller = await createHttpPoller< ServersImportDatabaseResponse, OperationState >(lro, { restoreFrom: options?.resumeFrom, intervalInMs: options?.updateIntervalInMs }); await poller.poll(); return poller; } /** * Imports a bacpac into a new database. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param serverName The name of the server. * @param parameters The database import request parameters. * @param options The options parameters. */ async beginImportDatabaseAndWait( resourceGroupName: string, serverName: string, parameters: ImportNewDatabaseDefinition, options?: ServersImportDatabaseOptionalParams ): Promise { const poller = await this.beginImportDatabase( resourceGroupName, serverName, parameters, options ); return poller.pollUntilDone(); } /** * Refresh external governance enablement status. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param serverName The name of the server. * @param options The options parameters. */ async beginRefreshStatus( resourceGroupName: string, serverName: string, options?: ServersRefreshStatusOptionalParams ): Promise< SimplePollerLike< OperationState, ServersRefreshStatusResponse > > { const directSendOperation = async ( args: coreClient.OperationArguments, spec: coreClient.OperationSpec ): Promise => { return this.client.sendOperationRequest(args, spec); }; const sendOperationFn = 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 = createLroSpec({ sendOperationFn, args: { resourceGroupName, serverName, options }, spec: refreshStatusOperationSpec }); const poller = await createHttpPoller< ServersRefreshStatusResponse, OperationState >(lro, { restoreFrom: options?.resumeFrom, intervalInMs: options?.updateIntervalInMs, resourceLocationConfig: "location" }); await poller.poll(); return poller; } /** * Refresh external governance enablement status. * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param serverName The name of the server. * @param options The options parameters. */ async beginRefreshStatusAndWait( resourceGroupName: string, serverName: string, options?: ServersRefreshStatusOptionalParams ): Promise { const poller = await this.beginRefreshStatus( resourceGroupName, serverName, options ); return poller.pollUntilDone(); } /** * ListNext * @param nextLink The nextLink from the previous successful call to the List method. * @param options The options parameters. */ private _listNext( nextLink: string, options?: ServersListNextOptionalParams ): Promise { return this.client.sendOperationRequest( { nextLink, options }, listNextOperationSpec ); } /** * ListByResourceGroupNext * @param resourceGroupName The name of the resource group that contains the resource. You can obtain * this value from the Azure Resource Manager API or the portal. * @param nextLink The nextLink from the previous successful call to the ListByResourceGroup method. * @param options The options parameters. */ private _listByResourceGroupNext( resourceGroupName: string, nextLink: string, options?: ServersListByResourceGroupNextOptionalParams ): Promise { return this.client.sendOperationRequest( { resourceGroupName, nextLink, options }, listByResourceGroupNextOperationSpec ); } } // Operation Specifications const serializer = coreClient.createSerializer(Mappers, /* isXml */ false); const checkNameAvailabilityOperationSpec: coreClient.OperationSpec = { path: "/subscriptions/{subscriptionId}/providers/Microsoft.Sql/checkNameAvailability", httpMethod: "POST", responses: { 200: { bodyMapper: Mappers.CheckNameAvailabilityResponse }, default: {} }, requestBody: Parameters.parameters100, queryParameters: [Parameters.apiVersion1], urlParameters: [Parameters.$host, Parameters.subscriptionId], headerParameters: [Parameters.contentType, Parameters.accept], mediaType: "json", serializer }; const listOperationSpec: coreClient.OperationSpec = { path: "/subscriptions/{subscriptionId}/providers/Microsoft.Sql/servers", httpMethod: "GET", responses: { 200: { bodyMapper: Mappers.ServerListResult }, default: {} }, queryParameters: [Parameters.apiVersion1, Parameters.expand], urlParameters: [Parameters.$host, Parameters.subscriptionId], headerParameters: [Parameters.accept], serializer }; const listByResourceGroupOperationSpec: coreClient.OperationSpec = { path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/servers", httpMethod: "GET", responses: { 200: { bodyMapper: Mappers.ServerListResult }, default: {} }, queryParameters: [Parameters.apiVersion1, Parameters.expand], urlParameters: [ Parameters.$host, Parameters.subscriptionId, Parameters.resourceGroupName ], headerParameters: [Parameters.accept], serializer }; const getOperationSpec: coreClient.OperationSpec = { path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/servers/{serverName}", httpMethod: "GET", responses: { 200: { bodyMapper: Mappers.Server }, default: {} }, queryParameters: [Parameters.apiVersion1, Parameters.expand], urlParameters: [ Parameters.$host, Parameters.subscriptionId, Parameters.resourceGroupName, Parameters.serverName ], headerParameters: [Parameters.accept], serializer }; const createOrUpdateOperationSpec: coreClient.OperationSpec = { path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/servers/{serverName}", httpMethod: "PUT", responses: { 200: { bodyMapper: Mappers.Server }, 201: { bodyMapper: Mappers.Server }, 202: { bodyMapper: Mappers.Server }, 204: { bodyMapper: Mappers.Server }, default: {} }, requestBody: Parameters.parameters101, queryParameters: [Parameters.apiVersion1], urlParameters: [ Parameters.$host, Parameters.subscriptionId, Parameters.resourceGroupName, Parameters.serverName ], headerParameters: [Parameters.contentType, Parameters.accept], mediaType: "json", serializer }; const deleteOperationSpec: coreClient.OperationSpec = { path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/servers/{serverName}", httpMethod: "DELETE", responses: { 200: {}, 201: {}, 202: {}, 204: {}, default: {} }, queryParameters: [Parameters.apiVersion1], urlParameters: [ Parameters.$host, Parameters.subscriptionId, Parameters.resourceGroupName, Parameters.serverName ], serializer }; const updateOperationSpec: coreClient.OperationSpec = { path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/servers/{serverName}", httpMethod: "PATCH", responses: { 200: { bodyMapper: Mappers.Server }, 201: { bodyMapper: Mappers.Server }, 202: { bodyMapper: Mappers.Server }, 204: { bodyMapper: Mappers.Server }, default: {} }, requestBody: Parameters.parameters102, queryParameters: [Parameters.apiVersion1], urlParameters: [ Parameters.$host, Parameters.subscriptionId, Parameters.resourceGroupName, Parameters.serverName ], headerParameters: [Parameters.contentType, Parameters.accept], mediaType: "json", serializer }; const importDatabaseOperationSpec: coreClient.OperationSpec = { path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/servers/{serverName}/import", httpMethod: "POST", responses: { 200: { bodyMapper: Mappers.ImportExportOperationResult }, 201: { bodyMapper: Mappers.ImportExportOperationResult }, 202: { bodyMapper: Mappers.ImportExportOperationResult }, 204: { bodyMapper: Mappers.ImportExportOperationResult }, default: {} }, requestBody: Parameters.parameters103, queryParameters: [Parameters.apiVersion1], urlParameters: [ Parameters.$host, Parameters.subscriptionId, Parameters.resourceGroupName, Parameters.serverName ], headerParameters: [Parameters.contentType, Parameters.accept], mediaType: "json", serializer }; const refreshStatusOperationSpec: coreClient.OperationSpec = { path: "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Sql/servers/{serverName}/refreshExternalGovernanceStatus", httpMethod: "POST", responses: { 200: { bodyMapper: Mappers.RefreshExternalGovernanceStatusOperationResult }, 201: { bodyMapper: Mappers.RefreshExternalGovernanceStatusOperationResult }, 202: { bodyMapper: Mappers.RefreshExternalGovernanceStatusOperationResult }, 204: { bodyMapper: Mappers.RefreshExternalGovernanceStatusOperationResult }, default: {} }, queryParameters: [Parameters.apiVersion1], urlParameters: [ Parameters.$host, Parameters.subscriptionId, Parameters.resourceGroupName, Parameters.serverName ], headerParameters: [Parameters.accept], serializer }; const listNextOperationSpec: coreClient.OperationSpec = { path: "{nextLink}", httpMethod: "GET", responses: { 200: { bodyMapper: Mappers.ServerListResult }, default: {} }, urlParameters: [ Parameters.$host, Parameters.subscriptionId, Parameters.nextLink ], headerParameters: [Parameters.accept], serializer }; const listByResourceGroupNextOperationSpec: coreClient.OperationSpec = { path: "{nextLink}", httpMethod: "GET", responses: { 200: { bodyMapper: Mappers.ServerListResult }, default: {} }, urlParameters: [ Parameters.$host, Parameters.subscriptionId, Parameters.resourceGroupName, Parameters.nextLink ], headerParameters: [Parameters.accept], serializer };