apache POI-获取生成的Excel文件的大小

Question

我正在使用Apache POI在Spring MVC应用程序中生成Excel文件。这是我的春季动作：

 @RequestMapping(value = "/excel", method = RequestMethod.POST)
public void companyExcelExport(@RequestParam String filter, @RequestParam String colNames, HttpServletResponse response) throws IOException{
    XSSFWorkbook workbook = new XSSFWorkbook();
    //code for generate excel file
    //....

    response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
    response.setHeader("Content-Disposition", "attachment; filename=test.xlsx");
    workbook.write(response.getOutputStream());

    response.setHeader("Content-Length", "" + /* How can i access workbook size here*/);
}

我用过，XSSFWorkbook因为我需要生成Excel 2007格式。但是我的问题是XSSFWorkbook没有getBytes或getSize方法。如何计算生成的xlsx文件的大小？

编辑：我ByteArrayOutputStream喜欢这里：

ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
workbook.write(response.getOutputStream()); 
workbook.write(byteArrayOutputStream);
response.setHeader("Content-Length", "" + byteArrayOutputStream.size());

Answer 1

正如@JB Nizet 所说：在写入响应之前设置标头。

所以你应该做的是：

ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
workbook.write(byteArrayOutputStream);
response.setHeader("Content-Length", "" + byteArrayOutputStream.size());
workbook.write(response.getOutputStream()); 

请参阅此处的答案，因为它描述了如何将 ByteArrayOutputStream 与 HSSFWorkbook 一起使用。

希望有帮助。