一句话总结
Excel 导出、导入时,根据注解自动添加单元格验证规则,避免用户因填写错误的枚举字段而反复修改 Excel
需求背景
对于 Java Web 项目,总是不可避免的出现 Excel 导入、导出的需求,而 Excel 导入导出时,枚举字段和枚举值的映射是非常常见的一种情况
例如:下面这张示例中的性别列
数据库表结构:
Excel 中用户需要输入:男,女,未知
常见的 Excel 框架都已经覆盖了枚举映射的功能,例如:EasyPOI
但是这种操作方式对于用户来说,并不是很方便,试想一下:假如用户在性别列输入了:男性,最终的结果一般就是程序抛出异常,用户得到提示:性别输入有误,贴心的开发者可能会加上:请输入 男 女 未知,做的更好一些的 可能在列头添加标签提示:该列仅能输入 男 女 未知,但是这种弱限制也无法从根本上解决问题
更好一点的解决方案是:利用 Excel 的数据验证功能,把单元格加上规则校验,让用户只能输入正确的枚举值,避免因一次输入错误而反复返工,浪费用户的时间和好心情
当用户输入了非枚举值之后,Excel 会提示用户输入不合规,禁止用户保存
这样的交互就能从源头保证用户输入正确的值
那这么友好的设计,在 Java 中如何能方便且可扩展性更强的实现呢?
需求实现
我这边的实现是基于 EasyPOI + 注解(EasyPOI 转换映射关系注解复用) + 反射 实现的,解决了以上需求痛点的同时,可以满足代码一处修改,多个功能都生效的目的
代码仓库
https://github.com/keatsCoder/excel
实体类
@Excel 注解中的 replace 属性,该属性是 EasyPOI 用来做字段映射的,我这里复用他做 Excel 验证的可选项,另外一个就是 orderNum 属性,用该值来自动获取某个字段在 Excel 中的列的位置
@Data
public class Human extends BaseEntity {
private Long id;
@Excel(name = “姓名”, orderNum = “1”, width = 15)
private String name;
@Excel(name = “年龄”, orderNum = “2”, width = 15)
private Integer age;
@Excel(name = “性别”, replace = {“男_1”, “女_2”, “未知_3”}, orderNum = “3”, width = 15)
private Integer gender;
}
获取列名和列位置的映射
该类在初始化时,需要指定当前导出 Excel 对应的实体类的类类型,然后通过遍历类中字段的注解,生成字段和列排序(位置)的映射关系
public class FieldOrderMappingHelper {
/**
* 支持的最大字段数
*/
private final static int MAX_LIST_SIZE = 26;
public FieldOrderMappingHelper(Class pojo) {
this.pojo = pojo;
initMap();
}
/**
* 解析注释的 pojo 对象
*/
private Class pojo;
/**
* 字段和序号的映射关系
*/
private HashMap<String, Integer> fieldAndOrderMap;
/**
* 功能:初始化类的字段内容,建立字段和序号以及字段和 excel 列名的映射关系
* @author kangshuai@gridsum.com
* @date 2021/4/9 12:06
*/
private void initMap() {
HashMap<String, Integer> fieldAndOrderMap = new HashMap<>(16);
HashSet existOrderNumSet = new HashSet<>(16);
List list = new ArrayList<>();
list = initList(list, pojo);
if (list.size() > MAX_LIST_SIZE) {
throw new RuntimeException(pojo.getName() + “目前最大支持 26 个字段,26+ 需要改代码”);
}
// 排序
list.sort(Comparator.comparing(FiledAndOrder::getOrder));
for (int i = 0; i < list.size(); i++) {
if (existOrderNumSet.contains(list.get(i).getOrder())) {
throw new RuntimeException(pojo.getName() + “类内部或与父类字段中存在重复的 excel 排序,请修改”);
}
existOrderNumSet.add(list.get(i).getOrder());
fieldAndOrderMap.put(list.get(i).getFiledName(), i);
}
this.fieldAndOrderMap = fieldAndOrderMap;
}
/**
* 功能:初始化类的字段信息,转换成 ArrayList
* @return java.util.List<com.gridsum.ad.ooh.project.entity.FiledAndOrder>
* @author kangshuai@gridsum.com
* @date 2021/4/9 12:09
*/
private List initList(List list, Class<?> pojoClass) {
if (Object.class.equals(pojoClass)) {
return list;
}
Field[] fields = pojoClass.getDeclaredFields();
for (Field f : fields) {
// 找到所有加了 Excel 注解的字段
Excel annotation = f.getAnnotation(Excel.class);
if (annotation == null) {
continue;
}
// 过滤隐藏行
if (annotation.isColumnHidden()) {
continue;
}
FiledAndOrder filedAndOrder = new FiledAndOrder(f.getName(), Integer.parseInt(annotation.orderNum()));
list.add(filedAndOrder);
}
// 递归查找父类
Class<?> superclass = pojoClass.getSuperclass();
return initList(list, superclass);
}
public HashMap<String, Integer> getFieldAndOrderMap() {
return fieldAndOrderMap;
}
}
设置验证规则
setValidation 方法有两个参数,第一个是导出 Excel 对应的实体类的类类型,第二个是 FieldOrderMappingHelper.getFieldAndOrderMap()
获取到的列名和排序映射,该类通过反射字段上的注解,自动为生成的 workbook 添加验证规则
public class ExcelStyleHelper {
/**
* 添加列值验证的最小行
*/
public static final int EXCEL_VALID_ROW_MIN = 1;
/**
* 添加列值验证的最大行
*/
public static final int EXCEL_VALID_ROW_MAX = (2 << 15) - 1;
/**
* Excel 对象
*/
private Workbook workbook;
/**
* Sheet 页,默认取第一个 sheet 页
*/
private Sheet sheet;
public ExcelStyleHelper(Workbook workbook) {
this.workbook = workbook;
this.sheet = workbook.getSheetAt(0);
}
/**
* 功能:单元格添加下拉框,仅支持 xls
* @author kangshuai@gridsum.com
* @date 2021/4/8 18:55
*/
public void setValidation(Class<?> pojoClass, HashMap<String, Integer> map) {
// 递归到 Object 就停下
自我介绍一下,小编13年上海交大毕业,曾经在小公司待过,也去过华为、OPPO等大厂,18年进入阿里一直到现在。
深知大多数Java工程师,想要提升技能,往往是自己摸索成长或者是报班学习,但对于培训机构动则几千的学费,着实压力不小。自己不成体系的自学效果低效又漫长,而且极易碰到天花板技术停滞不前!
因此收集整理了一份《2024年Java开发全套学习资料》,初衷也很简单,就是希望能够帮助到想自学提升又不知道该从何学起的朋友,同时减轻大家的负担。
既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,基本涵盖了95%以上Java开发知识点,真正体系化!
由于文件比较大,这里只是将部分目录截图出来,每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频,并且会持续更新!
如果你觉得这些内容对你有帮助,可以扫码获取!!(备注:Java)
更多:Java进阶核心知识集
包含:JVM,JAVA集合,网络,JAVA多线程并发,JAVA基础,Spring原理,微服务,Zookeeper,Kafka,RabbitMQ,Hbase,MongoDB,Cassandra,设计模式,负载均衡,数据库,一致性哈希,JAVA算法,数据结构,加密算法,分布式缓存等等
高效学习视频
《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取!
这些内容对你有帮助,可以扫码获取!!(备注:Java)**
[外链图片转存中…(img-oUudx4YG-1713844890035)]
更多:Java进阶核心知识集
包含:JVM,JAVA集合,网络,JAVA多线程并发,JAVA基础,Spring原理,微服务,Zookeeper,Kafka,RabbitMQ,Hbase,MongoDB,Cassandra,设计模式,负载均衡,数据库,一致性哈希,JAVA算法,数据结构,加密算法,分布式缓存等等
[外链图片转存中…(img-Q44Pwmj7-1713844890035)]
高效学习视频
《互联网大厂面试真题解析、进阶开发核心学习笔记、全套讲解视频、实战项目源码讲义》点击传送门即可获取!