在上一节讲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);