Struts2.1.8+Spring3.0+Ibatis2.3.4集成（六）

一、文件下载升级为stream方式。

 

原来的代码为：

 

public class HRGetZhSExcelParamAction extends HrpaAction implements SessionAware,ServletContextAware,ServletResponseAware { private ISalaryManager<Salary, Integer> salaryManager; public void setSalaryManager(ISalaryManager<Salary, Integer> salaryManager) { this.salaryManager = salaryManager; } private IExcelManager<Salary> excelManager; public void setExcelManager(IExcelManager<Salary> excelManager) { this.excelManager = excelManager; } private Map<String, Object> session; public void setSession(Map<String, Object> map){ this.session=map; } private ServletContext servletContext; @Override public void setServletContext(ServletContext servletContext) { this.servletContext=servletContext; } private HttpServletResponse servletResponse; public void setServletResponse(HttpServletResponse servletResponse) { this.servletResponse = servletResponse; } private String fileName; private Date dateSalary; public Date getDateSalary() { return dateSalary; } public void setDateSalary(Date dateSalary) { this.dateSalary = dateSalary; } public String getFileName() { return fileName; } public void setFileName(String fileName) { this.fileName = fileName; } public String execute() throws Exception { Account acc=(Account)session.get("HLAccInfo"); List<Salary> lstS; if(dateSalary==null){ lstS=salaryManager.getCurrentMonthZhS(acc.getAccId()); }else{ ParamContext param=new ParamContext(); param.setAccId(acc.getAccId()); param.setDateSalary(dateSalary); lstS=salaryManager.getCurrentMonthZhS(param); } String tempDir=servletContext.getRealPath("excelTemplete"); File fileTemp=new File(tempDir,"zhSTemplete.xls"); String resultDir=servletContext.getRealPath("excelResult"); String fileName=HrpaStringUtil.getZhSalaryExcelFileName(acc.getAccName()); File fileResult=new File(resultDir,fileName); if(fileTemp.exists()){ excelManager.addSalaryToSheet(fileTemp, fileResult, "明细", lstS); } if(fileResult.exists()){ downLoadData(servletResponse,fileResult,fileName); } return null; } public static void downLoadData(HttpServletResponse response,File returnFile,String downFileName){ response.setHeader("Content-disposition","attachment; filename="+downFileName); response.setHeader("Content-Type", "application/octet-stream"); BufferedInputStream bis = null; BufferedOutputStream bos = null; try{ bis = new BufferedInputStream(new FileInputStream(returnFile)); bos=new java.io.BufferedOutputStream(response.getOutputStream()); byte[] buff = new byte[2048]; int bytesRead; while(-1!= (bytesRead = bis.read(buff, 0, buff.length))){ bos.write(buff, 0, bytesRead); } }catch(Exception e){ e.printStackTrace(); }finally{ if (bis != null) try { bis.close(); } catch (IOException e) { e.printStackTrace(); } if (bos != null) try { bos.close(); } catch (IOException e) { e.printStackTrace(); } } } }

 

升级为stream的方式：

 

第一：一定要实现ServletResponseAware，不然会有很多想象不到的错误，但是如果是已经存在的固有文件是可以不实现它。建议实现它。

 

第二：写一个返回输入流的方法，

 

public InputStream getInputStream() throws Exception { Account acc=(Account)session.get("HLAccInfo"); List<Salary> lstS; if(dateSalary==null){ lstS=salaryManager.getCurrentMonthZhS(acc.getAccId()); }else{ ParamContext param=new ParamContext(); param.setAccId(acc.getAccId()); param.setDateSalary(dateSalary); lstS=salaryManager.getCurrentMonthZhS(param); } String tempDir=servletContext.getRealPath("excelTemplete"); File fileTemp=new File(tempDir,"zhSTemplete.xls"); String resultDir=servletContext.getRealPath("excelResult"); fileName=HrpaStringUtil.getZhSalaryExcelFileName(acc.getAccName()); File fileResult=new File(resultDir,fileName); if(fileTemp.exists()){ excelManager.addSalaryToSheet(fileTemp, fileResult, "明细", lstS); } if(fileResult.exists()){ try { return new FileInputStream(fileResult); } catch (IOException ex) { } } return null; }

 

第三：更改配置文件：

 

   <action name="HRGetZhSExcelFile" class="HRGetZhSExcelFile">
    <result name="success" type="stream">
     <param name="inputName">inputStream</param> ---名字和getXXX的一致
     <param name="contentType">application/octet-stream</param>
     <param name="contentDisposition">attachment;filename="${fileName}"</param>
     <param name="bufferSize">2048</param>
    </result>                         
  </action>

 

第四：页面代码

 

        <s:url id="url" action="HRGetZhSExcelFile">
        </s:url>
        <s:a href="%{url}">下载 </s:a>