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

Question

我正在使用 Apache POI用于在我的 spring mvc 应用程序中生成 excel 文件。这是我的 Spring 行动:

 @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 所说:在编写响应之前设置 header 。

所以你应该做的是:

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

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

希望对您有所帮助。