private static async Task <string> GetAccessToken(ApiToken apiToken)
{
var app = ConfidentialClientApplicationBuilder.Create(apiToken.ClientId.ToString())
.WithClientSecret(apiToken.ClientSecret)
.WithAuthority(new Uri($"https://login.microsoftonline.com/{apiToken.Tenant}"))
.Build();
var result = await app.AcquireTokenForClient(new[] { "https://westus2.api.loganalytics.io/.default" })
.ExecuteAsync();
return(result.AccessToken);
}
private static HttpClient GetHttpClient(ApiToken apiToken)
{
return(HttpClients.GetOrAdd(
apiToken.Id,
id =>
{
var client = new HttpClient();
if (!string.IsNullOrWhiteSpace(apiToken.Key))
{
client.DefaultRequestHeaders.Add("x-api-key", apiToken.Key);
}
else
{
var accessToken = GetAccessToken(apiToken).Result;
client.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Bearer", accessToken);
}
return client;
}));
}
public static async Task <List <TableResult> > GetTablesQuery(ApiToken apiToken, params string[] queries)
{
var httpClient = GetHttpClient(apiToken);
var url = string.Format(QueryUrl, apiToken.Type.ToString().ToLower(), apiToken.Id);
var jsonRequest = JsonConvert.SerializeObject(new { query = queries[0] });
var response = await httpClient.PostAsync(url, new StringContent(jsonRequest, Encoding.UTF8, "application/json"));
var responseJson = await response.Content.ReadAsStringAsync();
var queryResult = JObject.Parse(responseJson);
var result = new List <TableResult>();
foreach (var table in queryResult["tables"])
{
var tableResult = new TableResult();
var columns = table["columns"];
var rows = table["rows"];
foreach (var column in columns)
{
var columnType = Enum.TryParse <QueryResultType>(column["type"].Value <string>(), true, out var queryResultType)
? queryResultType
: QueryResultType.String;
tableResult.Columns.Add(new TableResultColumn(column["name"].Value <string>(), columnType));
}
foreach (var row in rows)
{
var rowResult = new List <dynamic>();
for (var i = 0; i < row.Count(); i++)
{
switch (tableResult.Columns[i].Type)
{
default:
case QueryResultType.String:
rowResult.Add(row[i].Value <string>());
break;
case QueryResultType.Long:
rowResult.Add(row[i].Value <string>() != "NaN" ? row[i].Value <long>() : 0);
break;
case QueryResultType.Real:
rowResult.Add(row[i].Value <string>() != "NaN" ? row[i].Value <double>() : 0);
break;
case QueryResultType.Datetime:
rowResult.Add(row[i].Value <DateTime>());
break;
}
}
tableResult.Rows.Add(rowResult);
}
result.Add(tableResult);
}
return(result);
}
public static async Task <TableResult> GetTableQuery(ApiToken apiToken, string query)
{
return((await GetTablesQuery(apiToken, query)).First());
}
