java.io.ByteArrayOutputStream 源码分析

ByteArrayOutputStream 内部包含了一个缓冲区，缓冲区会随着数据的不断写入而自动增长，俗称内存流。

 

首先看一下俩个属性，buf是内部缓冲区，count是记录写入了多少个字节。

protected byte buf[];
protected int count;

 

 

这个方法会在每次写入前都调用，检查buf的容量，如果缓冲区不够了会进行扩容。

    private void ensureCapacity(int minCapacity) {
        // overflow-conscious code
        if (minCapacity - buf.length > 0)
            grow(minCapacity);
    }

 

扩容的方法。

    private void grow(int minCapacity) {
　　　　int oldCapacity = buf.length;
　　　　//新的长度是原来的长度乘2  <<左移一位等于乘2
　　　　int newCapacity = oldCapacity << 1;
　　　　//如果新的长度仍然不够那么将minCapactity设置为缓冲区的大小
　　　　if (newCapacity - minCapacity < 0)
            newCapacity = minCapacity;
　　　　if (newCapacity < 0) {
　　　　　　if (minCapacity < 0) // overflow 判断溢出
　　　　　　　　throw new OutOfMemoryError();
             newCapacity = Integer.MAX_VALUE;
      }
　　　  //返回一些新的缓冲区，并且拷贝原来的字节数据。
       buf = Arrays.copyOf(buf, newCapacity);
    }

 

写入方法会将字节写入到缓冲区当中.

    public synchronized void write(int b) {
        ensureCapacity(count + 1);
        buf[count] = (byte) b;
        count += 1;
    }

    public synchronized void write(byte b[], int off, int len) {
        if ((off < 0) || (off > b.length) || (len < 0) ||
            ((off + len) - b.length > 0)) {
            throw new IndexOutOfBoundsException();
        }
        ensureCapacity(count + len);
        System.arraycopy(b, off, buf, count, len);
        count += len;
    }