java.io.ByteArrayOutputStream 源码分析

java

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;

}

以上是 java.io.ByteArrayOutputStream 源码分析 的全部内容, 来源链接: utcz.com/z/390685.html

回到顶部