Allow subclasses of InMemoryGraphImpl to specify an initial capacity.

PiperOrigin-RevId: 371115514
ptarjan · Apr 29, 2021 · 3ea1ffc · 3ea1ffc
1 parent b87e94d
commit 3ea1ffc
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/main/java/com/google/devtools/build/skyframe/InMemoryGraphImpl.java b/src/main/java/com/google/devtools/build/skyframe/InMemoryGraphImpl.java
@@ -34,7 +34,7 @@
  */
 public class InMemoryGraphImpl implements InMemoryGraph {
 
-  protected final ConcurrentMap<SkyKey, NodeEntry> nodeMap = new ConcurrentHashMap<>(1024);
+  protected final ConcurrentMap<SkyKey, NodeEntry> nodeMap;
   private final boolean keepEdges;
 
   @VisibleForTesting
@@ -43,7 +43,12 @@ public InMemoryGraphImpl() {
   }
 
   public InMemoryGraphImpl(boolean keepEdges) {
+    this(keepEdges, /*initialCapacity=*/ 1 << 10);
+  }
+
+  protected InMemoryGraphImpl(boolean keepEdges, int initialCapacity) {
     this.keepEdges = keepEdges;
+    this.nodeMap = new ConcurrentHashMap<>(initialCapacity);
   }
 
   @Override
@@ -120,9 +125,4 @@ public Map<SkyKey, NodeEntry> getAllValues() {
   public Map<SkyKey, ? extends NodeEntry> getAllValuesMutable() {
     return nodeMap;
   }
-
-  @VisibleForTesting
-  protected ConcurrentMap<SkyKey, ? extends NodeEntry> getNodeMap() {
-    return nodeMap;
-  }
 }