support slotinfo 244

Signed-off-by: catcherwong <[email protected]>
catcherwong · Oct 4, 2024 · 7887675 · 7887675
1 parent 702d263
commit 7887675
Show file tree

Hide file tree

Showing 8 changed files with 46 additions and 1 deletion.
diff --git a/src/RDBCli/Callbacks/KeysOnlyCallback.cs b/src/RDBCli/Callbacks/KeysOnlyCallback.cs
@@ -183,5 +183,9 @@ public void ZAdd(byte[] key, double score, byte[] member)
         public void SetIdleOrFreq(int val)
         {
         }
+
+        public void SetSlotInfo(ulong slotId, ulong slotSize, ulong expireSlotSize)
+        {
+        }
     }
 }
diff --git a/src/RDBCli/Callbacks/MemoryCallback.cs b/src/RDBCli/Callbacks/MemoryCallback.cs
@@ -483,5 +483,9 @@ public void SetIdleOrFreq(int val)
         {
             _idleOrFreq = val;
         }
+
+        public void SetSlotInfo(ulong slotId, ulong slotSize, ulong expireSlotSize)
+        {
+        }
     }
 }
diff --git a/src/RDBParser/BinaryReaderRDBParser.cs b/src/RDBParser/BinaryReaderRDBParser.cs
@@ -88,6 +88,15 @@ public void Parse(string path)
                             }
                         }
 
+                        if (opType == Constant.OpCode.SLOTINFO)
+                        {
+                            var slotId = br.ReadLength();
+                            var slotSize = br.ReadLength();
+                            var expireSlotSize = br.ReadLength();
+                            _callback.SetSlotInfo(slotId, slotSize, expireSlotSize);
+                            continue;
+                        }
+
                         if (opType == Constant.OpCode.SELECTDB)
                         {
                             if (!isFirstDb)

diff --git a/src/RDBParser/Callbacks/DefaultConsoleReaderCallBack.cs b/src/RDBParser/Callbacks/DefaultConsoleReaderCallBack.cs
@@ -91,6 +91,11 @@ public void SetIdleOrFreq(int val)
             Console.WriteLine($"SetIdleOrFreq, val={val}");
         }
 
+        public void SetSlotInfo(ulong slotId, ulong slotSize, ulong expireSlotSize)
+        {
+            Console.WriteLine($"SetSlotInfo, slotId={slotId}, slotSize={slotSize}, expireSlotSize={expireSlotSize}");
+        }
+
         public void StartDatabase(int database)
         {
             Console.WriteLine($"Start database = {database}");

diff --git a/src/RDBParser/Callbacks/IReaderCallback.cs b/src/RDBParser/Callbacks/IReaderCallback.cs
@@ -223,5 +223,13 @@ public interface IReaderCallback
         /// </summary>
         /// <param name="val"></param>
         void SetIdleOrFreq(int val);
+
+        /// <summary>
+        /// Record Slot Info
+        /// </summary>
+        /// <param name="slotId"></param>
+        /// <param name="slotSize"></param>
+        /// <param name="expireSlotSize"></param>
+        void SetSlotInfo(ulong slotId, ulong slotSize, ulong expireSlotSize);
     }
 }
diff --git a/src/RDBParser/Callbacks/NoOpReaderCallBack.cs b/src/RDBParser/Callbacks/NoOpReaderCallBack.cs
@@ -72,6 +72,10 @@ public void SetIdleOrFreq(int val)
         {
         }
 
+        public void SetSlotInfo(ulong slotId, ulong slotSize, ulong expireSlotSize)
+        {
+        }
+
         public void StartDatabase(int database)
         {
         }

diff --git a/src/RDBParser/Constant.cs b/src/RDBParser/Constant.cs
@@ -1,4 +1,6 @@
-namespace RDBParser
+using System.Dynamic;
+
+namespace RDBParser
 {
     public static class Constant
     {
@@ -31,6 +33,10 @@ public static class MagicCount
 
         public static class OpCode
         {
+            /// <summary>
+            /// Individual slot info, such as slot id and size (cluster mode only).
+            /// </summary>
+            public const int SLOTINFO = 244;
             public const int FUNCTION2 = 245;
             public const int FUNCTION = 246;
             public const int MODULE_AUX = 247;

diff --git a/tests/RDBParserTests/TestReaderCallback.cs b/tests/RDBParserTests/TestReaderCallback.cs
@@ -346,5 +346,10 @@ public void SetIdleOrFreq(int val)
         {
             _idleOrFreq = val;
         }
+
+        public void SetSlotInfo(ulong slotId, ulong slotSize, ulong expireSlotSize)
+        {
+            System.Diagnostics.Trace.WriteLine($"{slotId}, {slotSize}, {expireSlotSize}");
+        }
     }
 }