InputStream复用问题

青旅半醒 2022-11-27 08:51 130阅读 0赞

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

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 ); 
          }
      }
    }