mirror of
https://github.com/ckaczor/azuredatastudio.git
synced 2026-01-20 01:25:37 -05:00
Add SQL DB offline migration wizard experience (#20403)
* sql db wizard with target selection * add database table selection * add sqldb to service and IR page * Code complete * navigation bug fixes * fix target db selection * improve sqldb error and status reporting * fix error count bug * remove table status inference * address review feedback * update resource strings and content * fix migraton status string, use localized value * fix ux navigation issues * fix back/fwd w/o changes from changing data
This commit is contained in:
brian-harris
282
extensions/sql-migration/src/api/sqlUtils.ts
Normal file
282
extensions/sql-migration/src/api/sqlUtils.ts
Normal file
@@ -0,0 +1,282 @@
|
||||
/*---------------------------------------------------------------------------------------------
|
||||
* Copyright (c) Microsoft Corporation. All rights reserved.
|
||||
* Licensed under the Source EULA. See License.txt in the project root for license information.
|
||||
*--------------------------------------------------------------------------------------------*/
|
||||
|
||||
import * as azdata from 'azdata';
|
||||
import { azureResource } from 'azurecore';
|
||||
import { AzureSqlDatabase, AzureSqlDatabaseServer } from './azure';
|
||||
import { generateGuid } from './utils';
|
||||
import * as utils from '../api/utils';
|
||||
|
||||
const query_database_tables_sql = `
|
||||
SELECT
|
||||
DB_NAME() as database_name,
|
||||
QUOTENAME(SCHEMA_NAME(o.schema_id)) + '.' + QUOTENAME(o.name) AS table_name,
|
||||
SUM(p.Rows) AS row_count
|
||||
FROM
|
||||
sys.objects AS o
|
||||
INNER JOIN sys.partitions AS p
|
||||
ON o.object_id = p.object_id
|
||||
WHERE
|
||||
o.type = 'U'
|
||||
AND o.is_ms_shipped = 0x0
|
||||
AND index_id < 2
|
||||
GROUP BY
|
||||
o.schema_id,
|
||||
o.name
|
||||
ORDER BY table_name;`;
|
||||
|
||||
const query_target_databases_sql = `
|
||||
SELECT
|
||||
('servername') as server_name,
|
||||
SERVERPROPERTY ('collation') as server_collation,
|
||||
db.database_id as database_id,
|
||||
db.name as database_name,
|
||||
db.collation_name as database_collation,
|
||||
CASE WHEN 'A' = 'a' THEN 0 ELSE 1 END as is_server_case_sensitive,
|
||||
db.state as database_state,
|
||||
db.is_read_only as is_read_only
|
||||
FROM sys.databases db
|
||||
WHERE
|
||||
db.name not in ('master', 'tempdb', 'model', 'msdb')
|
||||
AND is_distributor <> 1
|
||||
ORDER BY db.name;`;
|
||||
|
||||
export const excludeDatabses: string[] = [
|
||||
'master',
|
||||
'tempdb',
|
||||
'msdb',
|
||||
'model'
|
||||
];
|
||||
|
||||
export enum AuthenticationType {
|
||||
Integrated = 'Integrated',
|
||||
SqlLogin = 'SqlLogin'
|
||||
}
|
||||
|
||||
export interface TableInfo {
|
||||
databaseName: string;
|
||||
tableName: string;
|
||||
rowCount: number;
|
||||
selectedForMigration: boolean;
|
||||
}
|
||||
|
||||
export interface TargetDatabaseInfo {
|
||||
serverName: string;
|
||||
serverCollation: string;
|
||||
databaseId: string;
|
||||
databaseName: string;
|
||||
databaseCollation: string;
|
||||
isServerCaseSensitive: boolean;
|
||||
databaseState: number;
|
||||
isReadOnly: boolean;
|
||||
sourceTables: Map<string, TableInfo>;
|
||||
targetTables: Map<string, TableInfo>;
|
||||
}
|
||||
|
||||
function getSqlDbConnectionProfile(
|
||||
serverName: string,
|
||||
tenantId: string,
|
||||
databaseName: string,
|
||||
userName: string,
|
||||
password: string): azdata.IConnectionProfile {
|
||||
return {
|
||||
id: generateGuid(),
|
||||
providerName: 'MSSQL',
|
||||
connectionName: '',
|
||||
serverName: serverName,
|
||||
databaseName: databaseName,
|
||||
userName: userName,
|
||||
password: password,
|
||||
authenticationType: AuthenticationType.SqlLogin,
|
||||
savePassword: false,
|
||||
saveProfile: false,
|
||||
options: {
|
||||
conectionName: '',
|
||||
server: serverName,
|
||||
database: databaseName,
|
||||
authenticationType: AuthenticationType.SqlLogin,
|
||||
user: userName,
|
||||
password: password,
|
||||
connectionTimeout: 60,
|
||||
columnEncryptionSetting: 'Enabled',
|
||||
encrypt: true,
|
||||
trustServerCertificate: false,
|
||||
connectRetryCount: '1',
|
||||
connectRetryInterval: '10',
|
||||
applicationName: 'azdata',
|
||||
azureTenantId: tenantId,
|
||||
originalDatabase: databaseName,
|
||||
databaseDisplayName: databaseName,
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
function getConnectionProfile(
|
||||
serverName: string,
|
||||
azureResourceId: string,
|
||||
userName: string,
|
||||
password: string): azdata.IConnectionProfile {
|
||||
return {
|
||||
serverName: serverName,
|
||||
id: generateGuid(),
|
||||
connectionName: undefined,
|
||||
azureResourceId: azureResourceId,
|
||||
userName: userName,
|
||||
password: password,
|
||||
authenticationType: AuthenticationType.SqlLogin,
|
||||
savePassword: false,
|
||||
groupFullName: '',
|
||||
groupId: '',
|
||||
providerName: 'MSSQL',
|
||||
saveProfile: false,
|
||||
options: {
|
||||
conectionName: '',
|
||||
server: serverName,
|
||||
authenticationType: AuthenticationType.SqlLogin,
|
||||
user: userName,
|
||||
password: password,
|
||||
connectionTimeout: 60,
|
||||
columnEncryptionSetting: 'Enabled',
|
||||
encrypt: true,
|
||||
trustServerCertificate: false,
|
||||
connectRetryCount: '1',
|
||||
connectRetryInterval: '10',
|
||||
applicationName: 'azdata',
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
export async function collectSourceDatabaseTableInfo(sourceConnectionId: string, sourceDatabase: string): Promise<TableInfo[]> {
|
||||
const ownerUri = await azdata.connection.getUriForConnection(sourceConnectionId);
|
||||
const connectionProvider = azdata.dataprotocol.getProvider<azdata.ConnectionProvider>(
|
||||
'MSSQL',
|
||||
azdata.DataProviderType.ConnectionProvider);
|
||||
await connectionProvider.changeDatabase(ownerUri, sourceDatabase);
|
||||
const queryProvider = azdata.dataprotocol.getProvider<azdata.QueryProvider>(
|
||||
'MSSQL',
|
||||
azdata.DataProviderType.QueryProvider);
|
||||
|
||||
const results = await queryProvider.runQueryAndReturn(
|
||||
ownerUri,
|
||||
query_database_tables_sql);
|
||||
|
||||
return results.rows.map(row => {
|
||||
return {
|
||||
databaseName: getSqlString(row[0]),
|
||||
tableName: getSqlString(row[1]),
|
||||
rowCount: getSqlNumber(row[2]),
|
||||
selectedForMigration: false,
|
||||
};
|
||||
}) ?? [];
|
||||
}
|
||||
|
||||
export async function collectTargetDatabaseTableInfo(
|
||||
targetServer: AzureSqlDatabaseServer,
|
||||
targetDatabaseName: string,
|
||||
tenantId: string,
|
||||
userName: string,
|
||||
password: string): Promise<TableInfo[]> {
|
||||
const connectionProfile = getSqlDbConnectionProfile(
|
||||
targetServer.properties.fullyQualifiedDomainName,
|
||||
tenantId,
|
||||
targetDatabaseName,
|
||||
userName,
|
||||
password);
|
||||
|
||||
const result = await azdata.connection.connect(connectionProfile, false, false);
|
||||
if (result.connected && result.connectionId) {
|
||||
const queryProvider = azdata.dataprotocol.getProvider<azdata.QueryProvider>(
|
||||
'MSSQL',
|
||||
azdata.DataProviderType.QueryProvider);
|
||||
|
||||
const ownerUri = await azdata.connection.getUriForConnection(result.connectionId);
|
||||
const results = await queryProvider.runQueryAndReturn(
|
||||
ownerUri,
|
||||
query_database_tables_sql);
|
||||
|
||||
return results.rows.map(row => {
|
||||
return {
|
||||
databaseName: getSqlString(row[0]),
|
||||
tableName: getSqlString(row[1]),
|
||||
rowCount: getSqlNumber(row[2]),
|
||||
selectedForMigration: false,
|
||||
};
|
||||
}) ?? [];
|
||||
}
|
||||
|
||||
throw new Error(result.errorMessage);
|
||||
}
|
||||
|
||||
export async function collectTargetDatabaseInfo(
|
||||
targetServer: AzureSqlDatabaseServer,
|
||||
userName: string,
|
||||
password: string): Promise<TargetDatabaseInfo[]> {
|
||||
|
||||
const connectionProfile = getConnectionProfile(
|
||||
targetServer.properties.fullyQualifiedDomainName,
|
||||
targetServer.id,
|
||||
userName,
|
||||
password);
|
||||
|
||||
const result = await azdata.connection.connect(connectionProfile, false, false);
|
||||
if (result.connected && result.connectionId) {
|
||||
const queryProvider = azdata.dataprotocol.getProvider<azdata.QueryProvider>(
|
||||
'MSSQL',
|
||||
azdata.DataProviderType.QueryProvider);
|
||||
|
||||
const ownerUri = await azdata.connection.getUriForConnection(result.connectionId);
|
||||
const results = await queryProvider.runQueryAndReturn(
|
||||
ownerUri,
|
||||
query_target_databases_sql);
|
||||
|
||||
return results.rows.map(row => {
|
||||
return {
|
||||
serverName: getSqlString(row[0]),
|
||||
serverCollation: getSqlString(row[1]),
|
||||
databaseId: getSqlString(row[2]),
|
||||
databaseName: getSqlString(row[3]),
|
||||
databaseCollation: getSqlString(row[4]),
|
||||
isServerCaseSensitive: getSqlBoolean(row[5]),
|
||||
databaseState: getSqlNumber(row[6]),
|
||||
isReadOnly: getSqlBoolean(row[7]),
|
||||
sourceTables: new Map(),
|
||||
targetTables: new Map(),
|
||||
};
|
||||
}) ?? [];
|
||||
}
|
||||
|
||||
throw new Error(result.errorMessage);
|
||||
}
|
||||
|
||||
export async function collectAzureTargetDatabases(
|
||||
account: azdata.Account,
|
||||
subscription: azureResource.AzureResourceSubscription,
|
||||
resourceGroup: string,
|
||||
targetServerName: string,
|
||||
): Promise<AzureSqlDatabase[]> {
|
||||
const databaseList: AzureSqlDatabase[] = [];
|
||||
if (resourceGroup && targetServerName) {
|
||||
databaseList.push(...
|
||||
await utils.getAzureSqlDatabases(
|
||||
account,
|
||||
subscription,
|
||||
resourceGroup,
|
||||
targetServerName));
|
||||
}
|
||||
return databaseList.filter(
|
||||
database => !excludeDatabses.includes(database.name)) ?? [];
|
||||
}
|
||||
|
||||
export function getSqlString(value: azdata.DbCellValue): string {
|
||||
return value.isNull ? '' : value.displayValue;
|
||||
}
|
||||
|
||||
export function getSqlNumber(value: azdata.DbCellValue): number {
|
||||
return value.isNull ? 0 : parseInt(value.displayValue);
|
||||
}
|
||||
|
||||
export function getSqlBoolean(value: azdata.DbCellValue): boolean {
|
||||
return value.isNull ? false : value.displayValue === '1';
|
||||
}
|
||||
Reference in New Issue
Block a user