From 1ee0c1d09a90ab6ddbcd6e23313fd357f2c188cc Mon Sep 17 00:00:00 2001 From: Monica Gupta Date: Thu, 8 Oct 2020 17:51:08 -0700 Subject: [PATCH] Make Kusto control commands work (#1094) Co-authored-by: Monica Gupta --- .../DataSource/KustoClient.cs | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/src/Microsoft.Kusto.ServiceLayer/DataSource/KustoClient.cs b/src/Microsoft.Kusto.ServiceLayer/DataSource/KustoClient.cs index d9800c38..e7f3001e 100644 --- a/src/Microsoft.Kusto.ServiceLayer/DataSource/KustoClient.cs +++ b/src/Microsoft.Kusto.ServiceLayer/DataSource/KustoClient.cs @@ -165,10 +165,21 @@ namespace Microsoft.Kusto.ServiceLayer.DataSource { var minimalQuery = codeBlock.Service.GetMinimalText(MinimalTextKind.RemoveLeadingWhitespaceAndComments); - IDataReader origReader = _kustoQueryProvider.ExecuteQuery( - KustoQueryUtils.IsClusterLevelQuery(minimalQuery) ? "" : databaseName, - minimalQuery, - clientRequestProperties); + + IDataReader origReader; + + if(minimalQuery.StartsWith(".") && !minimalQuery.StartsWith(".show")){ + origReader = _kustoAdminProvider.ExecuteControlCommand( + KustoQueryUtils.IsClusterLevelQuery(minimalQuery) ? "" : databaseName, + minimalQuery, + clientRequestProperties); + } + else{ + origReader = _kustoQueryProvider.ExecuteQuery( + KustoQueryUtils.IsClusterLevelQuery(minimalQuery) ? "" : databaseName, + minimalQuery, + clientRequestProperties); + } origReaders[index] = origReader; });