2015-01-24 2 views
4

Я использую Apache POI для генерации файла excel в моем приложении mvc spring. вот моя весна действие:apache POI - получить размер сгенерированного файла 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?

EDIT: Я использовал ByteArrayOutputStream как здесь:

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

Напишите книгу в ByteArrayOutputStream, получите размер от этого, а затем напишите байты в реальном потоке? – Gagravarr

+0

Я пробовал ваше решение, но все-таки браузер не показывает размер файла. – hamed

+0

, пожалуйста, обновите вопрос, как вы использовали ByteArrayOutputStream –

ответ

3

Как сказал @JB Nizet: установить заголовок перед записью ответа.

так, что вы должны сделать, это следующее:

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

и, пожалуйста, обратитесь к этому ответу here, как он описывает, как использовать ByteArrayOutputStream с HSSFWorkbook.

Надеюсь, что это поможет.

+0

Хорошо, это сработало. Благодарю. Но есть ли все-таки сделать это без ByteArrayOutputStream и с XSSFWorkbook? – hamed