常用开发功能——批量文件导出

        某天工作接到一个需求：批量导出场站的订单信息，一个场站一个Excel文档；

        与单个文件导出的区别在于，单个导出一次性只导出一个文件，在输出流中写入这一个文件即可，那么如何一次性导出多个文件？难道一次接口返回能被循环读取重复打开保存对话框？

        不是，其实是使用压缩文件，在内存中将生成的EXCEL文件依次加入到压缩文件中，最后形成一个.zip格式的文件，写入输出流作为接口的返回；

理解简单，话不多说，代码如下：

	public void exportLogBatch(ChargingParkCouponLogVO chargingParkCouponLog, HttpServletResponse response) throws IOException {
		// 设置响应头，指定下载的文件类型为压缩文件
		response.setContentType("application/zip");
		String start = DateUtil.format(chargingParkCouponLog.getStartDate(), "yyyy-MM-dd");
		String end = DateUtil.format(chargingParkCouponLog.getEndDate(), "yyyy-MM-dd");
		response.setHeader("Content-Disposition", "attachment; filename=\"订单停车券按场站批量导出(" + start + "~" + end + ").zip\"");
		String[] parkIdArr = chargingParkCouponLog.getParkIds().split(",");
		try (ZipOutputStream zipOut = new ZipOutputStream(response.getOutputStream())) {
			for (String s : parkIdArr) {
				try {
					ChargingParkConfig chargingParkConfig = chargingParkConfigService.getById(Long.valueOf(s));
					if (ObjectUtils.isEmpty(chargingParkConfig)) {
						continue;
					}
					List<ChargingParkCouponLogExport> exportList = generateExportFiles(chargingParkCouponLog);
					byte[] bytes = generateFileBytes(exportList);
					ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(bytes);
					// 创建压缩文件的条目，并将文件内容写入压缩文件
					ZipEntry zipEntry = new ZipEntry("道闸发券识别异常清单—" + chargingParkConfig.getParkName() + ".xlsx");
					zipOut.putNextEntry(zipEntry);

					byte[] buffer = new byte[1024];
					int length;
					while ((length = byteArrayInputStream.read(buffer)) >= 0) {
						zipOut.write(buffer, 0, length);
					}
					zipOut.closeEntry();
				} catch (Exception e) {
					throw new ServiceException("停车券发放记录列表导出异常，编号：" + s);
				}
			}
			zipOut.finish();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	private List<ChargingParkCouponLogExport> generateExportFiles(ChargingParkCouponLogVO chargingParkCouponLog) {
		Page page = new Page<>();
		page.setSize(10000L);
		List list = chargingParkCouponLogService.selectChargingParkCouponLogPage(page, chargingParkCouponLog).getRecords();
		List<ChargingParkCouponLogExport> exportList = new LinkedList<>();
		list.forEach(item -> {
			ChargingParkCouponLogExport logExport = new ChargingParkCouponLogExport();
			BeanUtils.copyProperties(item, logExport);
			exportList.add(logExport);
		});
		return exportList;
	}

	private byte[] generateFileBytes(List<ChargingParkCouponLogExport> exportList) {
		// 创建一个ByteArrayOutputStream对象，用于将Excel数据写入内存中的字节数组
		ByteArrayOutputStream outputStream = new ByteArrayOutputStream();

		// 使用ExcelWriter创建一个写入器，并将其与ByteArrayOutputStream关联
		ExcelWriter excelWriter = EasyExcel.write(outputStream, ChargingParkCouponLogExport.class).build();

		// 创建一个WriteSheet对象，设置Sheet名称和表头类
		WriteSheet writeSheet = EasyExcel.writerSheet().build();

		// 将数据列表写入Excel文件
		excelWriter.write(exportList, writeSheet);

		// 关闭ExcelWriter
		excelWriter.finish();

		// 获取ByteArrayOutputStream的字节数组
		byte[] bytes = outputStream.toByteArray();

		return bytes;
	}

调用接口：

 

返回结果：

总结下导出逻辑的变更：

         之前：准备要导出的场站ID，获取该场站要导出的数据，调用导出工具类写输出流；

         之后：准备要导出的场站的批量的ID列表，遍历ID列表循环获取场站各自要导出的数据，打开请求的总输出流作为ZipOutStream的输入流，对每个要导出的场站列表，创建单独的二进制输出流，将各自要导出的数据，写入输出流中，写入完后将输出流返回为二进制数组，作为输入流，写入创建的压缩文件中，将所有的导出文件二进制数据都写入各自的ZipEntry，最后写入总的ZipOutStream，循环执行，即可导出所有的文件作为压缩包；