easyexcel调用公共导出方法导出数据

easyexcel备忘

@Slf4j
public class ConditionDownloadUtil {

	//扫描在xboot 包下所有IService 接口的子类, 每次启动服务后, 重新扫描
	public final static Class[] classesExtendsIService = ClassUtil.scanPackageBySuper("cn.exrick.xboot", IService.class).toArray(new Class[1]);

	public static <T, E> void nonMultiQueryDownload(
		Class<T> clazz,
		QueryWrapper<T> queryWrapper,
		String excelName,
		Class<E> excelClazz,
		HttpServletResponse response) {

		if (ObjectUtils.isNotEmpty(clazz)) {

			//1. 根据入参中的class 来获取对应的IService Bean
			IService<T> targetIService = getTargetIService(clazz);

			/*
			 * 2. 由于查询/排序字段在每个模块中不同, 所以需要让调用方提供, 不再统一提供
			 * 调用IService 的list() 方法获取查询符合条件的数据
			 */
			List<T> list = targetIService.list(queryWrapper);

			if (list == null || list.size() == 0) {

				throw ExceptionUtil.wrapRuntime("数据为空, 无法下载excel!");

			}

			//3. 将实体类List 转换为Excel 类List
			List<E> excelList = list.stream().map(x -> BeanUtil.toBean(x, excelClazz, CopyOptions.create().setIgnoreError(true))).collect(Collectors.toList());

			//4. 调用EasyExcel 通用方法输出Excel
			afterMultiQueryDownload(excelName, excelList, excelClazz, new CustomMergeStrategy(excelClazz, excelList.size()), response);

		} else {

			throw ExceptionUtil.wrapRuntime("实体类不能为空!");

		}

	}

	public static <E> void afterMultiQueryDownload(
		String excelName,
		List<E> excelList,
		Class<E> excelClass,
		WriteHandler writeHandler,
		HttpServletResponse response) {

		//1. 判断需要输入的Excel List 是否为空, 如果不为空, 则输入, 否则, 抛出异常
		if (CollectionUtil.isNotEmpty(excelList)) {

			response.setContentType("application/vnd.ms-excel");
			response.setCharacterEncoding("utf-8");
			response.setHeader("Content-disposition", "attachment;filename=" + URLUtil.encode(excelName, StandardCharsets.UTF_8)  + ".xlsx");

			//2. 输入Excel
			try {

				ExcelWriterSheetBuilder excelWriterSheetBuilder = EasyExcel.write(response.getOutputStream())
						.excelType(ExcelTypeEnum.XLSX)
						.head(excelClass)
						.sheet();

				if (ObjectUtil.isNotEmpty(writeHandler)) {

					excelWriterSheetBuilder.registerWriteHandler(writeHandler);

				}

				//3. 清除Convertor 中的线程变量
				DictConverter.removeThreadLocal();

				excelWriterSheetBuilder.doWrite(excelList);

			} catch (Exception e) {

				StringWriter sw = new StringWriter();
				PrintWriter pw = new PrintWriter(sw);
				e.printStackTrace(pw);

				log.error(sw.toString());

				if (e.getCause() instanceof ExcelDataConvertException) {

					ExcelDataConvertException excelDataConvertException = (ExcelDataConvertException) e.getCause();
					String cellMsg = "";
					CellData cellData = excelDataConvertException.getCellData();
					//这里有一个celldatatype的枚举值,用来判断CellData的数据类型
					CellDataTypeEnum type = cellData.getType();
					if (type.equals(CellDataTypeEnum.NUMBER)) {
						cellMsg = cellData.getNumberValue().toString();
					} else if (type.equals(CellDataTypeEnum.STRING)) {
						cellMsg = cellData.getStringValue();
					} else if (type.equals(CellDataTypeEnum.BOOLEAN)) {
						cellMsg = cellData.getBooleanValue().toString();
					}

					String errorMsg = String.format("excel表格:第%s行,第%s列,数据值为:%s,该数据值不符合要求,请检验后重新导入!请检查其他的记录是否有同类型的错误!", excelDataConvertException.getRowIndex() + 1, excelDataConvertException.getColumnIndex(), cellMsg);
					log.error(errorMsg);

				}

			}

		} else {

			throw ExceptionUtil.wrapRuntime("数据为空, 无法下载excel!");

		}

	}

	public <T, E> void dynamicHeadDownload(
		List<List<String>> headList,
		String excelName,
		List data,
		HttpServletResponse response) {

		//1. 判断Head List 是否为空, 如果不为空, 则输入, 否则, 抛出异常
		if (ObjectUtil.isNotEmpty(headList)) {

			response.setContentType("application/vnd.ms-excel");
			response.setCharacterEncoding("utf-8");
			response.setHeader("Content-disposition", "attachment;filename=" + excelName + ".xlsx");

			//2. 输入Excel
			try {

				EasyExcel.write(response.getOutputStream())
					.head(headList)
					.sheet()
					.doWrite(data);

			} catch (Exception e) {

				StringWriter sw = new StringWriter();
				PrintWriter pw = new PrintWriter(sw);
				e.printStackTrace(pw);

				log.error(sw.toString());

				if (e.getCause() instanceof ExcelDataConvertException) {

					ExcelDataConvertException excelDataConvertException = (ExcelDataConvertException) e.getCause();
					String cellMsg = "";
					CellData cellData = excelDataConvertException.getCellData();
					//这里有一个celldatatype的枚举值,用来判断CellData的数据类型
					CellDataTypeEnum type = cellData.getType();
					if (type.equals(CellDataTypeEnum.NUMBER)) {
						cellMsg = cellData.getNumberValue().toString();
					} else if (type.equals(CellDataTypeEnum.STRING)) {
						cellMsg = cellData.getStringValue();
					} else if (type.equals(CellDataTypeEnum.BOOLEAN)) {
						cellMsg = cellData.getBooleanValue().toString();
					}

					String errorMsg = String.format("excel表格:第%s行,第%s列,数据值为:%s,该数据值不符合要求,请检验后重新导入!请检查其他的记录是否有同类型的错误!", excelDataConvertException.getRowIndex() + 1, excelDataConvertException.getColumnIndex(), cellMsg);
					log.error(errorMsg);

				}

			}

		} else {

			throw new RuntimeException("数据为空, 无法下载excel!");

		}

	}

	/**
	 * 添加表头
	 * @param head
	 * @param headers
	 */
	public static void addHead(String head, List<List<String>> headers) {

		ArrayList<String> list = new ArrayList<>(1);

		list.add(head);

		headers.add(list);

	}

	/*
	@SneakyThrows
	public void tablesDownload(ReportDownloadModel reportDownloadModel, HttpServletResponse response) {

		String excelName = URLEncoder.encode(reportDownloadModel.getExcelName(), "utf-8");
		List tableList = reportDownloadModel.getTables();

		response.setContentType("application/vnd.ms-excel");
		response.setCharacterEncoding("utf-8");
		response.setHeader("Content-disposition", "attachment;filename=" + excelName + ".xlsx");

		ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()).build();
		WriteSheet writeSheet = EasyExcel.writerSheet()
			.registerWriteHandler(new CustomCellWriteHandler())
			.registerWriteHandler(new CustomCellWriteHeightConfig())
			.needHead(Boolean.TRUE).build();

		//1. 定义标题二维数组
		ArrayList> titleHead = new ArrayList<>();

		//2. 定义最终输出的excel 二维数组
		ArrayList> finalData = new ArrayList<>();

		//3. 定义空行
		ArrayList