mirror of
https://github.com/Azure/cosmos-explorer.git
synced 2025-03-04 00:48:07 +00:00
Add read and write endpoint logging to cosmos client.
This commit is contained in:
parent
47672b650f
commit
b63735146b
@ -218,6 +218,41 @@ export function client(): Cosmos.CosmosClient {
|
||||
_defaultHeaders["x-ms-cosmos-priority-level"] = PriorityLevel.Default;
|
||||
}
|
||||
|
||||
// Parsing out endpoint from diagnostics. Used to find address I need to add to firewall rule.
|
||||
function parseEndpointFromDiag(json: string): string {
|
||||
const suffix: string = ".documents.azure.com";
|
||||
const start: number = json.indexOf("//") + "//".length;
|
||||
const end: number = json.indexOf(suffix) + suffix.length;
|
||||
const endpoint: string = json.substring(start, end);
|
||||
|
||||
return endpoint;
|
||||
}
|
||||
// Used to check current client region configuration.
|
||||
async function fetchConnectedRegions(client: Cosmos.CosmosClient) {
|
||||
// Check currently connected regions.
|
||||
try {
|
||||
const someMoreThings = await client.databases.readAll().fetchAll();
|
||||
console.log(`Current list of databases: ${JSON.stringify(someMoreThings)}`);
|
||||
const currentReadRegion = await client.getReadEndpoint();
|
||||
console.log(`Current read endpoint: ${JSON.stringify(currentReadRegion)}`);
|
||||
const currentReadRegions = await client.getReadEndpoints();
|
||||
console.log(`Current account endpoints: ${JSON.stringify(currentReadRegions)}`);
|
||||
// Getting primary region IP that needs to be blocked.
|
||||
// retrieve the regional endpoint of the account
|
||||
const regionalWriteEndpoint = await client.getWriteEndpoint();
|
||||
console.log(`Current write endpoint: ${JSON.stringify(regionalWriteEndpoint)}`);
|
||||
const parsedWriteEndpoint = parseEndpointFromDiag(JSON.stringify(regionalWriteEndpoint));
|
||||
console.log(`Current parsed write endpoint: ${JSON.stringify(parsedWriteEndpoint)}`);
|
||||
// const writeHostAddress = await findHostAddress(parsedWriteEndpoint);
|
||||
// console.log(`Current write host address: ${JSON.stringify(writeHostAddress)}`);
|
||||
} catch (error) {
|
||||
console.error("Error getting read endpoints:", error);
|
||||
}
|
||||
|
||||
const currentWriteRegion = await client.getWriteEndpoint();
|
||||
console.log(`Current write endpoint: ${JSON.stringify(currentWriteRegion)}`);
|
||||
}
|
||||
|
||||
const options: Cosmos.CosmosClientOptions = {
|
||||
endpoint: endpoint() || "https://cosmos.azure.com", // CosmosClient gets upset if we pass a bad URL. This should never actually get called
|
||||
key: userContext.dataPlaneRbacEnabled ? "" : userContext.masterKey,
|
||||
@ -244,5 +279,9 @@ export function client(): Cosmos.CosmosClient {
|
||||
}
|
||||
|
||||
_client = new Cosmos.CosmosClient(options);
|
||||
|
||||
// Log debug vals
|
||||
fetchConnectedRegions(_client).catch((error) => console.error(error));
|
||||
|
||||
return _client;
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user