Merge pull request #8 from alexdrenea/feature/offer-throughput

Fix for issue #7 - respect offer throughput for target collection
microsoft · Jul 31, 2019 · ffe7b5d · ffe7b5d
2 parents 019a668 + d3cc923
commit ffe7b5d
Show file tree

Hide file tree

Showing 5 changed files with 45 additions and 10 deletions.
diff --git a/CosmosClone/CosmicCloneUI/CloneOptionsPage.xaml b/CosmosClone/CosmicCloneUI/CloneOptionsPage.xaml
@@ -19,5 +19,9 @@
             <TextBlock Text="Offer Throughput" Style="{StaticResource TextBlockStyle}"/>
             <TextBox x:Name="OfferThroughput" Text="10000" Style="{StaticResource TextBoxStyle}" VerticalAlignment="Top"/>
         </StackPanel>
+        <StackPanel Orientation="Horizontal">
+            <Image x:Name="ConnectionIcon" Width="30" Height="30"/>
+            <TextBlock Name="ConnectionTestMsg" TextWrapping="Wrap" />
+        </StackPanel>
     </StackPanel>
 </Page>
diff --git a/CosmosClone/CosmicCloneUI/CloneOptionsPage.xaml.cs b/CosmosClone/CosmicCloneUI/CloneOptionsPage.xaml.cs
@@ -27,5 +27,31 @@ public CloneOptionsPage()
         {
             InitializeComponent();
         }
+
+
+        public bool TestCloneOptions()
+        {
+            var result = true;
+            if (!int.TryParse(OfferThroughput.Text, out int RU))
+                result = false;
+
+            if (RU < 400) result = false;
+            if (RU % 100 != 0) result = false;
+
+            if (result)
+            {
+                var connectionIcon = (Image)this.FindName("ConnectionIcon");
+                ConnectionIcon.Source = new BitmapImage(new Uri("/Images/success.png", UriKind.Relative));
+                ConnectionTestMsg.Text = "Validation Passed";
+            }
+            else
+            {
+                var connectionIcon = (Image)this.FindName("ConnectionIcon");
+                ConnectionIcon.Source = new BitmapImage(new Uri("/Images/fail.png", UriKind.Relative));
+                ConnectionTestMsg.Text = "Invalid Throughput provided. Make sure it's a number greater than 400 and multiple of 100.";
+            }
+
+            return result;
+        }
     }
 }
diff --git a/CosmosClone/CosmicCloneUI/MainWindow.xaml.cs b/CosmosClone/CosmicCloneUI/MainWindow.xaml.cs
@@ -157,12 +157,17 @@ private bool PerformAction(Page currentPage)
             }
             else if (GetPageNumber(currentPage) == 2)
             {
-                CloneSettings.CopyStoredProcedures = ((CheckBox)currentPage.FindName("SPs")).IsChecked.Value;
-                CloneSettings.CopyUDFs = ((CheckBox)currentPage.FindName("UDFs")).IsChecked.Value;
-                CloneSettings.CopyTriggers = ((CheckBox)currentPage.FindName("CosmosTriggers")).IsChecked.Value;
-                CloneSettings.CopyDocuments = ((CheckBox)currentPage.FindName("Documents")).IsChecked.Value;
-                CloneSettings.CopyIndexingPolicy = ((CheckBox)currentPage.FindName("IPs")).IsChecked.Value;
-                CloneSettings.CopyPartitionKey = ((CheckBox)currentPage.FindName("PKs")).IsChecked.Value;
+                var page = ((CloneOptionsPage)currentPage);
+                var valid = page.TestCloneOptions();
+                if (!valid) return false;
+
+                CloneSettings.CopyStoredProcedures = page.SPs.IsChecked.Value;
+                CloneSettings.CopyUDFs = page.UDFs.IsChecked.Value;
+                CloneSettings.CopyTriggers = page.CosmosTriggers.IsChecked.Value;
+                CloneSettings.CopyDocuments = page.Documents.IsChecked.Value;
+                CloneSettings.CopyIndexingPolicy = page.IPs.IsChecked.Value;
+                CloneSettings.CopyPartitionKey = page.PKs.IsChecked.Value;
+                CloneSettings.TargetMigrationOfferThroughputRUs = int.Parse(page.OfferThroughput.Text);
 
                 return true;                
             }

diff --git a/CosmosClone/CosmosCloneCommon/Model/RandomNumberGenerator.cs b/CosmosClone/CosmosCloneCommon/Model/RandomNumberGenerator.cs
@@ -8,7 +8,7 @@ namespace CosmosCloneCommon.Model
 
     public static class RandomNumberGenerator
     {
-        private static readonly Random _random = new Random(unchecked(Environment.TickCount * 31 + Thread.CurrentThread.ManagedThreadId));;
+        private static readonly Random _random = new Random(unchecked(Environment.TickCount * 31 + Thread.CurrentThread.ManagedThreadId));
 
         public static int GetNext(int maxValue)
         {

diff --git a/CosmosClone/CosmosCloneCommon/Utility/CloneSettings.cs b/CosmosClone/CosmosCloneCommon/Utility/CloneSettings.cs
@@ -69,8 +69,8 @@ static CloneSettings()
                 EndpointUrl = targetConfigs["EndpointUrl"],
                 AccessKey = targetConfigs["AccessKey"],
                 DatabaseName = targetConfigs["DatabaseName"],
-                CollectionName = targetConfigs["CollectionName"]
-               // OfferThroughputRUs = int.Parse(sourceConfigs["OfferThroughputRUs"])
+                CollectionName = targetConfigs["CollectionName"],
+                OfferThroughputRUs = int.Parse(targetConfigs["OfferThroughputRUs"])
             };
         }
 
@@ -95,6 +95,6 @@ public class CosmosCollectionValues
         public string AccessKey { get; set; }
         public string DatabaseName { get; set; }
         public string CollectionName { get; set; }
-
+        public int OfferThroughputRUs { get; set; }
     }
 }