java集合之ArrayList扩容机制

ArrayList是工作中经常都要用到的一个集合，今天我们来看看ArrayList是怎么扩容的。

一、ArrayList有三个构造函数

1. 无参构造

public ArrayList() {
    this.elementData = DEFAULTCAPACITY_EMPTY_ELEMENTDATA;
}

无参构造函数只创建了一个空的数组。
2. 有参构造函数

public ArrayList(int initialCapacity) {
		//判断传入的参数是否大于0
        if (initialCapacity > 0) {
        	//创建长度为传入参数的对象数组
            this.elementData = new Object[initialCapacity];
        } else if (initialCapacity == 0) {
        	//等于0赋值空数组
            this.elementData = EMPTY_ELEMENTDATA;
        } else {
        	//否则抛出异常
            throw new IllegalArgumentException("Illegal Capacity: "+
                    initialCapacity);
        }
    }

第二个有参构造

public ArrayList(Collection<? extends E> c) {
		//把传过来的集合转成数组
        elementData = c.toArray();
        //判断转换后的数组长度
        if ((size = elementData.length) != 0) {
            //不等0，判断是否是对象数组
            if (elementData.getClass() != Object[].class)
            	//不是的话复制到对象数组并复制给elementData 
                elementData = Arrays.copyOf(elementData, size, Object[].class);
        } else {
            // replace with empty array.
            //空的话直接赋值对数组
            this.elementData = EMPTY_ELEMENTDATA;
        }
    }

然后扩容机制的话我们主要看它的Add方法就行了

public boolean add(E e) {
        //确认list容量，如果不够，容量加1。注意：只加1，保证资源不被浪费
        ensureCapacityInternal(size + 1);  // Increments modCount!!
        //把元素放进数组中,并++
        elementData[size++] = e;
        return true;
    }

然后看ensureCapacityInternal这个方法

    private void ensureCapacityInternal(int minCapacity) {
        //数组容量检查，不够时则进行扩容
        ensureExplicitCapacity(calculateCapacity(elementData, minCapacity));
    }

calculateCapacity这个方法主要是计算容量的

    private static int calculateCapacity(Object[] elementData, int minCapacity) {
        // 若elementData == DEFAULTCAPACITY_EMPTY_ELEMENTDATA，则取minCapacity为DEFAULT_CAPACITY和参数minCapacity之间的最大值。
        // DEFAULT_CAPACITY在此之前已经定义为默认的初始化容量是10。
        if (elementData == DEFAULTCAPACITY_EMPTY_ELEMENTDATA) {
            return Math.max(DEFAULT_CAPACITY, minCapacity);
        }
        return minCapacity;
    }

然后回到上一层看这个方法ensureExplicitCapacity

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

        // overflow-conscious code
        //最小容量>数组缓冲区当前长度
        if (minCapacity - elementData.length > 0)
            //扩容
            grow(minCapacity);
    }

grow这个就是主要的扩容方法

private void grow(int minCapacity) {
        // overflow-conscious code
        //获取当前容器的容量
        int oldCapacity = elementData.length;
        //新容量等于旧容量的1.5倍
        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:
        //新的容量大小已经确定好了，就copy数组，改变容量大小。
        elementData = Arrays.copyOf(elementData, newCapacity);
    }

这个方法主要是看确定容量大小然后复制数组，赋值给elementData
执行完之后就会回到最外层进行赋值操作

    public boolean add(E e) {
        //确认list容量，如果不够，容量加1。注意：只加1，保证资源不被浪费
        ensureCapacityInternal(size + 1);  // Increments modCount!!
        //把元素放进数组中,并++
        elementData[size++] = e;
        return true;
    }

最后返回true，添加成功。