Minor updates to finding storage locations for grains

OrleansContrib · Jan 30, 2024 · ad2fe1b · ad2fe1b
1 parent 2fbde87
commit ad2fe1b
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 10 deletions.
diff --git a/OrleansShardedStorageProvider/AzureShardedGrainBase.cs b/OrleansShardedStorageProvider/AzureShardedGrainBase.cs
@@ -18,7 +18,7 @@ public AzureShardedGrainBase(string serviceId, AzureShardedStorageOptions option
             _options = options;
         }
 
-        protected int GetShardNumberFromKey(string pk)
+        public int GetShardNumberFromKey(string pk)
         {
             var hash = GetStableHashCode(pk);
             var storageNum = Math.Abs(hash % this._options.ConnectionStrings.Count());
@@ -51,7 +51,12 @@ protected int GetStableHashCode(string str)
         }
 
 
-        private const string KeyStringSeparator = "__";
+        protected const string KeyStringSeparator = "__";
+
+        public string GetKeyStringSeparator() 
+        { 
+            return KeyStringSeparator; 
+        }
 
         protected string GetKeyString(GrainId grainId)
         {

diff --git a/ZDataFinder/Program.cs b/ZDataFinder/Program.cs
@@ -41,6 +41,32 @@
 options.ConnectionStrings.AddRange(blobGrainStores);
 
 
+
+var serviceId = "OrleansBasics";
+var grainName = "smalldata";
+var grainId = "1F";
+AzureShardedGrainBase agb = new AzureShardedGrainBase(serviceId, options);
+
+var key = $"{serviceId}{agb.GetKeyStringSeparator()}{grainName}_{grainId}";
+var storageNumberFromKey = agb.GetShardNumberFromKey(key);
+Console.WriteLine($"storage location from key = {storageNumberFromKey}");
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 StorageDataFinder finder = new StorageDataFinder();
 await finder.Init(options);
 
@@ -49,7 +75,7 @@
 
 do
 {
-    Console.WriteLine("Enter the guid you wish to find the data location of");
+    Console.WriteLine("Enter the all (or part of) the guid you wish to find the data location of");
     input = Console.ReadLine();
 }
 while (String.IsNullOrWhiteSpace(input));

diff --git a/ZDataFinder/StorageDataFinder.cs b/ZDataFinder/StorageDataFinder.cs
@@ -13,6 +13,7 @@
 using Azure.Storage.Blobs.Models;
 using System.Collections.Concurrent;
 using Microsoft.Identity.Client;
+using System.Xml;
 
 namespace ZDataFinder
 {
@@ -105,7 +106,7 @@ await Parallel.ForEachAsync(this._blobClients, options, async (blobClient, ct) =
                             {
                                 retVal.Add($"{blobClient.AccountName} / {blobItem.Name}");
 
-                                if(exitAtFirstResult)
+                                if (exitAtFirstResult)
                                 {
                                     Console.WriteLine("Found something. Exiting parallel foreach...");
                                     source.Cancel();
@@ -122,16 +123,11 @@ await Parallel.ForEachAsync(this._blobClients, options, async (blobClient, ct) =
             {
                 // ... (the cts was cancelled)
             }
-            catch(Exception e)
+            catch (Exception e)
             {
                 Console.WriteLine(e.ToString());
             }
 
-
-
-
-
-
             return retVal;
         }
     }