diff --git a/src/sql/workbench/common/actions.ts b/src/sql/workbench/common/actions.ts index 0f7b646150..797aa9fd7d 100644 --- a/src/sql/workbench/common/actions.ts +++ b/src/sql/workbench/common/actions.ts @@ -26,6 +26,7 @@ import { IEditorService } from 'vs/workbench/services/editor/common/editorServic import { ServicesAccessor } from 'vs/platform/instantiation/common/instantiation'; import { INotificationService } from 'vs/platform/notification/common/notification'; import { IConfigurationService } from 'vs/platform/configuration/common/configuration'; +import { mssqlProviderName } from 'sql/platform/connection/common/constants'; export interface BaseActionContext { object?: ObjectMetadata; @@ -253,7 +254,7 @@ export class BackupAction extends Task { } if (profile) { const serverInfo = connectionManagementService.getServerInfo(profile.id); - if (serverInfo && serverInfo.isCloud) { + if (serverInfo && serverInfo.isCloud && profile.providerName === mssqlProviderName) { return accessor.get(INotificationService).info(nls.localize('backup.commandNotSupported', 'Backup command is not supported for Azure SQL databases.')); } } @@ -296,7 +297,7 @@ export class RestoreAction extends Task { } if (profile) { const serverInfo = connectionManagementService.getServerInfo(profile.id); - if (serverInfo && serverInfo.isCloud) { + if (serverInfo && serverInfo.isCloud && profile.providerName === mssqlProviderName) { return accessor.get(INotificationService).info(nls.localize('restore.commandNotSupported', 'Restore command is not supported for Azure SQL databases.')); } }