ArrayList源码分析（二）

在上一节讲ArrayList增加元素的源码时，还遗留了一个ensureCapacityInternal方法。

第一次调用add方法时

1、add方法如下：

public boolean add(E e) {
     ensureCapacityInternal(size + 1);  // Increments modCount!!
     elementData[size++] = e;
     return true;
}

size为0。
点击ensureCapacityInternal进入如下代码.

2、ensureCapacityInternal方法：

private void ensureCapacityInternal(int minCapacity) {
    ensureExplicitCapacity(calculateCapacity(elementData, minCapacity));
}

minCapacity为1。
点击calculateCapacity进入步骤3。
点击ensureExplicitCapacity进入步骤4。

3、calculateCapacity方法

private static int calculateCapacity(Object[] elementData, int minCapacity) {
    if (elementData == DEFAULTCAPACITY_EMPTY_ELEMENTDATA) {
         return Math.max(DEFAULT_CAPACITY, minCapacity);
    }
    return minCapacity;
}

其中minCapacity表示当前元素的size + 1，即为1。
DEFAULT_CAPACITY表示默认的大小10。
if条件判断数组是否为空，刚构造一个list时数组是为空的，这里返回10。

4、ensureExplicitCapacity方法

private void ensureExplicitCapacity(int minCapacity) {
    modCount++;

    // overflow-conscious code
    if (minCapacity - elementData.length > 0)
        grow(minCapacity);
}

这里的形参minCapacity为上面传入的DEFAULT_CAPACITY，即10。
点击grow方法进入如下代码：

5、grow方法

private void grow(int minCapacity) {
    // overflow-conscious code
      int oldCapacity = elementData.length;
      int newCapacity = oldCapacity + (oldCapacity >> 1);
      if (newCapacity - minCapacity < 0)
          newCapacity = minCapacity;
      if (newCapacity - MAX_ARRAY_SIZE > 0)
          newCapacity = hugeCapacity(minCapacity);
      // minCapacity is usually close to size, so this is a win:
      elementData = Arrays.copyOf(elementData, newCapacity);
}

minCapacity为10，oldCapacity为0。扩容后newCapacity还是为0，因此满足下面的判断，将newCapacity赋值为10.

if (newCapacity - minCapacity < 0)
    newCapacity = minCapacity;

MAX_ARRAY_SIZE 是一个非常大的数，这里我们不用管。

if (newCapacity - MAX_ARRAY_SIZE > 0)
       newCapacity = hugeCapacity(minCapacity);

6、总结

构造ArrayList时，数组初始为一个空数组。第一次调用add方法，会创建一个大小为10的数组。

后续调用add方法

这里以数组元素达到10触发第一次扩容为例。

1、add方法如下：

public boolean add(E e) {
     ensureCapacityInternal(size + 1);  // Increments modCount!!
     elementData[size++] = e;
     return true;
}

size为10。
点击ensureCapacityInternal进入如下代码.

2、ensureCapacityInternal方法：

private void ensureCapacityInternal(int minCapacity) {
    ensureExplicitCapacity(calculateCapacity(elementData, minCapacity));
}

minCapacity为11。
点击calculateCapacity进入步骤3。
点击ensureExplicitCapacity进入步骤4。

3、calculateCapacity方法

private static int calculateCapacity(Object[] elementData, int minCapacity) {
    if (elementData == DEFAULTCAPACITY_EMPTY_ELEMENTDATA) {
         return Math.max(DEFAULT_CAPACITY, minCapacity);
    }
    return minCapacity;
}

其中minCapacity表示当前元素的size + 1，即为11。
elementData 现在并不是一个空数组，故if条件不满足
直接返回minCapacity，即11.

4、ensureExplicitCapacity方法

private void ensureExplicitCapacity(int minCapacity) {
    modCount++;

    // overflow-conscious code
    if (minCapacity - elementData.length > 0)
        grow(minCapacity);
}

这里的形参minCapacity为上面返回minCapacity，即11.
点击grow方法进入如下代码：

5、grow方法

private void grow(int minCapacity) {
    // overflow-conscious code
      int oldCapacity = elementData.length;
      int newCapacity = oldCapacity + (oldCapacity >> 1);
      if (newCapacity - minCapacity < 0)
          newCapacity = minCapacity;
      if (newCapacity - MAX_ARRAY_SIZE > 0)
          newCapacity = hugeCapacity(minCapacity);
      // minCapacity is usually close to size, so this is a win:
      elementData = Arrays.copyOf(elementData, newCapacity);
}

minCapacity为11，oldCapacity为10。(oldCapacity >> 1)相当于扩容一半得newCapacity=15。
然后，将数据从老数组拷贝到新数组中

elementData = Arrays.copyOf(elementData, newCapacity);