Java利用Reflect实现封装Excel导出工具类_Java教程_编程语言

Java利用Reflect实现封装Excel导出工具类

2023-02-09 学习力0

核心提示：目录定义ExcelExportEnum定义导出方法校验入参利用java反射获取Service中的所有method提取method中的入参对象，获取其Class传入参数，调用导出方法注意事项最近遇到一个需求，需要对页面的列表数据做导出操作,考虑了很多实现方案之后，最终选择了hutool的Exce

最近遇到一个需求，需要对页面的列表数据做导出操作,考虑了很多实现方案之后，最终选择了hutool的ExcelWriter + Spring自带的ReflectionUtils工具类来实现这个功能，采用这种方式的主要原因是对现有代码改动较少，并且可以无缝切入系统，实现各个模块的导出操作，设计思路如下：

Java利用Reflect实现封装Excel导出工具类

如上图所示，首先我们定义一个接口，此接口完成Excel导出功能，在ExcelExportService 中实现功能。

话不多说，上代码：

定义ExcelExportEnum

Java利用Reflect实现封装Excel导出工具类

在导出Excel之前，我们需要先定义ExcelExportEnum，此枚举存储了我们后端需要导出的模块和模板名称

@Getter
@RequiredArgsConstructor
public enum ExcelExportEnum {

/**
 * 模块
 */
MODULE_A("moduleAService", "模块A");

private final String key;
private final String name;

}

定义导出方法

校验入参

首先，我们需要对入参进行校验，查看传入的 MODULE_NAME 是否和后端Enum的一致，并且我们需要对参数中的自定义方法名进行提取，我们通过ExcelExportEnum来获取导出的module的Service名称

    if (!map.containsKey(MODULE_NAME)) {
        throw new BaseException("缺少参数：moduleName");
    }
    String moduleName = map.get(MODULE_NAME).toString();
    if (!EnumUtil.contains(ExcelExportEnum.class, moduleName)) {
        throw new BaseException("模块查找失败");
    }
    String functionName = DEFAULT_FUNCTION_NAME;
    //如果传了自定义方法名
    if (map.containsKey(CUSTOM_FUNCTION_NAME)) {
        functionName = map.get(CUSTOM_FUNCTION_NAME).toString();
    }
    String serviceName = ExcelExportEnum.valueOf(moduleName).getKey();
    String tableName = ExcelExportEnum.valueOf(moduleName).getName();

利用java反射获取Service中的所有method

Map<String, Method> methodMap = Arrays.stream(SpringContextHolder.getBean(serviceName).getClass().getMethods()).collect(Collectors.toMap(Method::getName, Function.identity(), (key1, key2) -> key2));

在上述代码中SpringContextHolder.getBean(serviceName).getClass().getMethods()方法，首先通过getBean来获取第一步中得到的ServiceName的Java bean。然后通过Class.getMethods()方法获取此service中的所有方法

最后，使用Arrays.stream方法，将module对应的Service中的所有method获取到，并放入methodMap中，供后续代码使用

提取method中的入参对象，获取其Class

    Object o;
    try {
        o = methodMap.get(functionName).getParameterTypes()[0].getDeclaredConstructor().newInstance();
    } catch (Exception e) {
        throw new BaseException(e);
    }

通过Method.getParameterTypes()方法，将第二步中获取的method的入参的Class取到，并赋值给对象Object o，供后续调用分页列表接口传递入参匹配Class使用

传入参数，调用导出方法

    Map<String, Object> param = (Map<String, Object>) map.computeIfAbsent(PARAM, k -> new HashMap<String, Object>(4));
    param.put("limit", -1);
    CopyOptions copyOptions = new CopyOptions();
    copyOptions.setIgnoreError(true);
    PageUtils page = (PageUtils) methodMap.get(functionName).invoke(SpringContextHolder.getBean(serviceName), BeanUtil.mapToBean(param, o.getClass(), true, copyOptions));
    if (CollUtil.isEmpty(page.getList())) {
        throw new BaseException("数据获取失败");
    }
    Object resultObject = page.getList().get(0);
    ExcelUtils utils = new ExcelUtils(resultObject.getClass());
    utils.exportExcel(page.getList(), response, tableName, moduleName);

可以看到，在上述导出方法中，我们使用了Method.invoke方法，并且设置转义忽略

注意事项

使用hutool工具包输出到流

// 通过工具类创建writer，默认创建xls格式 
ExcelWriter writer = ExcelUtil.getWriter(); 
//创建xlsx格式的 
//ExcelWriter writer = ExcelUtil.getWriter(true); 
// 一次性写出内容，使用默认样式，强制输出标题 
writer.write(rows, true); 
//out为OutputStream，需要写出到的目标流 
writer.flush(out); 
// 关闭writer，释放内存 
writer.close();

注意 ExcelUtil.getWriter()默认创建xls格式的Excel，因此写出到客户端也需要自定义文件名为XXX.xls，否则会出现文件损坏的提示。若想生成xlsx格式，请使用ExcelUtil.getWriter(true)创建。

原文地址：https://juejin.cn/post/7163136768340918285

免责声明：本文仅代表作者个人观点，与乐学笔记（本网）无关。其原创性以及文中陈述文字和内容未经本站证实，对本文以及其中全部或者部分内容、文字的真实性、完整性、及时性本站不作任何保证或承诺，请读者仅作参考，并请自行核实相关内容。
本网站有部分内容均转载自其它媒体，转载目的在于传递更多信息，并不代表本网赞同其观点和对其真实性负责，若因作品内容、知识产权、版权和其他问题，请及时提供相关证明等材料并与我们留言联系，本网站将在规定时间内给予删除等相关处理.