From 4c55edc6f53989beb03bf36aabb469d76470b517 Mon Sep 17 00:00:00 2001
From: lohir <3399054449@qq.com>
Date: 星期三, 23 十月 2024 15:43:48 +0800
Subject: [PATCH] 完成创建工单时对工单进行白名单判断

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java |   36 ++++++++++++++++++++++++------------
 1 files changed, 24 insertions(+), 12 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
index aa99ac7..8585b6e 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
@@ -110,6 +110,11 @@
     @Transactional(rollbackFor = Exception.class)
     public synchronized Boolean innerAddWorkOrder(List<WorkOrder> workOrderList) {
         int total = workOrderList.size();
+        // 鏌ヨ鍑虹櫧鍚嶅崟鍒楄〃
+        List<String> serialNumbers = workOrderWhiteMapper.selectList().stream().map(WorkOrderWhite::getSerialNumber).collect(Collectors.toList());
+        // 閬嶅巻宸ュ崟鍒楄〃锛屽垽鏂槸鍚﹀湪鐧藉悕鍗曚腑
+        workOrderList.stream().filter(item -> !serialNumbers.contains(item.getSerialNumber()))
+                        .collect(Collectors.toList());
         workOrderList.stream().filter(item -> {
             return StringUtils.hasText(item.getSerialNumber()) && Objects.nonNull(item.getStatus()) && !CollectionUtils.isEmpty(item.getErrorTypeList());
         });
@@ -324,19 +329,26 @@
             return Result.error("宸ュ崟姝e湪娣诲姞锛岃绋嶅悗鍐嶈瘯");
         }
         WorkOrder entity = WorkOrderForm.getEntityByForm(form, null);
-        entity.setCreateTime(DateUtils.getNowDate());
-        entity.setStatus(WorkOrderStatusEnum.WAIT_DISTRIBUTE);
-        Date now = new Date();
-        entity.setCreateTime(now);
-        entity.setUpdateTime(now);
-        entity.setWorkOrderNo(IdUtils.workOrderNO(now, String.format("%05d", workOrderNum)));
-        entity.setErrorType(String.join(",", form.getErrorType()));
-        List<WorkOrderErrorType> workOrderErrorTypes = form.getErrorType().stream().map(errorType -> new WorkOrderErrorType(entity.getWorkOrderNo(), errorType)).toList();
-        workOrderErrorTypeService.getBaseMapper().insertWorkOrderErrorTypeList(workOrderErrorTypes);
-        if (baseMapper.insert(entity) > 0) {
-            return Result.ok("娣诲姞鎴愬姛");
+        // 鏌ヨ鍑虹櫧鍚嶅崟鍒楄〃
+        List<String> serialNumbers = workOrderWhiteMapper.selectList().stream().map(WorkOrderWhite::getSerialNumber).collect(Collectors.toList());
+        // 閬嶅巻宸ュ崟鍒楄〃锛屽垽鏂槸鍚﹀湪鐧藉悕鍗曚腑
+         if (serialNumbers.contains(entity.getSerialNumber())){
+            return Result.error("璇ヨ澶囧湪鐧藉悕鍗曚腑锛屼笉鑳芥坊鍔犲伐鍗�");
+        }else {
+            entity.setCreateTime(DateUtils.getNowDate());
+            entity.setStatus(WorkOrderStatusEnum.WAIT_DISTRIBUTE);
+            Date now = new Date();
+            entity.setCreateTime(now);
+            entity.setUpdateTime(now);
+            entity.setWorkOrderNo(IdUtils.workOrderNO(now, String.format("%05d", workOrderNum)));
+            entity.setErrorType(String.join(",", form.getErrorType()));
+            List<WorkOrderErrorType> workOrderErrorTypes = form.getErrorType().stream().map(errorType -> new WorkOrderErrorType(entity.getWorkOrderNo(), errorType)).toList();
+            workOrderErrorTypeService.getBaseMapper().insertWorkOrderErrorTypeList(workOrderErrorTypes);
+            if (baseMapper.insert(entity) > 0) {
+                return Result.ok("娣诲姞鎴愬姛");
+            }
+            return Result.error("娣诲姞澶辫触");
         }
-        return Result.error("娣诲姞澶辫触");
     }
 
     /**

--
Gitblit v1.8.0