Merge pull request #38 from koculu/add-btree-node-size-fine-tuning-op…

…tions Add B+Tree Node and Leaf size fine tuning options.
koculu · Jun 16, 2023 · f8b01ad · f8b01ad
2 parents 04f5caa + 765f8b8
commit f8b01ad
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/src/ZoneTree/Options/ZoneTreeOptions.cs b/src/ZoneTree/Options/ZoneTreeOptions.cs
@@ -99,6 +99,16 @@ public sealed class ZoneTreeOptions<TKey, TValue>
     /// </summary>
     public BTreeLockMode BTreeLockMode { get; set; } = BTreeLockMode.NodeLevelMonitor;
 
+    /// <summary>
+    /// The B+Tree node size.
+    /// </summary>
+    public int BTreeNodeSize { get; set; } = 128;
+
+    /// <summary>
+    /// The B+Tree leaf size.
+    /// </summary>
+    public int BTreeLeafSize { get; set; } = 128;
+
     /// <summary>
     /// ZoneTree Logger.
     /// </summary>

diff --git a/src/ZoneTree/Segments/InMemory/MutableSegment.cs b/src/ZoneTree/Segments/InMemory/MutableSegment.cs
@@ -54,7 +54,9 @@ public MutableSegment(ZoneTreeOptions<TKey, TValue> options,
 
         BTree = new(Comparer,
             Options.BTreeLockMode,
-            indexOpProvider);
+            indexOpProvider,
+            Options.BTreeNodeSize,
+            Options.BTreeLeafSize);
 
         MarkValueDeleted = options.MarkValueDeleted;
         MutableSegmentMaxItemCount = options.MutableSegmentMaxItemCount;
@@ -73,7 +75,12 @@ public MutableSegment(
         Options = options;
         Comparer = options.Comparer;
 
-        BTree = new(Comparer, Options.BTreeLockMode);
+        BTree = new(
+            Comparer,
+            Options.BTreeLockMode,
+            null,
+            Options.BTreeNodeSize,
+            Options.BTreeLeafSize);
         BTree.SetNextOpIndex(nextOpIndex);
 
         MarkValueDeleted = options.MarkValueDeleted;