JAVA IO源码学习系列一（ByteArrayOutputStream）

介绍

1. 输出流：ByteArrayOutputStream

上一篇简单介绍了输出流的超类OutputStream,也大概的讲述了输出流的作用，本篇就介绍一下，输出流的一种实现，字节数组输出流，该输出流是为了处理字节的基础流，本质上就是写入数据到类中的缓冲字节数组中;

2. ByteArrayOutputStream 源代码介绍

(1)属性内容：属性内容相较输入流的会有存放数据的缓冲区，也就是字节数组；

count：写操作的时候的计数，也可以算作数据的大小

// 存储数据的缓冲区(字节数组)。
protected byte buf[];
//缓冲区中的有效字节数(写入的内容大小，也可以看做是位置)
protected int count;

（2）构造函数：默认的构造函数会初始化缓冲区大小为32个字节，即缓冲数组的大小为32；

//默认构造函数
public ByteArrayOutputStream() {
this(32);
}

//指定的缓冲区大小
public ByteArrayOutputStream(int size) {
if (size < 0) {
throw new IllegalArgumentException("Negative initial size: "
+ size);
}
buf = new byte[size];
}

（3）主要方法：输出流的主要方法当然就是写，ByteArrayOutputStream中有三种写方法，第一是写入一个int 数据以字节的形式存到缓冲数组中，第二个就是写入指定的字节数组，可以指定从该数组的哪个位置开始写，和写多少；最后一个是写到指定的输出流中，相当于继续流，

// 将指定的字节写入此 byte 数组输出流。
public synchronized void write(int b) {
ensureCapacity(count + 1);
buf[count] = (byte) b;
count += 1;
}
//将指定 byte 数组中从偏移量 off 开始的 len 个字节写入此 byte 数组输出流。
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;
}
//将此 byte 数组输出流的全部内容写入到指定的输出流参数中
public synchronized void writeTo(OutputStream out) throws IOException {
out.write(buf, 0, count);
}

扩容方法：
在上面的写方法中我们可以看到其中还有一个中间处理的方法，也就是扩容的方法，在数据写入的过程中，每次会判断当前的缓冲区容量是否够写入，如果不够就两倍扩容；

//判断是否需要扩容
private void ensureCapacity(int minCapacity) {
// overflow-conscious code
if (minCapacity - buf.length > 0)
grow(minCapacity);
}

//进行扩容操作
private void grow(int minCapacity) {
int oldCapacity = buf.length;
int newCapacity = oldCapacity << 1;
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);
}

（4）其他方法介绍

reset():重置方法仅仅是将count值设置为0，这样写数据的时候就可以重新从0开始写，此时缓冲区的数据其实还存在；

public synchronized void reset() {
count = 0;
}

size():返回当前写入了多少数据，其实就是count的值；

public synchronized int size() {
return count;
}

toByteArray():以字节数组的形式返回当前缓冲数组中的数据，当然位置是从0，到count,所以reset操作之后即使缓冲数组中的数据很多，但只显示到count;

public synchronized byte toByteArray()[] {
return Arrays.copyOf(buf, count);
}

还有toString()的方法：字符串显示，类似toByteArray();

3. 附上自己重写的代码

只实现核心的方法功能，没有继承其他类；

