一.业务背景

工作中接到的一个业务需求，需要将某部分数据按照某个条件拆分成多个excel然后打包成压缩包，导出压缩包。

二.解决方式探索

作为一个码农，当不知道该如何解决问题的时候，第一步自然是先面向百度(谷歌)编程。于是开始了漫长的搜索。大多数的解决方式都是先在本地建一个临时的文件夹，将每个excel放到这个文件夹中，全部添加完成后打包这个文件夹，再删除这个文件夹。

看到这种解决方式，我第一反应就是恶心！什么？这样搞也太蠢了把。于是我又继续搜啊搜，终于让我找到了一个非常非常舒服的解决方式，非常感谢这位博主。

三.解决问题

@ResponseBody

@RequestMapping(value = "/xxx")

public BaseResp unDirectExport(Date date, HttpServletResponse response) throws Exception {

String fileName = xxx;

ResponseUtil.setMultipartHeader(response, fileName);

List list = xxxService.query(xx);

//按某个条件分组

Map> map = list.stream().collect(Collectors.groupingBy(T::getXXX));

exportExcel(response, map);

return BaseResp.succResp();

}

private void exportExcel(HttpServletResponse response, Map> map) throws IOException {

//Excel sheet 标题

String[] headers = {""};

ServletOutputStream outputStream = response.getOutputStream();

ZipOutputStream zipOutputStream = new ZipOutputStream(outputStream);

try {

map.forEach((k, v) -> {

//新建一个Excel 并设置下sheet头

HSSFWorkbook workbook = createExcelAndSetHeaders(headers, k);

//向sheet中 继续填充对象的数据

setSheetCellValue(workbook.getSheet(k), v);

try {

//重点开始，创建压缩文件

ZipEntry zipEntry = new ZipEntry(k + ".xls");

zipOutputStream.putNextEntry(zipEntry);

} catch (IOException e) {

logger.error("向XXX压缩包中添加Excel失败");

throw new Exception("向XXX压缩包中添加Excel失败");

}

try {

//写入一个压缩文件

workbook.write(zipOutputStream);

} catch (IOException e) {

logger.error("向zipOutputStream中写入流数据失败");

throw new Exception("向zipOutputStream中写入流数据失败");

}

});

zipOutputStream.flush();

} catch (Exception e) {

//重复丢出异常，有点多余，但是为了套上try catch finally 关闭数据流

logger.error("导XXX失败,原因" + e.getErrorCode());

throw new Exception("导出XXX结算数据失败，原因：" + e.getErrorCode());

} finally {

//关闭数据流，注意关闭的顺序

zipOutputStream.close();

outputStream.close();

}

}

private HSSFWorkbook createExcelAndSetHeaders(String[] headers, String sheetName) {

HSSFWorkbook hssfWorkbook = new HSSFWorkbook();

HSSFSheet hssfSheet = hssfWorkbook.createSheet(sheetName);

HSSFRow row0 = hssfSheet.createRow(0);

for (int i = 0; i < headers.length; i++) {

HSSFCell cellHeader = row0.createCell(i);

cellHeader.setCellValue(headers[i]);

}

return hssfWorkbook;

}

private void setSheetCellValue(HSSFSheet hssfSheet, List dtos) {

for (T dto : dtos) {

//从当前sheet页的最后一行后新增一行，开始填充数据

HSSFRow row = hssfSheet.createRow(hssfSheet.getLastRowNum() + 1);

int count = -1;

row.createCell(++count).setCellValue(dto.getXXX);

row.createCell(++count).setCellValue(dto.getXXX);

row.createCell(++count).setCellValue(dto.getXXX);

row.createCell(++count).setCellValue(dto.getXXX);

row.createCell(++count).setCellValuedto.getXXX);

}

}

import javax.servlet.http.HttpServletResponse;

import java.io.UnsupportedEncodingException;

import java.net.URLEncoder;

/**

* @author zhaojq

* @email zhaojq@tsintergy.com

* @date 2019 2019/10/14 17:11

*/

public class ResponseUtil {

private ResponseUtil() {

}

public static void setMultipartHeader(HttpServletResponse response, String fileName) {

try {

response.setCharacterEncoding("utf-8");

response.setContentType("multipart/form-data");

response.setHeader("Content-Disposition", "attachment;filename*=UTF-8''" + URLEncoder.encode(fileName, "UTF-8"));

} catch (UnsupportedEncodingException e) {

throw new BusinessException("文件名编码异常");

}

}

}