问题描述
在做系统的导入导出模块时需要在导入时判断用户导入的表格是否符合给定的模板,该系统导入导出使用的是EasyExcel,因此在实现该功能时是基于EasyExcel的
解决方案
创建Spring Boot项目,并添加如下依赖
<dependency><groupId>com.alibaba</groupId><artifactId>easyexcel</artifactId><version>2.2.7</version></dependency>
创建导入时数据对应的实体类,并使用@ExcelProperty注解配置模板表头
public class EasyExcelData {@ExcelProperty(value = "学号", index = 0)private String no;@ExcelProperty(value = "姓名", index = 1)private String name;@ExcelProperty(value = "性别", index = 2)private String gender;private String errorReason;public String getNo() {return no;}public void setNo(String no) {this.no = no;}public String getName() {return name;}public void setName(String name) {this.name = name;}public String getGender() {return gender;}public void setGender(String gender) {this.gender = gender;}public String getErrorReason() {return errorReason;}public void setErrorReason(String errorReason) {this.errorReason = errorReason;}@Overridepublic String toString() {return "EasyExcelData{" +"no='" + no + '\'' +", name='" + name + '\'' +", gender='" + gender + '\'' +", errorReason='" + errorReason + '\'' +'}';}}
创建监听类并继承AnalysisEventListener类,重写invokeHeadMap方法,在该方法中判断是否符合模板
public class EasyExcelDemoListener extends AnalysisEventListener<EasyExcelData> {private List<EasyExcelData> correctData = new ArrayList<>();private List<EasyExcelData> errorData = new ArrayList<>();/** * 在这里进行模板的判断 * @param headMap 存放着导入表格的表头,键是索引,值是名称 * @param context */@Overridepublic void invokeHeadMap(Map<Integer, String> headMap, AnalysisContext context) {/*count 记录模板表头有几个,用以判断用户导入的表格是否和模板完全一致如果用户导入表格较模板的表头多,但其余符合模板,这样不影响则不需要 */int count = 0;// 获取数据实体的字段列表Field[] fields = EasyExcelData.class.getDeclaredFields();// 遍历字段进行判断for (Field field : fields) {// 获取当前字段上的ExcelProperty注解信息ExcelProperty fieldAnnotation = field.getAnnotation(ExcelProperty.class);// 判断当前字段上是否存在ExcelProperty注解if (fieldAnnotation != null) {++count;// 存在ExcelProperty注解则根据注解的index索引到表头中获取对应的表头名String headName = headMap.get(fieldAnnotation.index());// 判断表头是否为空或是否和当前字段设置的表头名不相同if (StringUtils.isEmpty(headName) || !headName.equals(fieldAnnotation.value()[0])) {// 如果为空或不相同,则抛出异常不再往下执行throw new RuntimeException("模板错误,请检查导入模板");}}}// 判断用户导入表格的标题头是否完全符合模板if (count != headMap.size()) {throw new RuntimeException("模板错误,请检查导入模板");}}/** * 在这里可以进行数据的解析,每读取一条数据都会执行这个方法 * @param easyExcelData 当前读取的数据 * @param analysisContext */@Overridepublic void invoke(EasyExcelData easyExcelData, AnalysisContext analysisContext) {if (StringUtils.isEmpty(easyExcelData.getNo())) {easyExcelData.setErrorReason("学号为空");errorData.add(easyExcelData);} else if (StringUtils.isEmpty(easyExcelData.getName())) {easyExcelData.setErrorReason("姓名为空");errorData.add(easyExcelData);} else if (StringUtils.isEmpty(easyExcelData.getGender())) {easyExcelData.setErrorReason("性别为空");errorData.add(easyExcelData);} else {correctData.add(easyExcelData);}}/** * 这里是读取完成后执行的方法 * @param analysisContext */@Overridepublic void doAfterAllAnalysed(AnalysisContext analysisContext) {// 打印没有错误的数据System.out.println(correctData);// 打印有错误的数据System.out.println(errorData);}}
进行测试,测试用到的表格可到EasyExcel判断导入时是否符合给定模板测试表格获取
@Testpublic void testEasyExcelReadError() {// 读取错误表格String fileName = "excel" + File.separator + "demo-error.xlsx";EasyExcel.read(fileName, EasyExcelData.class, new EasyExcelDemoListener()).sheet().doRead();}@Testpublic void testEasyExcelReadCorrect() {// 读取正确表格String fileName = "excel" + File.separator + "demo-correct.xlsx";EasyExcel.read(fileName, EasyExcelData.class, new EasyExcelDemoListener()).sheet().doRead();}