mirror of
https://github.com/Azure/cosmos-explorer.git
synced 2025-04-07 10:28:57 +01:00
1c80ced259
* Fix settings tab issue for cassandra and add feature flag * Adding the rest of the changes.... Co-authored-by: Steve Faulkner <southpolesteve@gmail.com>
72 lines
3.3 KiB
TypeScript
72 lines
3.3 KiB
TypeScript
import * as DataModels from "../../Contracts/DataModels";
|
|
import { AuthType } from "../../AuthType";
|
|
import { DefaultAccountExperienceType } from "../../DefaultAccountExperienceType";
|
|
import { client } from "../CosmosClient";
|
|
import { listSqlContainers } from "../../Utils/arm/generatedClients/2020-04-01/sqlResources";
|
|
import { listCassandraTables } from "../../Utils/arm/generatedClients/2020-04-01/cassandraResources";
|
|
import { listMongoDBCollections } from "../../Utils/arm/generatedClients/2020-04-01/mongoDBResources";
|
|
import { listGremlinGraphs } from "../../Utils/arm/generatedClients/2020-04-01/gremlinResources";
|
|
import { listTables } from "../../Utils/arm/generatedClients/2020-04-01/tableResources";
|
|
import { logConsoleProgress, logConsoleError } from "../../Utils/NotificationConsoleUtils";
|
|
import { logError } from "../Logger";
|
|
import { sendNotificationForError } from "./sendNotificationForError";
|
|
import { userContext } from "../../UserContext";
|
|
|
|
export async function readCollections(databaseId: string): Promise<DataModels.Collection[]> {
|
|
let collections: DataModels.Collection[];
|
|
const clearMessage = logConsoleProgress(`Querying containers for database ${databaseId}`);
|
|
try {
|
|
if (
|
|
window.authType === AuthType.AAD &&
|
|
!userContext.useSDKOperations &&
|
|
userContext.defaultExperience !== DefaultAccountExperienceType.MongoDB &&
|
|
userContext.defaultExperience !== DefaultAccountExperienceType.Table
|
|
) {
|
|
collections = await readCollectionsWithARM(databaseId);
|
|
} else {
|
|
const sdkResponse = await client()
|
|
.database(databaseId)
|
|
.containers.readAll()
|
|
.fetchAll();
|
|
collections = sdkResponse.resources as DataModels.Collection[];
|
|
}
|
|
} catch (error) {
|
|
logConsoleError(`Error while querying containers for database ${databaseId}:\n ${JSON.stringify(error)}`);
|
|
logError(JSON.stringify(error), "ReadCollections", error.code);
|
|
sendNotificationForError(error);
|
|
throw error;
|
|
}
|
|
clearMessage();
|
|
return collections;
|
|
}
|
|
|
|
async function readCollectionsWithARM(databaseId: string): Promise<DataModels.Collection[]> {
|
|
let rpResponse;
|
|
const subscriptionId = userContext.subscriptionId;
|
|
const resourceGroup = userContext.resourceGroup;
|
|
const accountName = userContext.databaseAccount.name;
|
|
const defaultExperience = userContext.defaultExperience;
|
|
|
|
switch (defaultExperience) {
|
|
case DefaultAccountExperienceType.DocumentDB:
|
|
rpResponse = await listSqlContainers(subscriptionId, resourceGroup, accountName, databaseId);
|
|
break;
|
|
case DefaultAccountExperienceType.MongoDB:
|
|
rpResponse = await listMongoDBCollections(subscriptionId, resourceGroup, accountName, databaseId);
|
|
break;
|
|
case DefaultAccountExperienceType.Cassandra:
|
|
rpResponse = await listCassandraTables(subscriptionId, resourceGroup, accountName, databaseId);
|
|
break;
|
|
case DefaultAccountExperienceType.Graph:
|
|
rpResponse = await listGremlinGraphs(subscriptionId, resourceGroup, accountName, databaseId);
|
|
break;
|
|
case DefaultAccountExperienceType.Table:
|
|
rpResponse = await listTables(subscriptionId, resourceGroup, accountName);
|
|
break;
|
|
default:
|
|
throw new Error(`Unsupported default experience type: ${defaultExperience}`);
|
|
}
|
|
|
|
return rpResponse?.value?.map(collection => collection.properties?.resource as DataModels.Collection);
|
|
}
|