BufferedInputStream实现原理分析

转载自：http://diaocow.iteye.com/blog/1746210

高我们的读取效率，现在我们看下BufferedInputStream的实现原理： 
BufferedInputStream内部有一个缓冲区，默认大小为8M，每次调用read方法的时候，它首先尝试从缓冲区里读取数据，若读取失败（缓冲区无可读数据），则选择从物理数据源（譬如文件）读取新数据（这里会尝试尽可能读取多的字节）放入到缓冲区中，最后再将缓冲区中的内容部分或全部返回给用户.由于从缓冲区里读取数据远比直接从物理数据源（譬如文件）读取速度快，所以BufferedInputStream的效率很高！ 


在具体看源码之前，我们还需要了解BufferedInputStream的mark操作：void mark（int markLimit) 
当你调用mark方法时，内部会保存一个markPos标志，它的值为目前读取字节流的pos位置，倘若你调用reset方法，这时候会把pos重置为markPos的值，这样你就可以重读已经读过的字节.好像说的不是很清楚，那我们打个比方：有一段字节流是abcdefg, 当你读取完字母a调用mark方法（此时markPos指向字母b），接着你继续读取字母b，字母c，字母d，然后此时你调用reset方法（内部把pos重置为markPos)，当你再读取下一个字节的时候，你会发现你读取到的是b而不是字母e，这样通过mark方法我们就是实现了重复读（re-read the same bytes) 

mark方法中还有个参数markLimit，它的值表示在调用mark方法后reset方法前最多允许读取的字节数（根据我的测试，以及查看源代码发现，这个最大字节数，其实是由markLimit和buffer.size中较大的那个决定的），如果超过这个限制，则在调用reset方法时会报：Reseting to invalid mark 异常 


说完了这么多，让我们来赶紧看看源码吧： 

Java代码  

1. // BufferedInputStream主要有这两个构造方法  
2.   
3. public BufferedInputStream(InputStream in) {  
4.      this(in, defaultBufferSize);   // 默认缓冲区为8M  
5.     }  
6.   
7. public BufferedInputStream(InputStream in, int size) {  
8.     super(in);  
9.     if (size <= 0) {  
10.         throw new IllegalArgumentException("Buffer size <= 0");  
11.     }  
12.     buf = new byte[size];  
13. }  

你需要指定InputStream（装饰模式的体现）以及bufferSize(可选） 

当我们调用read()方法时，它在内部做了一下事情： 

Java代码  

1. public synchronized int read() throws IOException {  
2.     if (pos >= count) {         // 检查是否有可读缓冲数据  
3.         fill();                 // 没有缓冲数据可读，则从物理数据源读取数据并填充缓冲区  
4.         if (pos >= count)       // 若物理数据源也没有多于可读数据，则返回-1，标示EOF  
5.         return -1;  
6.     }  
7.   
8.  // 从缓冲区读取buffer[pos]并返回（由于这里读取的是一个字节，而返回的是整型，所以需要把高位置0）  
9.     return getBufIfOpen()[pos++] & 0xff;     
10. }  
11.   
12. private byte[] getBufIfOpen() throws IOException {  
13.     byte[] buffer = buf;      // buf为内部缓冲区  
14.     if (buffer == null)  
15.         throw new IOException("Stream closed");  
16.         return buffer;  
17. }  

其中pos为缓冲区buffer下一个可读的数组下标，我们可以一直从缓冲区里读取数据，直到pos变为count（此时只能从物理数据源读取数据），下面我们就分析下，当缓冲区里没有数据可读时，BufferInputStream是如何处理的： 

1.若用户没有开启re-read功能（即未调用mark方法），当pos == count时，我们只需要将pos重新置为0，然后从物理源读取数据（假设读到了n个字节），最后把count设置成 n + pos 即可 (其实就是n，因为pos之前被设置成了0）, 当下次你在调用read方法时，就直接从缓冲读取，非常快速（如下图）； 


2.若用户调用了mark方法，情况就变得很复杂了，为什么呢？ 这意味着我们需要保存从markPos到pos这段数据（以供用户re-read），现在我们分情况讨论： 

a.此时pos < buffer.length，这意味着缓冲区还有多余空间，所以我们可以继续从物理数据源读取数据放入到缓冲区中（如下图）； 


b.pos == buffer.length，这意味着缓冲区已经没有多余空间，所以只能清空缓冲区内容，但是不要忘了，我们还必须保留原来 
markPos到pos那段数据，以供用户re-read，所以我需要这样做： 

Java代码  

1. // 计算需要保留多少字节的数据  
2. int sz = pos - markPos;     
3. // 然后拷贝到缓冲头部                                
4. System.arraycopy(buffer, markpos, buffer, 0, sz);   
5. // 重置pos以及markPos        
6. pos=sz;                                                                                  
7. markPos=0;  

                                   
接着从缓冲区的sz 到 buffer.length又变成可用区间，用来存放从物理数据源读到的数据（如下图） 


到这里似乎问题完美的解决了，其实不然，我们忘记考虑markPos失效问题，以及若pos - markPos == buffer.length，那么移了等于白移动，还是没有挪出多余空间，所以实我们应该这样做（后面讨论都是建立在pos == buffer.length的基础上）： 


2.1 若markPos > 0， 那么 pos - makrPos一定小于缓冲区大小，这样意味着我们按照刚才的算法挪动后，缓冲区就有了空余空间 

2.2 若makrPos == 0,  这意味着需要保存的数据满满的充斥着缓冲区，所以这时候我们是无法通过挪动位置来使缓冲区有多余空间的，所以我们只可以清空或扩展缓冲区 

2.2.1 当buffer.length >= marklimit时 ，此时意味着markPos已经失效，用户不可以在进行re-read，所以此时我们就可以简单释放整个缓冲区了：pos=0, markPos=-1; 

2.2.2 其余情况，意味着markPos还有效，所以我们只能通过扩展缓冲区的方式来使缓冲区有多余空间 

说了这么多，我们还是看下相关代码吧： 

Java代码  

1. private void fill() throws IOException {  
2.         byte[] buffer = getBufIfOpen();        // 得到当前缓冲区  
3.      if (markpos < 0)                          // 对应情况1  
4.          pos = 0;           
5.      else if (pos >= buffer.length)            // 对应情况2  
6.          if (markpos > 0) {                    // 对应情况2.1  
7.                int sz = pos - markpos;  
8.                System.arraycopy(buffer, markpos, buffer, 0, sz);  
9.                pos = sz;  
10.                markpos = 0;  
11.          } else if (buffer.length >= marklimit) {  // 对应情况2.2.1  
12.                markpos = -1;      
13.                pos = 0;      
14.          } else {                                  // 对应情况2.2.2  
15.                int nsz = pos * 2;  
16.                if (nsz > marklimit)  
17.                    nsz = marklimit;  
18.                byte nbuf[] = new byte[nsz];  
19.                System.arraycopy(buffer, 0, nbuf, 0, pos);  
20.             ...  
21.          }  
22.     count = pos;  
23.      int n = getInIfOpen().read(buffer, pos, buffer.length - pos);  
24.     if (n > 0)  
25.         count = n + pos;  
26. }  

好了关于BufferedInputStream就说道这里，它的 read(byte b[], int off, int len)其实内部实现也大概如此：先从缓冲区读，如果读不到则从物理数据源读取并刷新到缓冲区（可能需要对缓冲区原来内容作必要的挪动或者对缓冲区大小进行扩展）