| | |
|---|
| | | package cn.iocoder.yudao.framework.excel.core.util; |
|---|
| | | |
|---|
| | | import cn.idev.excel.ExcelWriter; |
|---|
| | | import cn.idev.excel.FastExcelFactory; |
|---|
| | | import cn.idev.excel.converters.longconverter.LongStringConverter; |
|---|
| | | import cn.iocoder.yudao.framework.common.util.http.HttpUtils; |
|---|
| | |
|---|
| | | public static <T> void write(HttpServletResponse response, String filename, String sheetName, |
|---|
| | | Class<T> head, List<T> data) throws IOException { |
|---|
| | | // 输出 Excel |
|---|
| | | FastExcelFactory.write(response.getOutputStream(), head) |
|---|
| | | ExcelWriter excelWriter = FastExcelFactory.write(response.getOutputStream(), head) |
|---|
| | | .autoCloseStream(false) // 不要自动关闭，交给 Servlet 自己处理 |
|---|
| | | .registerWriteHandler(new ColumnWidthMatchStyleStrategy()) // 基于 column 长度，自动适配。最大 255 宽度 |
|---|
| | | .registerWriteHandler(new SelectSheetWriteHandler(head)) // 基于固定 sheet 实现下拉框 |
|---|
| | | .registerConverter(new LongStringConverter()) // 避免 Long 类型丢失精度 |
|---|
| | | .sheet(sheetName).doWrite(data); |
|---|
| | | .build(); |
|---|
| | | try { |
|---|
| | | excelWriter.write(data, FastExcelFactory.writerSheet(sheetName).build()); |
|---|
| | | } finally { |
|---|
| | | // finish() 将 workbook 写入到 response 输出流 |
|---|
| | | excelWriter.finish(); |
|---|
| | | // close() 触发 dispose() 清理临时文件； |
|---|
| | | // POI SXSSFWorkbook.dispose() 可能输出 WARN "Failed to dispose sheet"， |
|---|
| | | // 这是 POI 已知问题（临时文件清理时 BufferedWriter 已关闭），不影响导出结果 |
|---|
| | | excelWriter.close(); |
|---|
| | | } |
|---|
| | | // 设置 header 和 contentType。写在最后的原因是，避免报错时，响应 contentType 已经被修改了 |
|---|
| | | response.addHeader("Content-Disposition", "attachment;filename=" + HttpUtils.encodeUtf8(filename)); |
|---|
| | | response.setContentType("application/vnd.ms-excel;charset=UTF-8"); |
|---|
