mirror of
https://github.com/Azure/cosmos-explorer.git
synced 2024-12-11 06:47:01 +00:00
6925fa8e4e
* Use Az login with OpenID connection to get test credentials. * Set subscription id environment variable. * Update testExplorer and cleanup job. * Retrieve access token in test case and pass to testExplorer. * Add debug tracing for tests. * Set up other mongo test to use Az CLI creds. * Revert subscription id retrieval. * Add CLI credentials retrieval to rest of tests. * Fix missing imports. * Clean up redundant code. * Remove commented import statement.
96 lines
4.6 KiB
JavaScript
96 lines
4.6 KiB
JavaScript
const msRestNodeAuth = require("@azure/ms-rest-nodeauth");
|
|
const { CosmosDBManagementClient } = require("@azure/arm-cosmosdb");
|
|
const ms = require("ms");
|
|
|
|
const subscriptionId = process.env["AZURE_SUBSCRIPTION_ID"];
|
|
const resourceGroupName = "runners";
|
|
|
|
const thirtyMinutesAgo = new Date(Date.now() - 1000 * 60 * 30).getTime();
|
|
|
|
function friendlyTime(date) {
|
|
try {
|
|
return ms(date);
|
|
} catch (error) {
|
|
return "Unknown";
|
|
}
|
|
}
|
|
|
|
async function main() {
|
|
const credentials = await msRestNodeAuth.AzureCliCredentials.create();
|
|
const client = new CosmosDBManagementClient(credentials, subscriptionId);
|
|
const accounts = await client.databaseAccounts.list(resourceGroupName);
|
|
for (const account of accounts) {
|
|
if (account.kind === "MongoDB") {
|
|
const mongoDatabases = await client.mongoDBResources.listMongoDBDatabases(resourceGroupName, account.name);
|
|
for (const database of mongoDatabases) {
|
|
// Unfortunately Mongo does not provide a timestamp in ARM. There is no way to tell how old the DB is other thn encoding it in the ID :(
|
|
const timestamp = Number(database.name.split("-")[1]);
|
|
if (timestamp && timestamp < thirtyMinutesAgo) {
|
|
await client.mongoDBResources.deleteMongoDBDatabase(resourceGroupName, account.name, database.name);
|
|
console.log(`DELETED: ${account.name} | ${database.name} | Age: ${friendlyTime(Date.now() - timestamp)}`);
|
|
} else {
|
|
console.log(`SKIPPED: ${account.name} | ${database.name} | Age: ${friendlyTime(Date.now() - timestamp)}`);
|
|
}
|
|
}
|
|
} else if (account.capabilities.find((c) => c.name === "EnableCassandra")) {
|
|
const cassandraDatabases = await client.cassandraResources.listCassandraKeyspaces(
|
|
resourceGroupName,
|
|
account.name,
|
|
);
|
|
for (const database of cassandraDatabases) {
|
|
const timestamp = Number(database.resource._ts) * 1000;
|
|
if (timestamp && timestamp < thirtyMinutesAgo) {
|
|
await client.cassandraResources.deleteCassandraKeyspace(resourceGroupName, account.name, database.name);
|
|
console.log(`DELETED: ${account.name} | ${database.name} | Age: ${friendlyTime(Date.now() - timestamp)}`);
|
|
} else {
|
|
console.log(`SKIPPED: ${account.name} | ${database.name} | Age: ${friendlyTime(Date.now() - timestamp)}`);
|
|
}
|
|
}
|
|
} else if (account.capabilities.find((c) => c.name === "EnableTable")) {
|
|
const tablesDatabase = await client.tableResources.listTables(resourceGroupName, account.name);
|
|
for (const database of tablesDatabase) {
|
|
const timestamp = Number(database.resource._ts) * 1000;
|
|
if (timestamp && timestamp < thirtyMinutesAgo) {
|
|
await client.tableResources.deleteTable(resourceGroupName, account.name, database.name);
|
|
console.log(`DELETED: ${account.name} | ${database.name} | Age: ${friendlyTime(Date.now() - timestamp)}`);
|
|
} else {
|
|
console.log(`SKIPPED: ${account.name} | ${database.name} | Age: ${friendlyTime(Date.now() - timestamp)}`);
|
|
}
|
|
}
|
|
} else if (account.capabilities.find((c) => c.name === "EnableGremlin")) {
|
|
const graphDatabases = await client.gremlinResources.listGremlinDatabases(resourceGroupName, account.name);
|
|
for (const database of graphDatabases) {
|
|
const timestamp = Number(database.resource._ts) * 1000;
|
|
if (timestamp && timestamp < thirtyMinutesAgo) {
|
|
await client.gremlinResources.deleteGremlinDatabase(resourceGroupName, account.name, database.name);
|
|
console.log(`DELETED: ${account.name} | ${database.name} | Age: ${friendlyTime(Date.now() - timestamp)}`);
|
|
} else {
|
|
console.log(`SKIPPED: ${account.name} | ${database.name} | Age: ${friendlyTime(Date.now() - timestamp)}`);
|
|
}
|
|
}
|
|
} else if (account.kind === "GlobalDocumentDB") {
|
|
const sqlDatabases = await client.sqlResources.listSqlDatabases(resourceGroupName, account.name);
|
|
for (const database of sqlDatabases) {
|
|
const timestamp = Number(database.resource._ts) * 1000;
|
|
if (timestamp && timestamp < thirtyMinutesAgo) {
|
|
await client.sqlResources.deleteSqlDatabase(resourceGroupName, account.name, database.name);
|
|
console.log(`DELETED: ${account.name} | ${database.name} | Age: ${friendlyTime(Date.now() - timestamp)}`);
|
|
} else {
|
|
console.log(`SKIPPED: ${account.name} | ${database.name} | Age: ${friendlyTime(Date.now() - timestamp)}`);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
main()
|
|
.then(() => {
|
|
console.log("Completed");
|
|
process.exit(0);
|
|
})
|
|
.catch((err) => {
|
|
console.log(err);
|
|
console.log("Cleanup failed! Exiting with success. Cleanup should always fail safe.");
|
|
process.exit(0);
|
|
});
|