ckaczor/sqltoolsservice
1
0
Fork 0
mirror of https://github.com/ckaczor/sqltoolsservice.git synced 2026-01-22 01:25:44 -05:00
Code Issues Packages Projects Releases Wiki Activity

3487 Created new function GetTablesForDashboard in KustoDataSource to load tables for ADS Dashboard (#1056)

Browse Source
This commit is contained in:
Justin M
2020-08-25 11:55:07 -07:00
committed by GitHub
parent 14f5a3e0f1
commit 3f725b5aec
1 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/Microsoft.Kusto.ServiceLayer/DataSource/KustoDataSource.cs
View File

@@ -572,7 +572,9 @@ namespace Microsoft.Kusto.ServiceLayer.DataSource
return GetDatabaseMetadata(includeSizeDetails);
case DataSourceMetadataType.Database: // show folders, tables, and functions
return GetDatabaseSchema(objectMetadata, includeSizeDetails);
return includeSizeDetails
? GetTablesForDashboard(objectMetadata)
: GetDatabaseSchema(objectMetadata);
case DataSourceMetadataType.Table: // show columns
var table = objectMetadata as TableMetadata;
@@ -621,13 +623,13 @@ namespace Microsoft.Kusto.ServiceLayer.DataSource
}
/// <inheritdoc/>
private IEnumerable<DataSourceObjectMetadata> GetDatabaseSchema(DataSourceObjectMetadata objectMetadata, bool includeSizeDetails)
private IEnumerable<DataSourceObjectMetadata> GetDatabaseSchema(DataSourceObjectMetadata objectMetadata)
{
// Check if the database exists
ValidationUtils.IsTrue<ArgumentException>(DatabaseExists(objectMetadata.Name).Result, $"Database '{objectMetadata}' does not exist.");
var allMetadata = GetAllMetadata(objectMetadata.Urn);
// if the records have already been loaded them return them
if (allMetadata.Any())
{
@@ -636,13 +638,19 @@ namespace Microsoft.Kusto.ServiceLayer.DataSource
LoadTableSchema(objectMetadata);
LoadFunctionSchema(objectMetadata);
return GetAllMetadata(objectMetadata.Urn);
}
if (!includeSizeDetails)
{
return GetAllMetadata(objectMetadata.Urn);
}
private IEnumerable<DataSourceObjectMetadata> GetTablesForDashboard(DataSourceObjectMetadata objectMetadata)
{
string newKey = $"{DatabaseKeyPrefix}.{objectMetadata.Urn}";
if (!_tableMetadata.ContainsKey(newKey) || !_tableMetadata[newKey].Any())
{
LoadTableSchema(objectMetadata);
}
return _tableMetadata[newKey].OrderBy(x => x.PrettyName, StringComparer.OrdinalIgnoreCase);
}