mirror of
https://github.com/ckaczor/sqltoolsservice.git
synced 2026-01-17 17:23:48 -05:00
3278 Kusto Unit Test Refactor (#1053)
* 3278 Moved functions related to Metadata from DataSourceFactory to MetadataFactory.cs * 3278 Refactored DataSourceFactory to not be static. Added IDataSourceFactory interface. * 3278 Refactored IKustoIntellisenseHelper.cs to not be static. Added IKustoIntellisenseHelper.cs interface. * 3278 Removed unused functions from Scripter and deleted ScripterCore.cs because it was unused. Refactored Scripter.cs methods to not be static and created IScripter.cs * 3278 Refactored datsasourceConnectionFactory in ConnectionService to use MEF through the InitializeService function * 3278 Refactored IAutoCompleteHelper to not be static and added IAutoCompleteHelper interface. * 3278 Removed unused classes DatabaseFullAccessException and FeatureWithFullDbAccess. Refactored ObjectExplorerService to use ImportingConstructor attribute. Removed commented out in KustoResultsReader. Removed unused functions from DatabaseLocksManager. Removed unused IDataSourceConnectionFactory from ConnectionService * 3278 Moved SqlConnectionOpener class to new file. Added new interfaces IConnectedBindingQueue.cs and ISqlConnectionOpener.cs Added sqlConnectionOpener to dependency in ConnectedBindingQueue. * 3278 Removed needsMetadata param from ConnectedBindingQueue. Added param to AddConnectionContext where it's used * 3278 Refactored ConnectedBindingQueue to use MEF. Refactored usages to run against interface. * 3278 Corrected ServiceHost namespace. Removed unused dependency to sqlToolsContext in LanguageService. Updated dependency in MetadataService and LanguageService to IProtocolEndpoint instead of ServiceHost * 3278 Added back NextResult function and summary to KustoResultsReader. Renamed instance to _instance in ConnectionService and DatabaseLocksManager to stay consistent. Changed OpenDataSourceConnection to private in ConnectionService * 3278 Converted helper methods back to static and removed backing interfaces * 3278 Reverted AdminService > InitializeService to use ServiceHost as param
This commit is contained in:
@@ -0,0 +1,104 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using Microsoft.Kusto.ServiceLayer.Admin.Contracts;
|
||||
using Microsoft.Kusto.ServiceLayer.Metadata.Contracts;
|
||||
using Microsoft.Kusto.ServiceLayer.Utility;
|
||||
|
||||
namespace Microsoft.Kusto.ServiceLayer.DataSource.Metadata
|
||||
{
|
||||
public class MetadataFactory
|
||||
{
|
||||
public static DataSourceObjectMetadata CreateClusterMetadata(string clusterName)
|
||||
{
|
||||
ValidationUtils.IsArgumentNotNullOrWhiteSpace(clusterName, nameof(clusterName));
|
||||
|
||||
return new DataSourceObjectMetadata
|
||||
{
|
||||
MetadataType = DataSourceMetadataType.Cluster,
|
||||
MetadataTypeName = DataSourceMetadataType.Cluster.ToString(),
|
||||
Name = clusterName,
|
||||
PrettyName = clusterName,
|
||||
Urn = $"{clusterName}"
|
||||
};
|
||||
}
|
||||
|
||||
public static DataSourceObjectMetadata CreateDatabaseMetadata(DataSourceObjectMetadata clusterMetadata,
|
||||
string databaseName)
|
||||
{
|
||||
ValidationUtils.IsTrue<ArgumentException>(clusterMetadata.MetadataType == DataSourceMetadataType.Cluster,
|
||||
nameof(clusterMetadata));
|
||||
ValidationUtils.IsArgumentNotNullOrWhiteSpace(databaseName, nameof(databaseName));
|
||||
|
||||
return new DatabaseMetadata
|
||||
{
|
||||
ClusterName = clusterMetadata.Name,
|
||||
MetadataType = DataSourceMetadataType.Database,
|
||||
MetadataTypeName = DataSourceMetadataType.Database.ToString(),
|
||||
Name = databaseName,
|
||||
PrettyName = databaseName,
|
||||
Urn = $"{clusterMetadata.Urn}.{databaseName}"
|
||||
};
|
||||
}
|
||||
|
||||
public static FolderMetadata CreateFolderMetadata(DataSourceObjectMetadata parentMetadata, string path, string name)
|
||||
{
|
||||
ValidationUtils.IsNotNull(parentMetadata, nameof(parentMetadata));
|
||||
|
||||
return new FolderMetadata
|
||||
{
|
||||
MetadataType = DataSourceMetadataType.Folder,
|
||||
MetadataTypeName = DataSourceMetadataType.Folder.ToString(),
|
||||
Name = name,
|
||||
PrettyName = name,
|
||||
ParentMetadata = parentMetadata,
|
||||
Urn = $"{path}.{name}"
|
||||
};
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Converts database details shown on cluster manage dashboard to DatabaseInfo type. Add DataSourceType as param if required to show different properties
|
||||
/// </summary>
|
||||
/// <param name="clusterDBDetails"></param>
|
||||
/// <returns></returns>
|
||||
public static List<DatabaseInfo> ConvertToDatabaseInfo(IEnumerable<DataSourceObjectMetadata> clusterDBDetails)
|
||||
{
|
||||
var databaseDetails = new List<DatabaseInfo>();
|
||||
|
||||
if (typeof(DatabaseMetadata) == clusterDBDetails.FirstOrDefault().GetType())
|
||||
{
|
||||
foreach (var dbDetail in clusterDBDetails)
|
||||
{
|
||||
DatabaseInfo databaseInfo = new DatabaseInfo();
|
||||
Int64.TryParse(dbDetail.SizeInMB.ToString(), out long sum_OriginalSize);
|
||||
databaseInfo.Options["name"] = dbDetail.Name;
|
||||
databaseInfo.Options["sizeInMB"] = (sum_OriginalSize / (1024 * 1024)).ToString();
|
||||
databaseDetails.Add(databaseInfo);
|
||||
}
|
||||
}
|
||||
|
||||
return databaseDetails;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Converts tables details shown on database manage dashboard to ObjectMetadata type. Add DataSourceType as param if required to show different properties
|
||||
/// </summary>
|
||||
/// <param name="dbChildDetails"></param>
|
||||
/// <returns></returns>
|
||||
public static List<ObjectMetadata> ConvertToObjectMetadata(IEnumerable<DataSourceObjectMetadata> dbChildDetails)
|
||||
{
|
||||
var databaseChildDetails = new List<ObjectMetadata>();
|
||||
|
||||
foreach (var childDetail in dbChildDetails)
|
||||
{
|
||||
ObjectMetadata dbChildInfo = new ObjectMetadata();
|
||||
dbChildInfo.Name = childDetail.PrettyName;
|
||||
dbChildInfo.MetadataTypeName = childDetail.MetadataTypeName;
|
||||
dbChildInfo.MetadataType = MetadataType.Table; // Add mapping here.
|
||||
databaseChildDetails.Add(dbChildInfo);
|
||||
}
|
||||
|
||||
return databaseChildDetails;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user