From 3eebe0f41ed53473166a36dee612d5254e3dc4da Mon Sep 17 00:00:00 2001
From: Vidisha Bhatt <vidishabhatt5@gmail.com>
Date: Tue, 25 Aug 2020 11:18:19 -0500
Subject: [PATCH] Add select query

---
 CosmosClone/CosmicCloneUI/SourcePage.xaml                 | 8 ++++++--
 CosmosClone/CosmicCloneUI/SourcePage.xaml.cs              | 3 ++-
 .../CosmosCloneCommon/Migrator/DocumentMigrator.cs        | 2 +-
 CosmosClone/CosmosCloneCommon/Utility/CloneSettings.cs    | 1 +
 CosmosClone/CosmosCloneCommon/Utility/CosmosDBHelper.cs   | 5 ++---
 5 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/CosmosClone/CosmicCloneUI/SourcePage.xaml b/CosmosClone/CosmicCloneUI/SourcePage.xaml
index 16ccce5..a30ebc4 100644
--- a/CosmosClone/CosmicCloneUI/SourcePage.xaml
+++ b/CosmosClone/CosmicCloneUI/SourcePage.xaml
@@ -16,6 +16,7 @@
             <RowDefinition Height="30" />
             <RowDefinition Height="30" />
             <RowDefinition Height="30" />
+            <RowDefinition Height="30" />
             <RowDefinition Height="*" />
         </Grid.RowDefinitions>
         <Grid.ColumnDefinitions>
@@ -38,10 +39,13 @@
         <Label Content="Collection" Grid.Row="4" Grid.Column="0" Style="{StaticResource GridRowLabelStyle}" />
         <TextBox x:Name="SourceCollection" Grid.Row="4" Grid.Column="1" Style="{StaticResource GridRowTextBoxStyle}"/>
 
-        <StackPanel Grid.Row="5" Grid.Column="0" Grid.ColumnSpan="2" Orientation="Horizontal" HorizontalAlignment="Right">
+        <Label Content="SelectCriteria" Grid.Row="5" Grid.Column="0" Style="{StaticResource GridRowLabelStyle}" />
+        <TextBox x:Name="SelectQuery" Text="SELECT * FROM c" Grid.Row="5" Grid.Column="1" Style="{StaticResource GridRowTextBoxStyle}"/>
+
+        <StackPanel Grid.Row="6" Grid.Column="0" Grid.ColumnSpan="2" Orientation="Horizontal" HorizontalAlignment="Right">
             <Image x:Name="ConnectionIcon" Width="30" Height="30"/>
             <Button Click="BtnTestSource" Content="Test Connection" Style="{StaticResource ButtonStyle}"/>
         </StackPanel>
-        <TextBlock Grid.Row="6" Grid.Column="0" Grid.ColumnSpan="2" Name="ConnectionTestMsg" TextWrapping="Wrap" />
+        <TextBlock Grid.Row="7" Grid.Column="0" Grid.ColumnSpan="2" Name="ConnectionTestMsg" TextWrapping="Wrap" />
     </Grid>
 </Page>
diff --git a/CosmosClone/CosmicCloneUI/SourcePage.xaml.cs b/CosmosClone/CosmicCloneUI/SourcePage.xaml.cs
index 8ce0b49..b408706 100644
--- a/CosmosClone/CosmicCloneUI/SourcePage.xaml.cs
+++ b/CosmosClone/CosmicCloneUI/SourcePage.xaml.cs
@@ -41,7 +41,8 @@ public bool TestSourceConnection()
                 EndpointUrl = SourceURL.Text.ToString(),
                 AccessKey = SourceKey.Text.ToString(),
                 DatabaseName = SourceDB.Text.ToString(),
-                CollectionName = SourceCollection.Text.ToString()
+                CollectionName = SourceCollection.Text.ToString(),
+                SelectQuery = SelectQuery.Text.ToString()
             };
 
             var result = cosmosHelper.TestSourceConnection();
diff --git a/CosmosClone/CosmosCloneCommon/Migrator/DocumentMigrator.cs b/CosmosClone/CosmosCloneCommon/Migrator/DocumentMigrator.cs
index 7fdc9f5..47309f0 100644
--- a/CosmosClone/CosmosCloneCommon/Migrator/DocumentMigrator.cs
+++ b/CosmosClone/CosmosCloneCommon/Migrator/DocumentMigrator.cs
@@ -134,7 +134,7 @@ public async Task InitializeMigration()
             {
                 TotalRecordsInSource = cosmosHelper.GetSourceRecordCount();
                 logger.LogInfo($"Total records in Source: {TotalRecordsInSource} ");
-                SourceCommonDataFetchQuery = cosmosHelper.GetSourceEntityDocumentQuery<dynamic>(sourceClient, CloneSettings.ReadBatchSize);
+                SourceCommonDataFetchQuery = cosmosHelper.GetSourceEntityDocumentQuery<dynamic>(sourceClient, CloneSettings.SourceSettings.SelectQuery, CloneSettings.ReadBatchSize);
                 await cosmosBulkImporter.InitializeBulkExecutor(targetClient, targetCollection);
             }
             else
diff --git a/CosmosClone/CosmosCloneCommon/Utility/CloneSettings.cs b/CosmosClone/CosmosCloneCommon/Utility/CloneSettings.cs
index 0eae3c1..b01cc19 100644
--- a/CosmosClone/CosmosCloneCommon/Utility/CloneSettings.cs
+++ b/CosmosClone/CosmosCloneCommon/Utility/CloneSettings.cs
@@ -95,6 +95,7 @@ public class CosmosCollectionValues
         public string AccessKey { get; set; }
         public string DatabaseName { get; set; }
         public string CollectionName { get; set; }
+        public string SelectQuery { get; set; }
         public int OfferThroughputRUs { get; set; }
     }
 }
diff --git a/CosmosClone/CosmosCloneCommon/Utility/CosmosDBHelper.cs b/CosmosClone/CosmosCloneCommon/Utility/CosmosDBHelper.cs
index be06081..45b8ea8 100644
--- a/CosmosClone/CosmosCloneCommon/Utility/CosmosDBHelper.cs
+++ b/CosmosClone/CosmosCloneCommon/Utility/CosmosDBHelper.cs
@@ -273,7 +273,7 @@ public async Task<DocumentCollection> CreateTargetDocumentCollection(DocumentCli
             }
         }
 
-        public IQueryable<T> GetSourceEntityDocumentQuery<T>(DocumentClient sourceClient, int batchSize = -1)
+        public IQueryable<T> GetSourceEntityDocumentQuery<T>(DocumentClient sourceClient, string selectQuery, int batchSize = -1)
         {
             try
             {
@@ -281,8 +281,7 @@ public IQueryable<T> GetSourceEntityDocumentQuery<T>(DocumentClient sourceClient
                 string sourceDatabaseName = CloneSettings.SourceSettings.DatabaseName;
                 string sourceCollectionName = CloneSettings.SourceSettings.CollectionName;
                 FeedOptions queryOptions = new FeedOptions { MaxItemCount = batchSize, EnableCrossPartitionQuery = true };
-                string EntityDataQuery = $"SELECT * FROM c";
-                var documentQuery = sourceClient.CreateDocumentQuery<T>(UriFactory.CreateDocumentCollectionUri(sourceDatabaseName, sourceCollectionName), EntityDataQuery, queryOptions);
+                var documentQuery = sourceClient.CreateDocumentQuery<T>(UriFactory.CreateDocumentCollectionUri(sourceDatabaseName, sourceCollectionName), selectQuery, queryOptions);
                 return documentQuery;
             }
             catch (Exception ex)