diff --git a/extensions/sql-migration/src/api/azure.ts b/extensions/sql-migration/src/api/azure.ts index c23683bd7c..4edf1a13a1 100644 --- a/extensions/sql-migration/src/api/azure.ts +++ b/extensions/sql-migration/src/api/azure.ts @@ -200,11 +200,13 @@ export async function createSqlMigrationService(account: azdata.Account, subscri if (response.errors.length > 0) { throw new Error(response.errors.toString()); } - const asyncUrl = response.response.headers['azure-asyncoperation']; + const asyncUrl = response.response.headers['azure-asyncoperation'] + .replace('https://management.azure.com/', ''); + const maxRetry = 24; let i = 0; for (i = 0; i < maxRetry; i++) { - const asyncResponse = await api.makeAzureRestRequest(account, subscription, asyncUrl.replace('https://management.azure.com/', ''), azurecore.HttpRequestMethod.GET, undefined, true); + const asyncResponse = await api.makeAzureRestRequest(account, subscription, asyncUrl, azurecore.HttpRequestMethod.GET, undefined, true); const creationStatus = asyncResponse.response.data.status; if (creationStatus === ProvisioningState.Succeeded) { break; diff --git a/extensions/sql-migration/src/dialog/createSqlMigrationService/createSqlMigrationServiceDialog.ts b/extensions/sql-migration/src/dialog/createSqlMigrationService/createSqlMigrationServiceDialog.ts index e4cea41df8..06cc8323f4 100644 --- a/extensions/sql-migration/src/dialog/createSqlMigrationService/createSqlMigrationServiceDialog.ts +++ b/extensions/sql-migration/src/dialog/createSqlMigrationService/createSqlMigrationServiceDialog.ts @@ -5,7 +5,7 @@ import * as azdata from 'azdata'; import * as vscode from 'vscode'; -import { createSqlMigrationService, getSqlMigrationService, getSqlMigrationServiceAuthKeys, getSqlMigrationServiceMonitoringData, SqlMigrationService } from '../../api/azure'; +import { createSqlMigrationService, getResourceName, getSqlMigrationService, getSqlMigrationServiceAuthKeys, getSqlMigrationServiceMonitoringData, SqlMigrationService } from '../../api/azure'; import { MigrationStateModel, NetworkContainerType } from '../../models/stateMachine'; import { logError, TelemetryViews } from '../../telemtery'; import * as constants from '../../constants/strings'; @@ -499,7 +499,8 @@ export class CreateSqlMigrationServiceDialog { private async refreshStatus(): Promise { const subscription = this._model._targetSubscription; - const resourceGroup = (this.migrationServiceResourceGroupDropdown.value as azdata.CategoryValue).name; + const resourceGroupId = (this.migrationServiceResourceGroupDropdown.value as azdata.CategoryValue).name; + const resourceGroup = getResourceName(resourceGroupId); const location = this._model._targetServerInstance.location; const maxRetries = 5; @@ -562,7 +563,8 @@ export class CreateSqlMigrationServiceDialog { } private async refreshAuthTable(): Promise { const subscription = this._model._targetSubscription; - const resourceGroup = (this.migrationServiceResourceGroupDropdown.value as azdata.CategoryValue).name; + const resourceGroupId = (this.migrationServiceResourceGroupDropdown.value as azdata.CategoryValue).name; + const resourceGroup = getResourceName(resourceGroupId); const location = this._model._targetServerInstance.location; const keys = await getSqlMigrationServiceAuthKeys( this._model._azureAccount,