mirror of
https://github.com/ckaczor/azuredatastudio.git
synced 2026-02-16 18:46:40 -05:00
Remove extra id property from azure subs and add type name helper (#12145)
* Remove extra id property from azure subs and add type name helper * case * Change a few names
This commit is contained in:
@@ -7,6 +7,20 @@ declare module 'azureResource' {
|
||||
import { TreeDataProvider } from 'vscode';
|
||||
import { DataProvider, Account, TreeItem } from 'azdata';
|
||||
export namespace azureResource {
|
||||
|
||||
export const enum AzureResourceType {
|
||||
resourceGroup = 'microsoft.resources/subscriptions/resourcegroups',
|
||||
sqlServer = 'microsoft.sql/servers',
|
||||
sqlDatabase = 'microsoft.sql/servers/databases',
|
||||
sqlManagedInstance = 'microsoft.sql/managedinstances',
|
||||
azureArcSqlManagedInstance = 'microsoft.azuredata/sqlinstances',
|
||||
virtualMachines = 'microsoft.compute/virtualmachines',
|
||||
kustoClusters = 'microsoft.kusto/clusters',
|
||||
azureArcPostgresServer = 'microsoft.azuredata/postgresinstances',
|
||||
postgresServer = 'microsoft.dbforpostgresql/servers',
|
||||
azureArcService = 'microsoft.azuredata/datacontrollers'
|
||||
}
|
||||
|
||||
export interface IAzureResourceProvider extends DataProvider {
|
||||
getTreeDataProvider(): IAzureResourceTreeDataProvider;
|
||||
}
|
||||
@@ -28,7 +42,7 @@ declare module 'azureResource' {
|
||||
tenant?: string;
|
||||
}
|
||||
|
||||
export interface AzureResourceSubscription extends AzureResource {
|
||||
export interface AzureResourceSubscription extends Omit<AzureResource, 'subscriptionId'> {
|
||||
}
|
||||
|
||||
export interface AzureSqlResource extends AzureResource {
|
||||
|
||||
@@ -21,7 +21,7 @@ export class AzureResourceDatabaseService implements IAzureResourceService<azure
|
||||
|
||||
// Query servers and databases in parallel (start both promises before waiting on the 1st)
|
||||
let serverQueryPromise = queryGraphResources<GraphData>(resourceClient, subscriptions, serversQuery);
|
||||
let dbQueryPromise = queryGraphResources<GraphData>(resourceClient, subscriptions, 'where type == "microsoft.sql/servers/databases"');
|
||||
let dbQueryPromise = queryGraphResources<GraphData>(resourceClient, subscriptions, `where type == "${azureResource.AzureResourceType.sqlDatabase}"`);
|
||||
let servers: DbServerGraphData[] = await serverQueryPromise as DbServerGraphData[];
|
||||
let dbByGraph: DatabaseGraphData[] = await dbQueryPromise as DatabaseGraphData[];
|
||||
|
||||
|
||||
@@ -15,7 +15,7 @@ export interface DbServerGraphData extends GraphData {
|
||||
};
|
||||
}
|
||||
|
||||
export const serversQuery = 'where type == "microsoft.sql/servers"';
|
||||
export const serversQuery = `where type == "${azureResource.AzureResourceType.sqlServer}"`;
|
||||
|
||||
export class AzureResourceDatabaseServerService extends ResourceServiceBase<DbServerGraphData, azureResource.AzureResourceDatabaseServer> {
|
||||
|
||||
|
||||
@@ -14,7 +14,7 @@ export interface KustoGraphData extends GraphData {
|
||||
};
|
||||
}
|
||||
|
||||
const instanceQuery = 'where type == "microsoft.kusto/clusters"';
|
||||
const instanceQuery = `where type == "${azureResource.AzureResourceType.kustoClusters}"`;
|
||||
|
||||
export class KustoResourceService extends ResourceServiceBase<KustoGraphData, azureResource.AzureResourceDatabaseServer> {
|
||||
|
||||
|
||||
@@ -12,7 +12,7 @@ export interface PostgresArcServerGraphData extends GraphData {
|
||||
};
|
||||
}
|
||||
|
||||
export const serversQuery = 'where type == "microsoft.azuredata/postgresinstances"';
|
||||
export const serversQuery = `where type == "${azureResource.AzureResourceType.azureArcPostgresServer}"`;
|
||||
|
||||
export class PostgresServerArcService extends ResourceServiceBase<PostgresArcServerGraphData, azureResource.AzureResourceDatabaseServer> {
|
||||
|
||||
|
||||
@@ -15,7 +15,7 @@ interface DbServerGraphData extends GraphData {
|
||||
};
|
||||
}
|
||||
|
||||
const serversQuery = 'where type == "microsoft.dbforpostgresql/servers"';
|
||||
const serversQuery = `where type == "${azureResource.AzureResourceType.postgresServer}"`;
|
||||
|
||||
export class PostgresServerService extends ResourceServiceBase<DbServerGraphData, azureResource.AzureResourceDatabaseServer> {
|
||||
|
||||
|
||||
@@ -10,7 +10,7 @@ import { ResourceServiceBase } from '../resourceTreeDataProviderBase';
|
||||
export class AzureResourceGroupService extends ResourceServiceBase<DbServerGraphData, azureResource.AzureResourceResourceGroup> {
|
||||
|
||||
protected get query(): string {
|
||||
return 'ResourceContainers | where type=="microsoft.resources/subscriptions/resourcegroups"';
|
||||
return `ResourceContainers | where type=="${azureResource.AzureResourceType.resourceGroup}"`;
|
||||
}
|
||||
|
||||
protected convertResource(resource: DbServerGraphData): azureResource.AzureResourceResourceGroup {
|
||||
|
||||
@@ -13,7 +13,7 @@ interface SqlInstanceGraphData extends GraphData {
|
||||
};
|
||||
}
|
||||
|
||||
const instanceQuery = 'where type == "microsoft.sql/managedinstances"';
|
||||
const instanceQuery = `where type == "${azureResource.AzureResourceType.sqlManagedInstance}"`;
|
||||
|
||||
export class SqlInstanceResourceService extends ResourceServiceBase<SqlInstanceGraphData, azureResource.AzureResourceDatabaseServer> {
|
||||
|
||||
|
||||
@@ -13,7 +13,7 @@ export interface SqlInstanceArcGraphData extends GraphData {
|
||||
};
|
||||
}
|
||||
|
||||
const instanceQuery = 'where type == "microsoft.azuredata/sqlinstances"';
|
||||
const instanceQuery = `where type == "${azureResource.AzureResourceType.azureArcSqlManagedInstance}"`;
|
||||
export class SqlInstanceArcResourceService extends ResourceServiceBase<SqlInstanceArcGraphData, azureResource.AzureResourceDatabaseServer> {
|
||||
|
||||
protected get query(): string {
|
||||
|
||||
@@ -18,7 +18,6 @@ export class AzureResourceSubscriptionService implements IAzureResourceSubscript
|
||||
subs.forEach((sub) => subscriptions.push({
|
||||
id: sub.subscriptionId,
|
||||
name: sub.displayName,
|
||||
subscriptionId: sub.subscriptionId,
|
||||
tenant: tenantId
|
||||
}));
|
||||
|
||||
|
||||
Reference in New Issue
Block a user