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

请参考这个答案here，因为它描述了如何将 ByteArrayOutputStream 与 HSSFWorkbook 一起使用。

希望对您有所帮助。