This commit is contained in:
Krishan Mistry 2023-09-04 17:14:01 +01:00
Родитель 65b94830b5
Коммит af220f6d16
2 изменённых файлов: 4 добавлений и 4 удалений

Просмотреть файл

@ -27,7 +27,7 @@ impl AuthorizationContext {
} }
/// Executes a KQL query to get the Kusto identity token from the management endpoint /// Executes a KQL query to get the Kusto identity token from the management endpoint
async fn execute_kql_mgmt_query(&self) -> Result<KustoIdentityToken> { async fn query_kusto_identity_token(&self) -> Result<KustoIdentityToken> {
let results = self let results = self
.client .client
.execute_command("NetDefaultDB", ".get kusto identity token", None) .execute_command("NetDefaultDB", ".get kusto identity token", None)
@ -101,7 +101,7 @@ impl AuthorizationContext {
} }
// Fetch new token from Kusto, update the cache, and return the token // Fetch new token from Kusto, update the cache, and return the token
let token = self.execute_kql_mgmt_query().await?; let token = self.query_kusto_identity_token().await?;
auth_context_cache.update(Some(token.clone())); auth_context_cache.update(Some(token.clone()));
Ok(token) Ok(token)

Просмотреть файл

@ -109,7 +109,7 @@ impl IngestClientResources {
} }
/// Executes a KQL management query that retrieves resource URIs for the various Azure resources used for ingestion /// Executes a KQL management query that retrieves resource URIs for the various Azure resources used for ingestion
async fn execute_kql_mgmt_query(&self) -> Result<InnerIngestClientResources> { async fn query_ingestion_resources(&self) -> Result<InnerIngestClientResources> {
let results = self let results = self
.client .client
.execute_command("NetDefaultDB", ".get ingestion resources", None) .execute_command("NetDefaultDB", ".get ingestion resources", None)
@ -142,7 +142,7 @@ impl IngestClientResources {
} }
} }
let new_resources = self.execute_kql_mgmt_query().await?; let new_resources = self.query_ingestion_resources().await?;
resources.update(Some(new_resources.clone())); resources.update(Some(new_resources.clone()));
Ok(new_resources) Ok(new_resources)