Add support to update solr

Author: Maarten Willebrands

SyncTokenGnerator/Program.cs

@@ -1,9 +1,12 @@
 using Microsoft.Azure.SqlDatabase.ElasticScale.ShardManagement;
+using Newtonsoft.Json;
+using Newtonsoft.Json.Linq;
 using System;
 using System.Collections.Generic;
 using System.Data.SqlClient;
 using System.IO;
 using System.Linq;
+using System.Net;
 using System.Runtime.Serialization.Formatters.Binary;
 
 namespace SyncTokenGenerator
@@ -13,9 +16,9 @@ internal class Program
         static void Main(string[] args)
         {
             Console.WriteLine("Please enter the connectionstring to the shardmapmanager");
-
             var shardMapManagerConnectionString = Console.ReadLine();
             Dictionary<string, long?> shardChangeVersions = new Dictionary<string, long?>();
+            string syncToken = "";
 
             try
             {
@@ -56,16 +59,52 @@ static void Main(string[] args)
                 {
                     BinaryFormatter binaryFormatter = new BinaryFormatter();
                     binaryFormatter.Serialize(memoryStream, shardChangeVersions);
+                    syncToken = Convert.ToBase64String(memoryStream.ToArray());
                     Console.WriteLine("The new sync token is:");
-                    Console.WriteLine(Convert.ToBase64String(memoryStream.ToArray()));
+                    Console.WriteLine(syncToken);
                 }
             }
-            catch(Exception ex)
+            catch (Exception ex)
             {
-                Console.WriteLine("Unable to access the Shard Map Manager database");
-                Console.WriteLine(ex.Message);
+                WriteError("Unable to access the Shard Map Manager database");
+                WriteError(ex.Message);
+            }
+
+            Console.WriteLine();
+            Console.WriteLine("Update XDB SOLR Index? (Y/N)");
+            bool updateSolr = Console.ReadLine().ToLowerInvariant() == "y";
+            if (updateSolr)
+            {
+                try
+                {
+                    Console.WriteLine("Please enter the SOLR XDB connectionstring, for ex https://localhost:8983/solr/sitecore_xdb");
+                    var solrConnectionString = Console.ReadLine();
+                    if (!solrConnectionString.EndsWith("/"))
+                    {
+                        solrConnectionString += "/";
+                    }
+                    solrConnectionString += "update?commit=true";
+                    using (var webclient = new WebClient())
+                    {
+                        var postBody = $"[{{ \"id\":\"xdb-index-token\",\"xdbtokenbytes_s\":\"{syncToken}\"}}]";
+                        webclient.Headers.Add("content-type", "application/json");
+                        var response = webclient.UploadString(solrConnectionString, "post", postBody);
+                        var jsonResponse = !string.IsNullOrEmpty(response) ? JObject.Parse(response) : null; ;
+                        if (!int.TryParse((jsonResponse["responseHeader"]?["status"] as JValue)?.Value?.ToString(), out var status) || status != 0)
+                        {
+                            WriteError("Invalid response recieved from SOLR");
+                            WriteError(jsonResponse?.ToString());
+                        }
+                    }
+                }
+                catch (Exception ex)
+                {
+                    WriteError("Unable to update Solr");
+                    WriteError(ex.Message);
+                }
             }
-                
+
+            Console.WriteLine("Done, press any key to exit");
             Console.ReadLine();
         }
 
@@ -84,5 +123,13 @@ static string Credentials(string connectionString)
             }
             return result.ToString();
         }
+
+        static void WriteError(string message)
+        {
+            var originalColor = Console.ForegroundColor;
+            Console.ForegroundColor = ConsoleColor.Red;
+            Console.Error.WriteLine(message);
+            Console.ForegroundColor = originalColor;
+        }
     }
 }

SyncTokenGnerator/Properties/AssemblyInfo.cs

@@ -32,5 +32,5 @@
 // You can specify all the values or you can default the Build and Revision Numbers
 // by using the '*' as shown below:
 // [assembly: AssemblyVersion("1.0.*")]
-[assembly: AssemblyVersion("1.0.0.0")]
-[assembly: AssemblyFileVersion("1.0.0.0")]
+[assembly: AssemblyVersion("1.1.0.0")]
+[assembly: AssemblyFileVersion("1.1.0.0")]

SyncTokenGnerator/SyncTokenGenerator.csproj

@@ -12,6 +12,7 @@
     <FileAlignment>512</FileAlignment>
     <AutoGenerateBindingRedirects>true</AutoGenerateBindingRedirects>
     <Deterministic>true</Deterministic>
+    <IsWebBootstrapper>false</IsWebBootstrapper>
     <PublishUrl>publish\</PublishUrl>
     <Install>true</Install>
     <InstallFrom>Disk</InstallFrom>
@@ -22,9 +23,8 @@
     <UpdatePeriodically>false</UpdatePeriodically>
     <UpdateRequired>false</UpdateRequired>
     <MapFileExtensions>true</MapFileExtensions>
-    <ApplicationRevision>0</ApplicationRevision>
+    <ApplicationRevision>1</ApplicationRevision>
     <ApplicationVersion>1.0.0.%2a</ApplicationVersion>
-    <IsWebBootstrapper>false</IsWebBootstrapper>
     <UseApplicationTrust>false</UseApplicationTrust>
     <PublishWizardCompleted>true</PublishWizardCompleted>
     <BootstrapperEnabled>true</BootstrapperEnabled>
@@ -64,6 +64,9 @@
     <Reference Include="Microsoft.Azure.SqlDatabase.ElasticScale.Client, Version=2.3.0.0, Culture=neutral, PublicKeyToken=b1e0ec4a89ab0a4a, processorArchitecture=MSIL">
       <HintPath>..\packages\Microsoft.Azure.SqlDatabase.ElasticScale.Client.2.3.0\lib\net451\Microsoft.Azure.SqlDatabase.ElasticScale.Client.dll</HintPath>
     </Reference>
+    <Reference Include="Newtonsoft.Json, Version=13.0.0.0, Culture=neutral, PublicKeyToken=30ad4fe6b2a6aeed, processorArchitecture=MSIL">
+      <HintPath>..\packages\Newtonsoft.Json.13.0.1\lib\net45\Newtonsoft.Json.dll</HintPath>
+    </Reference>
     <Reference Include="System" />
     <Reference Include="System.Core" />
     <Reference Include="System.Xml.Linq" />

SyncTokenGnerator/packages.config

@@ -1,4 +1,5 @@
 <?xml version="1.0" encoding="utf-8"?>
 <packages>
   <package id="Microsoft.Azure.SqlDatabase.ElasticScale.Client" version="2.3.0" targetFramework="net472" />
+  <package id="Newtonsoft.Json" version="13.0.1" targetFramework="net472" />
 </packages>