1、工单导入剔除重复数据

jeecg-boot-base/jeecg-boot-base-core/src/main/java/org/jeecg/modules/workorderTemplate/service/impl/WorkorderExcelServiceImpl.java

@@ -1,6 +1,7 @@
 package org.jeecg.modules.workorderTemplate.service.impl;
 
 import cn.hutool.core.date.DateTime;
+import org.apache.commons.lang.StringUtils;
 import org.apache.poi.ss.usermodel.*;
 import org.apache.poi.ss.util.CellRangeAddress;
 import org.apache.poi.xssf.streaming.SXSSFWorkbook;
@@ -39,6 +40,7 @@ import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.io.InputStream;
 import java.util.ArrayList;
+import java.util.Iterator;
 import java.util.List;
 
 @Service
@@ -82,9 +84,40 @@ public class WorkorderExcelServiceImpl implements WorkorderExcelService {
     /******************************************************************************************************************/
     @Override
     public Result<?> templateImport1(List<ImportBean> importBeanList) {
+        //返回信息
+        String message = "数据导入失败！";
+        //导入数据
         List<WorkorderTemplate> templateList = new ArrayList<>();
 
         try {
+            //0、重复数据剔除
+            List<String> taskNos = new ArrayList<>();
+            for (ImportBean importBean : importBeanList) {
+                taskNos.add(importBean.getTaskNo());
+            }
+            List<WorkorderTemplate> list = workorderTemplateService
+                    .lambdaQuery()
+                    .in(WorkorderTemplate::getTaskNo, taskNos)
+                    .list();
+            StringBuffer buffer = new StringBuffer();//重复任务号
+            Iterator<ImportBean> it_b=importBeanList.iterator();
+            while(it_b.hasNext()){
+                ImportBean a=it_b.next();
+                for (WorkorderTemplate template : list) {
+                    if (a.getTaskNo().equals(template.getTaskNo())) {
+                        buffer.append(a.getTaskNo() + "、");
+                        it_b.remove();
+                        break;
+                    }
+                }
+            }
+
+            if(StringUtils.isEmpty(buffer.toString())){
+                message = "数据全部导入成功！";
+            }else {
+                message = "以下任务号已存在："+buffer + "请检查数据！";
+            }
+
             //1、获取excel数据
             for (ImportBean importBean : importBeanList) {
                 WorkorderTemplate workorderTemplate = new WorkorderTemplate();
@@ -102,14 +135,15 @@ public class WorkorderExcelServiceImpl implements WorkorderExcelService {
 
             //2、将数据录入到数据库
             Boolean result = workorderTemplateService.saveBatch(templateList);
-            if(result){
-                return Result.OK("数据导入成功！");
-            }else {
-                return Result.error("数据导入失败！");
+            if(!result){
+                message = "数据导入失败";
             }
+
+            return Result.OK(message);
+
         } catch (Exception e) {
             e.printStackTrace();
-            return Result.error("数据导入失败！");
+            return Result.error(message);
         }
     }