导入模板校验表头和空数据

胡翰林
1 parent 158d9f1f
Showing 5 changed files with 72 additions and 5 deletions
src/main/java/com/ash/excelData/CaseExcelData.java
src/main/java/com/ash/listener/CaseExcelListener.java
src/main/java/com/ash/listener/WarningInstanceExcelListener.java
src/main/java/com/ash/service/CaseService.java
src/main/java/com/ash/service/WarningInstanceService.java
--- a/src/main/java/com/ash/excelData/CaseExcelData.java
View file @68ae793
+++ b/src/main/java/com/ash/excelData/CaseExcelData.java
View file @68ae793
@@ -5,9 +5,7 @@ import com.alibaba.excel.annotation.write.style.ColumnWidth;
 import com.alibaba.excel.annotation.write.style.ContentRowHeight;
 import com.alibaba.excel.annotation.write.style.ContentStyle;
 import com.alibaba.excel.annotation.write.style.HeadRowHeight;
-import com.ash.base.excelOpt.DatetimeConverter;
 import com.ash.base.excelOpt.ExcelCheck;
-import com.baomidou.mybatisplus.annotation.TableField;
 import lombok.AllArgsConstructor;
 import lombok.Builder;
 import lombok.Data;
--- a/src/main/java/com/ash/listener/CaseExcelListener.java
View file @68ae793
+++ b/src/main/java/com/ash/listener/CaseExcelListener.java
View file @68ae793
@@ -17,6 +17,8 @@ import com.ash.base.excelOpt.ValidateAnnotation;
 import com.ash.entity.dao.CaseMapper;
 import com.ash.excelData.CaseExcelData;
 import com.ash.service.CaseService;
+import com.sun.javafx.binding.StringFormatter;
+import javafx.beans.binding.StringExpression;
 import lombok.Getter;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.collections.CollectionUtils;
@@ -48,6 +50,7 @@ public class CaseExcelListener extends AnalysisEventListener<CaseExcelData>
     private final Map<Field, ValidateAnnotation> fieldMap = new HashMap<>();
     private boolean existValidate = true;
+    private boolean headCheck = true;
     private static final int CORE_POOL_SIZE = 5;// 核心线程数
     private static final int MAX_POOL_SIZE = 10;// 最大线程数
