From 3ea1ffcc89008cc5157bcaebd2bf85212f44b981 Mon Sep 17 00:00:00 2001
From: jhorvitz <jhorvitz@google.com>
Date: Thu, 29 Apr 2021 07:26:33 -0700
Subject: [PATCH] Allow subclasses of `InMemoryGraphImpl` to specify an initial
 capacity.

PiperOrigin-RevId: 371115514
---
 .../devtools/build/skyframe/InMemoryGraphImpl.java   | 12 ++++++------
 1 file changed, 6 insertions(+), 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
index a69c228d6fdc5e..01e1e7036efa82 100644
--- 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;
-  }
 }