private (#248)

Author: bitfaster

BitFaster.Caching.UnitTests/Lfu/CmSketchTests.cs

@@ -16,7 +16,7 @@ public class CmSketchTests
         [Fact]
         public void WhenCapacityIsZeroDefaultsSelected()
         {
-            sketch.EnsureCapacity(0);
+            sketch = new CmSketch<int>(0, EqualityComparer<int>.Default);
 
             sketch.ResetSampleSize.Should().Be(10);
         }
@@ -55,7 +55,7 @@ public void WhenSampleSizeExceededCountIsReset()
         {
             bool reset = false;
 
-            sketch.EnsureCapacity(64);
+            sketch = new CmSketch<int>(64, EqualityComparer<int>.Default);
 
             for (int i = 1; i < 20 * 64; i++)
             {

BitFaster.Caching/Lfu/CmSketch.cs

@@ -47,12 +47,7 @@ public CmSketch(long maximumSize, IEqualityComparer<T> comparer)
         /// </summary>
         public int Size => this.size;
 
-        /// <summary>
-        /// Initialize such that the count min sketch can accurately estimate the count for values given
-        /// a maximum size.
-        /// </summary>
-        /// <param name="maximumSize">The maximum size.</param>
-        public void EnsureCapacity(long maximumSize)
+        private void EnsureCapacity(long maximumSize)
         {
             int maximum = (int)Math.Min(maximumSize, int.MaxValue >> 1);
 

BitFaster.Caching/Lfu/ConcurrentLfu.cs

@@ -87,8 +87,7 @@ public ConcurrentLfu(int concurrencyLevel, int capacity, IScheduler scheduler, I
             int writeBufferSize = Math.Min(BitOps.CeilingPowerOfTwo(capacity), 128);
             this.writeBuffer = new MpscBoundedBuffer<LfuNode<K, V>>(writeBufferSize);
 
-            this.cmSketch = new CmSketch<K>(1, comparer);
-            this.cmSketch.EnsureCapacity(capacity);
+            this.cmSketch = new CmSketch<K>(capacity, comparer);
             this.windowLru = new LfuNodeList<K, V>();
             this.probationLru = new LfuNodeList<K, V>();
             this.protectedLru = new LfuNodeList<K, V>();