@@ -75,6 +78,7 @@ public class CaseExcelListener extends AnalysisEventListener<CaseExcelData>
     @Override
     public void invokeHead(Map<Integer, ReadCellData<?>> headMap, AnalysisContext context) {
         ExcelReadHeadProperty excelReadHeadProperty = context.currentReadHolder().excelReadHeadProperty();
+        StringBuilder headError = new StringBuilder();
         for (Head head : excelReadHeadProperty.getHeadMap().values()) {
             Field field = head.getField();
             ExcelProperty excelProperty = field.getAnnotation(ExcelProperty.class);
@@ -84,6 +88,24 @@ public class CaseExcelListener extends AnalysisEventListener<CaseExcelData>
                 field.setAccessible(true);
                 fieldMap.put(field, new ValidateAnnotation(excelProperty, excelCheck, headName));
             }
+
+            //校验表头
+            if (excelProperty != null) {
+                int epIndex = excelProperty.index();
+                String headDescription = excelProperty.value()[0];
+                if (!headMap.containsKey(epIndex)) {
+                    headError.append(String.format(" 第%s列缺失,应为:%s"
+                            , epIndex + 1, headDescription));
+                } else if (!headMap.get(epIndex).getStringValue().equals(headDescription)) {
+                    headError.append(String.format(" 第%s列错误,%s 应为:%s"
+                            , epIndex + 1, headMap.get(epIndex).getStringValue(), headDescription));
+                }
+            }
+        }
+
+        if (StringUtils.isNotBlank(headError.toString())) {
+            headCheck = false;
+            addError(0, "模板错误请使用下载的模板导入!" + headError);
         }
         if (fieldMap.isEmpty()) {
             existValidate = false;
@@ -103,6 +125,9 @@ public class CaseExcelListener extends AnalysisEventListener<CaseExcelData>
     @Override
     public void invoke(CaseExcelData data, AnalysisContext analysisContext) {
+        if (!headCheck) {
+            return;
+        }
         log.info("接收案件信息" + data);
         if (data != null) {
             if (existValidate) {
@@ -129,6 +154,12 @@ public class CaseExcelListener extends AnalysisEventListener<CaseExcelData>
         if (CollectionUtils.isNotEmpty(errorList)) {
             throw new ExcelAnalysisException("数据校验错误!");
         }
+
+        if (CollectionUtils.isEmpty(list)) {
+            addError(0, "模板数据为空!");
+            throw new ExcelAnalysisException("模板数据为空!");
+        }
+
         log.info("解析结束，开始插入数据");
         ExecutorService executor = new ThreadPoolExecutor(CORE_POOL_SIZE, MAX_POOL_SIZE,
                 KEEP_ALIVE_TIME, TimeUnit.SECONDS,
@@ -149,7 +180,7 @@ public class CaseExcelListener extends AnalysisEventListener<CaseExcelData>
                 endPosition = (i + 1) * singleThreadDealcount;
             }
             CaseMapper caseMapper = SpringJobBeanFactory.getBean(CaseMapper.class);
-            CaseService caseService= SpringJobBeanFactory.getBean(CaseService.class);
+            CaseService caseService = SpringJobBeanFactory.getBean(CaseService.class);
             CaseThread thread = new CaseThread(count, caseMapper, caseService, list, startPosition, endPosition);
             executor.execute(thread);
         }
--- a/src/main/java/com/ash/listener/WarningInstanceExcelListener.java
View file @68ae793
+++ b/src/main/java/com/ash/listener/WarningInstanceExcelListener.java
View file @68ae793
@@ -48,6 +48,7 @@ public class WarningInstanceExcelListener extends AnalysisEventListener<WarningI
     private final Map<Field, ValidateAnnotation> fieldMap = new HashMap<>();
     private boolean existValidate = true;
+    private boolean headCheck = true;
     private static final int CORE_POOL_SIZE = 5;// 核心线程数
     private static final int MAX_POOL_SIZE = 10;// 最大线程数
@@ -75,6 +76,7 @@ public class WarningInstanceExcelListener extends AnalysisEventListener<WarningI
     @Override
     public void invokeHead(Map<Integer, ReadCellData<?>> headMap, AnalysisContext context) {
         ExcelReadHeadProperty excelReadHeadProperty = context.currentReadHolder().excelReadHeadProperty();
+        StringBuilder headError = new StringBuilder();
         for (Head head : excelReadHeadProperty.getHeadMap().values()) {
             Field field = head.getField();
             ExcelProperty excelProperty = field.getAnnotation(ExcelProperty.class);
@@ -84,7 +86,27 @@ public class WarningInstanceExcelListener extends AnalysisEventListener<WarningI
                 field.setAccessible(true);
                 fieldMap.put(field, new ValidateAnnotation(excelProperty, excelCheck, headName));
             }
+
+            //校验表头
+            if (excelProperty != null) {
+                int epIndex = excelProperty.index();
+                String headDescription = excelProperty.value()[0];
+                if (!headMap.containsKey(epIndex)) {
+                    headError.append(String.format(" 第%s列缺失,应为:%s"
+                            , epIndex + 1, headDescription));
+                } else if (!headMap.get(epIndex).getStringValue().equals(headDescription)) {
+                    headError.append(String.format(" 第%s列错误,%s 应为:%s"
+                            , epIndex + 1, headMap.get(epIndex).getStringValue(), headDescription));
+                }
+            }
+
+        }
+
+        if (StringUtils.isNotBlank(headError.toString())) {
+            headCheck = false;
+            addError(0, "模板错误请使用下载的模板导入!" + headError);
         }
+
         if (fieldMap.isEmpty()) {
             existValidate = false;
         }
@@ -103,6 +125,9 @@ public class WarningInstanceExcelListener extends AnalysisEventListener<WarningI
     @Override
     public void invoke(WarningInstanceExcelData data, AnalysisContext analysisContext) {
+        if (!headCheck) {
+            return;
+        }
         log.info("接收警情信息" + data);
         if (data != null) {
             if (existValidate) {
@@ -128,6 +153,13 @@ public class WarningInstanceExcelListener extends AnalysisEventListener<WarningI
         if (CollectionUtils.isNotEmpty(errorList)) {
             throw new ExcelAnalysisException("数据校验错误!");
         }
+
+        if (CollectionUtils.isEmpty(list)) {
+            addError(0, "模板数据为空!");
+            throw new ExcelAnalysisException("模板数据为空!");
+        }
+
+
         log.info("解析结束，开始插入数据");
         ExecutorService executor = new ThreadPoolExecutor(CORE_POOL_SIZE, MAX_POOL_SIZE,
                 KEEP_ALIVE_TIME, TimeUnit.SECONDS,
--- a/src/main/java/com/ash/service/CaseService.java
View file @68ae793
+++ b/src/main/java/com/ash/service/CaseService.java
View file @68ae793
@@ -3,6 +3,7 @@ package com.ash.service;
 import com.alibaba.excel.EasyExcel;
 import com.alibaba.excel.exception.ExcelAnalysisException;
 import com.alibaba.excel.exception.ExcelDataConvertException;
+import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
 import com.ash.base.*;
 import com.ash.base.excelOpt.ExcelErrorMessage;
@@ -683,7 +684,9 @@ public class CaseService {
             return listener.getErrorList();
         } catch (ExcelDataConvertException e) {
             List<ExcelErrorMessage> errorList = new ArrayList<>();
-            ExcelErrorMessage eem = new ExcelErrorMessage().setMessage("模板错误");
+            String error = String.format("第%s行，第%s列解析异常，数据为:%s", e.getRowIndex(),
+                    e.getColumnIndex(), JSON.toJSONString(e.getCellData()));
+            ExcelErrorMessage eem = new ExcelErrorMessage().setMessage("数据类型错误!" + error);
             errorList.add(eem);
             return errorList;
         } catch (Exception ex) {
--- a/src/main/java/com/ash/service/WarningInstanceService.java
View file @68ae793
+++ b/src/main/java/com/ash/service/WarningInstanceService.java
View file @68ae793
@@ -3,6 +3,7 @@ package com.ash.service;
 import com.alibaba.excel.EasyExcel;
 import com.alibaba.excel.exception.ExcelAnalysisException;
 import com.alibaba.excel.exception.ExcelDataConvertException;
+import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
 import com.ash.base.*;
 import com.ash.base.excelOpt.ExcelErrorMessage;
@@ -410,7 +411,9 @@ public class WarningInstanceService {
             return listener.getErrorList();
         } catch (ExcelDataConvertException e) {
             List<ExcelErrorMessage> errorList = new ArrayList<>();
-            ExcelErrorMessage eem = new ExcelErrorMessage().setMessage("模板错误");
+            String error = String.format("第%s行，第%s列解析异常，数据为:%s", e.getRowIndex(),
+                    e.getColumnIndex(), JSON.toJSONString(e.getCellData()));
+            ExcelErrorMessage eem = new ExcelErrorMessage().setMessage("数据类型错误!" + error);
             errorList.add(eem);
             return errorList;
         } catch (Exception ex) {