caculate capacity in the default constructor of InternalThreadLocalMap (

#8697)
apache · Sep 11, 2021 · d173d16 · d173d16
1 parent cfece57
commit d173d16
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/dubbo-common/src/main/java/org/apache/dubbo/common/threadlocal/InternalThreadLocalMap.java b/dubbo-common/src/main/java/org/apache/dubbo/common/threadlocal/InternalThreadLocalMap.java
@@ -125,11 +125,23 @@ public int size() {
     }
 
     private static Object[] newIndexedVariableTable() {
-        Object[] array = new Object[32];
+        int variableIndex = NEXT_INDEX.get();
+        int newCapacity = variableIndex < 32 ? 32 : newCapacity(variableIndex);
+        Object[] array = new Object[newCapacity];
         Arrays.fill(array, UNSET);
         return array;
     }
 
+    private static int newCapacity(int index) {
+        int newCapacity = index;
+        newCapacity |= newCapacity >>> 1;
+        newCapacity |= newCapacity >>> 2;
+        newCapacity |= newCapacity >>> 4;
+        newCapacity |= newCapacity >>> 8;
+        newCapacity |= newCapacity >>> 16;
+        return ++newCapacity;
+    }
+
     private static InternalThreadLocalMap fastGet(InternalThread thread) {
         InternalThreadLocalMap threadLocalMap = thread.threadLocalMap();
         if (threadLocalMap == null) {
@@ -151,14 +163,7 @@ private static InternalThreadLocalMap slowGet() {
     private void expandIndexedVariableTableAndSet(int index, Object value) {
         Object[] oldArray = indexedVariables;
         final int oldCapacity = oldArray.length;
-        int newCapacity = index;
-        newCapacity |= newCapacity >>> 1;
-        newCapacity |= newCapacity >>> 2;
-        newCapacity |= newCapacity >>> 4;
-        newCapacity |= newCapacity >>> 8;
-        newCapacity |= newCapacity >>> 16;
-        newCapacity++;
-
+        int newCapacity = newCapacity(index);
         Object[] newArray = Arrays.copyOf(oldArray, newCapacity);
         Arrays.fill(newArray, oldCapacity, newArray.length, UNSET);
         newArray[index] = value;