MP-11740 Restrict TreeMap to 1 bucket

Author: sarveswaran-m

src/main/java/com/trivago/fastutilconcurrentwrapper/ConcurrentLongShortTreeMapBuilder.java

@@ -5,7 +5,9 @@
 
 public final class ConcurrentLongShortTreeMapBuilder {
     private MapMode mapMode = MapMode.BLOCKING;
-    private int buckets = 8;
+    // floorEntry() requires all keys to reside in a single bucket to return
+    // correct results, so the bucket count is fixed at 1.
+    private static final int BUCKETS = 1;
     private short defaultValue = LongShortMap.DEFAULT_VALUE;
 
     private ConcurrentLongShortTreeMapBuilder() {
@@ -15,10 +17,6 @@ public static ConcurrentLongShortTreeMapBuilder newBuilder() {
         return new ConcurrentLongShortTreeMapBuilder();
     }
 
-    public ConcurrentLongShortTreeMapBuilder withBuckets(int buckets) {
-        this.buckets = buckets;
-        return this;
-    }
 
     public ConcurrentLongShortTreeMapBuilder withMode(MapMode mapMode) {
         this.mapMode = mapMode;
@@ -39,15 +37,15 @@ public enum MapMode {
             @Override
             LongShortTreeMap createMap(ConcurrentLongShortTreeMapBuilder builder) {
                 return new ConcurrentBusyWaitingLongShortTreeMap(
-                        builder.buckets,
+                        BUCKETS,
                         builder.defaultValue);
             }
         },
         BLOCKING {
             @Override
             LongShortTreeMap createMap(ConcurrentLongShortTreeMapBuilder builder) {
                 return new ConcurrentLongShortTreeMap(
-                        builder.buckets,
+                        BUCKETS,
                         builder.defaultValue);
             }
         };

src/test/java/com/trivago/fastutilconcurrentwrapper/ConcurrentLongShortTreeMapBuilderTest.java

@@ -13,7 +13,6 @@ public class ConcurrentLongShortTreeMapBuilderTest {
     @Test
     public void buildsBusyWaitingMap() {
         ConcurrentLongShortTreeMapBuilder b = ConcurrentLongShortTreeMapBuilder.newBuilder()
-                .withBuckets(2)
                 .withDefaultValue(DEFAULT_VALUE)
                 .withMode(ConcurrentLongShortTreeMapBuilder.MapMode.BUSY_WAITING);
         LongShortTreeMap map = b.build();
@@ -27,7 +26,6 @@ public void buildsBusyWaitingMap() {
     @Test
     public void buildsBlockingMap() {
         ConcurrentLongShortTreeMapBuilder b = ConcurrentLongShortTreeMapBuilder.newBuilder()
-                .withBuckets(2)
                 .withDefaultValue(DEFAULT_VALUE)
                 .withMode(ConcurrentLongShortTreeMapBuilder.MapMode.BLOCKING);
         LongShortTreeMap map = b.build();