diff --git a/extensions/sql-migration/src/models/stateMachine.ts b/extensions/sql-migration/src/models/stateMachine.ts index 6246f5dc27..ca1b9df1fd 100644 --- a/extensions/sql-migration/src/models/stateMachine.ts +++ b/extensions/sql-migration/src/models/stateMachine.ts @@ -185,12 +185,22 @@ export class MigrationStateModel implements Model, vscode.Disposable { return finalResult; } - public async getDatabaseAssessments(): Promise { + public async getDatabaseAssessments(targetType: MigrationTargetType): Promise { const ownerUri = await azdata.connection.getUriForConnection(this.sourceConnectionId); try { - this._assessmentApiResponse = (await this.migrationService.getAssessments(ownerUri, this._databaseAssessment))!; + const response = (await this.migrationService.getAssessments(ownerUri, this._databaseAssessment))!; + if (response?.assessmentResult) { + response.assessmentResult.items = response.assessmentResult.items?.filter( + issue => issue.appliesToMigrationTargetPlatform === targetType); + + response.assessmentResult.databases?.forEach( + database => database.items = database.items?.filter( + issue => issue.appliesToMigrationTargetPlatform === targetType)); + } + + this._assessmentApiResponse = response; this._assessmentResults = { - issues: this._assessmentApiResponse?.assessmentResult?.items ?? [], + issues: this._assessmentApiResponse?.assessmentResult?.items || [], databaseAssessments: this._assessmentApiResponse?.assessmentResult?.databases?.map(d => { return { name: d.name, diff --git a/extensions/sql-migration/src/wizard/skuRecommendationPage.ts b/extensions/sql-migration/src/wizard/skuRecommendationPage.ts index c9063e1a09..848cf46a71 100644 --- a/extensions/sql-migration/src/wizard/skuRecommendationPage.ts +++ b/extensions/sql-migration/src/wizard/skuRecommendationPage.ts @@ -444,7 +444,7 @@ export class SKURecommendationPage extends MigrationWizardPage { const serverName = (await this.migrationStateModel.getSourceConnectionProfile()).serverName; this._igComponent.value = constants.ASSESSMENT_COMPLETED(serverName); try { - await this.migrationStateModel.getDatabaseAssessments(); + await this.migrationStateModel.getDatabaseAssessments(MigrationTargetType.SQLMI); this._detailsComponent.value = constants.SKU_RECOMMENDATION_ALL_SUCCESSFUL(this.migrationStateModel._assessmentResults.databaseAssessments.length); const errors: string[] = [];