Java使用输出流OutputStream导出Excel遇到的问题及解决方法

这半年一直在参与一个新系统的软件开发，再此期间遇到了一个小小的问题，就是使用原生POI导出Excel时，会生成非Excel格式的文件，而且文件名称也不是设置好的，而是导出的方法名，如下图；

不过这种文件选择打开方式Excel或者WPS都可以，但是对客户就不是很友好了，所以必须要查找一下问题所在。
在经过与同事的正常导出代码进行对比后，发现如下问题：

1. 在文件导出时，生成的Workbook数据要存入到字节流或字符流当中，并将流写入到response响应中；
2. 需要将文件名称格式化，并设置响应头；
   现在将失败和成功代码分享如下：

错误

 // 将工作簿写入输出流中
OutputStream outputStream = response.getOutputStream();
workbook.write(outputStream);
// 设置响应头，指定文件名和文件类型
response.setContentType("application/vnd.ms-excel"); // 设置为 Excel 文件类型
response.setHeader("Content-Disposition", "attachment; filename=\"" + fileName + ".xls\""); // 设置文件名为 fileName + ".xls"
// 关闭输出流和工作簿对象
outputStream.close();
workbook.close();

正确

// 将工作簿写入输出流中
ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
workbook.write(byteArrayOutputStream);
byte[] bytes = byteArrayOutputStream.toByteArray();
// 设置响应头，指定文件名和文件类型
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
response.setHeader("Content-Disposition", "attachment; filename="+fileName+"-"+ DateUtils.getDate("yyyyMMddHHmmss")+".xls");
response.setContentLength(bytes.length);
// 关闭输出流和工作簿对象
ServletOutputStream outputStream = response.getOutputStream();
outputStream.write(bytes);
outputStream.flush();
outputStream.close();

根据代码对比可以看到，错版没有把workbook写入到文件流当中，只是将流写到了响应里边，就会导致当数据量超过分页时的数据量则导出的文件不是“xls”或者“xlsx”格式，而当数据量小于等于分页的数据量则可以导出“xls”或者“xlsx”文件格式。

这个问题确实疏忽了文件流的问题，一个是因为测试时数据量比较少，所以用错误写法导出时也是正常显示“xls”格式；另一个是因为只看到了响应中的getOutputStream()方法，忽略了正常应该使用的字节流或字符流。

欢迎大家加入CSDN开发云