导入基础数据做格式验证

陈威
1 parent e9764e35
Showing 2 changed files with 32 additions and 11 deletions Show diff stats
zr-cloud/zr-modules/zr-schsf/src/main/java/com/chinagas/modules/schsf/utils/listener/McBasicCmtyImportInspection.java
zr-cloud/zr-modules/zr-schsf/src/main/java/com/chinagas/modules/schsf/utils/listener/McBasicMoneyImportInspection.java
@@ -5,18 +5,14 @@ import com.alibaba.excel.event.AnalysisEventListener;
 import com.alibaba.excel.exception.ExcelDataConvertException;
 import com.chinagas.common.core.utils.StringUtils;
 import com.chinagas.modules.schsf.domain.McBasicCmty;
-import com.chinagas.modules.schsf.service.ICommonService;
-import org.springframework.beans.factory.annotation.Autowired;
-import java.util.HashMap;
+import java.util.Map;
-public class McBasicCmtyImportInspection extends AnalysisEventListener {
+public class McBasicCmtyImportInspection extends AnalysisEventListener<McBasicCmty> {
     @Override
-    public void invoke(Object data, AnalysisContext context) {
-
-        McBasicCmty mcBasicCmty = (McBasicCmty) data;
-        System.out.println(mcBasicCmty);
+    public void invoke(McBasicCmty data, AnalysisContext context) {
+        McBasicCmty mcBasicCmty = data;
         if (mcBasicCmty.getFiscalYear() == null) {
             throw new RuntimeException(String.format("第%s行  财年  为空，请核实", context.readRowHolder().getRowIndex() + 1));
         }
@@ -58,6 +54,20 @@ public class McBasicCmtyImportInspection extends AnalysisEventListener {
     @Override
+    public void invokeHeadMap(Map<Integer, String> headMap, AnalysisContext context) {
+//        // 检查标题行，headMap 是一个 Map，key 是列索引，value 是列标题
+//        if (headMap.containsKey(0) && "预期标题".equals(headMap.get(0))) { // 假设预期的标题在第一列
+//
+//        }
+        if (headMap==null || headMap.size()!=36){
+            throw new RuntimeException(String.format("导入文件格式错误，请检查"));
+        }
+    }
+
+
+
+
+    @Override
     public void doAfterAllAnalysed(AnalysisContext analysisContext) {
     }
@@ -6,13 +6,15 @@ import com.alibaba.excel.exception.ExcelDataConvertException;
 import com.chinagas.common.core.utils.StringUtils;
 import com.chinagas.modules.schsf.domain.McBasicMoney;
-public class McBasicMoneyImportInspection extends AnalysisEventListener {
+import java.util.Map;
+
+public class McBasicMoneyImportInspection extends AnalysisEventListener<McBasicMoney>  {
     @Override
-    public void invoke(Object data, AnalysisContext context) {
-        McBasicMoney mcBasicMoney = (McBasicMoney) data;
+    public void invoke(McBasicMoney data, AnalysisContext context) {
+        McBasicMoney mcBasicMoney =  data;
         if (StringUtils.isEmpty(mcBasicMoney.getCompanyName())) {
             throw new RuntimeException(String.format("第%s行  项目公司  为空，请核实", context.readRowHolder().getRowIndex() + 1));
         }
@@ -40,4 +42,13 @@ public class McBasicMoneyImportInspection extends AnalysisEventListener {
     public void doAfterAllAnalysed(AnalysisContext analysisContext) {
     }
+
+
+    @Override
+    public void invokeHeadMap(Map<Integer, String> headMap, AnalysisContext context) {
+        if (headMap==null || headMap.size()!=33){
+            throw new RuntimeException(String.format("导入文件格式错误，请检查"));
+        }
+    }
+
 }