io流(文件流的重复读取)

ゞ浴缸里的玫瑰 2022-11-13 13:23 278阅读 0赞

## 场景 ##

传统的io流是无法重复读取的。  
但是我们经常会用到一些需要重复利用流的场景，例如上传图片并解析文件。

## 过程 ##

大部分的IO流是无法重复读取的，只能读取一次，再读取时，会抛出异常，我们可以使用ByteArrayOutputStream将流数据缓存到内存中，达到多次读取的目的。

1、将不可重复的IO输入流读取到ByteArrayOutputStream中，ByteArrayOutputStream是一个可重复使用的输出流。

private ByteArrayOutputStream getBOS(InputStream in) { 
    	ByteArrayOutputStream bos = new ByteArrayOutputStream(); 
    	try { 
    		BufferedInputStream br = new BufferedInputStream(in);
    		for(int c=0;(c=br.read())!=-1;){ 
    			bos.write(c);
    		}
    		br.close();
    	} catch (Exception e) { 
    		logger.error("",e);
    	} 
    	return bos;
    }

2、生成一个新的可重复读取的InputStream

这个时候的bis是可以被多次重复读取，close对其无效，但是要注意每次读前，调用bis.reset();方法需要将游标重置到流的头部。

InputStream bis = new ByteArrayInputStream(bos.toByteArray());

### 尝试代码 ###

读写一次，内容正确。代码：

public static void main(String[] args) throws Exception{ 
        String fileName="in.txt";
        String outFileName="out.txt";
        try { 
            FileInputStream in = new FileInputStream("d://" + fileName);
            FileOutputStream out = new FileOutputStream("d://" + outFileName);
            for(int c=0;(c=in.read())!=-1;){ 
                out.write(c);
            }
            out.close();
            in.close();
        } catch (FileNotFoundException e) { 
            e.printStackTrace();
        }
    }

读之后写2次，第二次写为空(out2.txt没内容)，因为常规文件流不能复用。代码：

public static void main(String[] args) throws Exception{ 
        String fileName="in.txt";
        String outFileName="out.txt";
        String outFileName2="out2.txt";
        try { 
            FileInputStream in = new FileInputStream("d://" + fileName);
            FileOutputStream out = new FileOutputStream("d://" + outFileName);
            FileOutputStream out2 = new FileOutputStream("d://" + outFileName2);
            // out的写入
            for(int c=0;(c=in.read())!=-1;){ 
                out.write(c);
            }
    
            // out2的写入
            for(int c=0;(c=in.read())!=-1;){ 
                out2.write(c);
            }
    
            out.close();
            out2.close();
            in.close();
    
    
        } catch (FileNotFoundException e) { 
            e.printStackTrace();
        }
    }

使用可复用的文件流，2次写入均成功：

public static void main(String[] args) throws Exception{ 
    String fileName="in.txt";
    String outFileName="out.txt";
    String outFileName2="out2.txt";
    try { 
    
        FileInputStream fileInputStream = new FileInputStream("d://" + fileName);
        InputStream in = new ByteArrayInputStream(getBOS(fileInputStream).toByteArray());
        FileOutputStream out = new FileOutputStream("d://" + outFileName);
        FileOutputStream out2 = new FileOutputStream("d://" + outFileName2);
        // out的写入
        for(int c=0;(c=in.read())!=-1;){ 
            out.write(c);
        }
        in.reset();
        // out2的写入
        for(int c=0;(c=in.read())!=-1;){ 
            out2.write(c);
        }
    
        out.close();
        out2.close();
        in.close();
    
    
    } catch (FileNotFoundException e) { 
        e.printStackTrace();
    }
    
    private static ByteArrayOutputStream getBOS(InputStream in) { 
        ByteArrayOutputStream bos = new ByteArrayOutputStream();
        try { 
            BufferedInputStream br = new BufferedInputStream(in);
            for(int c=0;(c=br.read())!=-1;){ 
                bos.write(c);
            }
            br.close();
        } catch (Exception e) { 
            logger.error("",e);
        }
        return bos;
    }