欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

InputStream复用问题

程序员文章站 2022-05-25 19:05:45
...

【问题】:
后台获取了前台传输过来的文件,先判断文件里的内容是否正确,正确之后,再将文件上传到服务器。
但是当我拿到文件的内容之后,上传到服务器的文件内容却是空的。代码如下:

FileIoUtil.java
class FileIoUtil{
//获取文件里的内容
  public List<String> getFileIo(<List<?> fileList){
   Workbook wb - null;
   List<String> parm = new ArrayList<String>();
   //一
   InputStream ins =  FileUtil.getInputStreamFromFileUpload(((List<FileUpload>fileList).get(0));
   try{
      wb = WorkbookFactory.create(ins);
      Sheet dataSheet = wb.getSheetAt(0);
      int rownum = dataSheet.getLastRowNum();
      for(int i=0;i<rownum;i++){

        ....
        //一系列操作,将excel文件里的内容加入到list——parm中
      }
   }catch(Exception e){

      ...
   }finally {
			if (ins != null) {
				try {
					ins.close();
				} catch (IOException e) {
				}
            }
   }

//上传文件到服务器
public void uploadFileToService(String path,String fileName,InputStream ins){

  //...

}


Test1.java
class Test1{
   /**
   *fileList就是前台传过来的文件信息
   */
   public void uploadFile(List<?> fileList){
      List<String> param = FileIoUtil.getFileIo(fileList);
      //判断是否有重复的内容
      param = findRepeatedRecord(param);
      //证明有
      if(param.size()>0){
        reutrn;
      }else{
     //没有重复,则需要上传
        String  path  = "E:\\test\\";
        FileUpload fileUpload = (FileUpload)fileList.get(0);
        String  fileName = fileUpload.getFileName();
        //二
        InputStream ins = FileUtil.getInputStreamFromFileUpload(fileUpload);
        //上传文件
			FileIoUtil.uploadFileToService(path, fileName, ins); 
      }
  }
}

【原因】:
我代码中有两处用到了InputStream,实际上第一次读取InputStream对象后,第二次再读取时可能已经到Stream的结尾了(EOFException)或者Stream已经close掉了,所以第二次再使用InputStream,我们就没有办法得到内容了。
【解决办法】:
InputStream对象本身不能复制,因为它没有实现Cloneable接口。因此我们可以此时,可以先把InputStream转化成ByteArrayOutputStream(将其缓存在到ByteArrayOutputStream中),后面要使用InputStream对象时,再从ByteArrayOutputStream转化回来就好了。
修改后的代码

FileIoUtil.java
class FileIoUtil{
//获取文件里的内容
  public List<String> getFileIo(InputStream ins){
       Workbook workbook = null;
       List<String> parm = new ArrayList<String>();
	   try {
		  workbook = WorkbookFactory.create(ins);	
		  Sheet dataSheet = workbook.getSheetAt(0);
		  int rownum = dataSheet.getLastRowNum();
		  for (int i = 1; i <= rownum; i++) {
           ....
        //一系列操作,将excel文件里的内容加入到list——parm中
         }
      }catch(Exception e){

      ...
      }finally {
			if (ins != null) {
				try {
					ins.close();
				} catch (IOException e) {
				}
            }
   }

//上传文件到服务器
 public void uploadFileToService(String path,String fileName,InputStream ins){

  //...

 }


Test1.java
class Test1{
   /**
   *fileList就是前台传过来的文件信息
   */
   public void uploadFile(List<?> fileList){
   //获取文件流
	FileUpload fileUpload = (FileUpload)fileList.get(0);
	InputStream ins = FileUtil.getInputStreamFromFileUpload(fileUpload);
    /** * 将InputStream中的字节保存到ByteArrayOutputStream中 */
    ByteArrayOutputStream baos = FileIoUtil.inputStreamCacher(ins);
    InputStream checkIns = new  ByteArrayInputStream(baos.toByteArray());  
    List<String> param = FileIoUtil.getFileIo(checkIns);
     //判断是否有重复的内容
    param = findRepeatedRecord(param);
     //证明有
     if(param.size()>0){
        reutrn;
     }else{
     //没有重复,则需要上传
        String  path  = "E:\\test\\";
        FileUpload fileUpload = (FileUpload)fileList.get(0);
        String  fileName = fileUpload.getFileName();
        InputStream attachIns = new ByteArrayInputStream(baos.toByteArray());  
        //上传文件
			FileIoUtil.uploadFileToService(path, fileName, attachIns ); 
      }
  }
}
相关标签: 经验之谈