//字节数组输出流
public class MyByteArrayOutputStream {

//缓冲的字节数组
protected byte[] buffer;

//输出流的内容大小，写入的大小
protected int counts;

//默认构造初始化时定义32字节大小的缓冲数组空间
public MyByteArrayOutputStream() {
this(32);
}
//创建指定大小的缓冲数组空间
public MyByteArrayOutputStream(int size) {
if(size<0){
throw new IllegalArgumentException("size 值必须大于0"+size);
}
buffer = new byte[size];
}

//输出流写入一个int b :即将数据写到缓冲数组中
public synchronized void write(int b){
//每次写数据前判断当前缓冲数组空间是否够写入，不够则进行扩容
ensureCapacity(counts+1);
buffer[counts] = (byte)b;
//统计写入的大小
counts+=1;
}

//从指定的字节数组向输出流中写数据，可以指定数组中的啥位置开始写入，和写多少
public synchronized void write(byte[] b,int offset,int length){
if(b==null){
throw new NullPointerException();
}else if((offset < 0) ||(b.length<offset)|| length < 0 || (b.length<length+offset)){
throw new IndexOutOfBoundsException("写入的大小不正常，越界了");
}
//每次写数据前判断当前缓冲数组空间是否够写入，不够则进行扩容
ensureCapacity(counts+length);
//将指定的数据写入到缓冲数组中
System.arraycopy(b, offset, buffer, counts, length);
//统计写入的大小
counts+=length;

}

//将缓冲数组输出流中的数据写入指定的输出流中
public synchronized void writeTo(OutputStream out) throws IOException{
//类似回调方法，写入指定输出流中
out.write(buffer,0,counts);

}

//返回字节数组：即将当前缓冲数组中的数据以字节数组的形式返回
public synchronized byte[] toByteArray(){
return Arrays.copyOf(buffer, counts);
}
//返回字符串：即将当前缓冲数组中的数据以字符串的形式返回
public synchronized String toString(){
return new String(buffer,0,counts);
}
//返回字符串：即将当前缓冲数组中的数据以字符串的形式返回，指定了字符集
public synchronized String toString(String charsetName) throws UnsupportedEncodingException{

return new String(buffer,0,counts,charsetName);
}

//返回当前输出流的大小，即缓冲数组中写入的数据
public synchronized int size(){
return counts;
}

//判断当前是否需要扩容：即比较当前的容量是否大于新增的数量，不够则进行扩容
private void ensureCapacity(int capacity){

if(capacity > buffer.length){
growCapacity(capacity);
}
}

//扩容方法：两倍当前容量进行扩容，如果两倍容量还小于需要的容量，则使用需要的容量，当然最大不能超过数组的最大容量
private void growCapacity(int capacity){
int oldCapacity = buffer.length;
int newCapacity = oldCapacity << 1;
if(newCapacity < capacity){
newCapacity = capacity;
}
if(newCapacity < 0){
if(capacity<0){
throw new IndexOutOfBoundsException("增加的容量不正常");
}
newCapacity=Integer.MAX_VALUE;
}
buffer = Arrays.copyOf(buffer, newCapacity);
}

//重置：将写入的位置设置为0，这样就可以从0位置写数据，重置后缓冲数组中可能存在之前的数据
public synchronized void reset(){
counts = 0;
}
}

4.最后召唤神兽

/**
*
*                                                    __----~~~~~~~~~~~------___
*                                   .  .   ~~//====......          __--~ ~~
*                   -.            \_|//     |||\\  ~~~~~~::::... /~
*                ___-==_       _-~o~  \/    |||  \\            _/~~-
*        __---~~~.==~||\=_    -_--~/_-~|-   |\\   \\        _/~
*    _-~~     .=~    |  \\-_    '-~7  /-   /  ||    \      /
*  .~       .~       |   \\ -_    /  /-   /   ||      \   /
* /  ____  /         |     \\ ~-_/  /|- _/   .||       \ /
* |~~    ~~|--~~~~--_ \     ~==-/   | \~--===~~        .\
*          '         ~-|      /|    |-~\~~       __--~~
*                      |-~~-_/ |    |   ~\_   _-~            /\
*                           /  \     \__   \/~                \__
*                       _--~ _/ | .-~~____--~-/                  ~~==.
*                      ((->/~   '.|||' -_|    ~~-/ ,              . _||
*                                 -_     ~\      ~~---l__i__i__i--~~_/
*                                 _-~-__   ~)  \--______________--~~
*                               //.-~~~-~_--~- |-------~~~~~~~~
*                                      //.-~~~--\
*                               神兽保佑
*                              代码无BUG!