From 4da3b0321291a1ad8a1341f8f843a41d4b516903 Mon Sep 17 00:00:00 2001
From: 龚焕茏 <2842157468@qq.com>
Date: 星期三, 24 四月 2024 18:09:11 +0800
Subject: [PATCH] 合同规则分三层、合同积分新增

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java |  136 +++++++++++++++++++++++++++++++++++----------
 1 files changed, 105 insertions(+), 31 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 e311724..e29849f 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
@@ -1,44 +1,43 @@
 package com.ycl.platform.service.impl;
 
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
 import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ycl.exception.ServiceException;
 import com.ycl.platform.domain.entity.*;
 import com.ycl.platform.domain.form.*;
 import com.ycl.platform.domain.query.DistributeWorkOrderQuery;
+import com.ycl.platform.domain.query.WorkOrderQuery;
 import com.ycl.platform.domain.vo.DistributeWorkOrderVO;
+import com.ycl.platform.domain.vo.WorkOrderVO;
 import com.ycl.platform.mapper.*;
 import com.ycl.platform.service.NotifyService;
 import com.ycl.platform.service.WorkOrderService;
 import com.ycl.system.Result;
-import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.ycl.platform.domain.vo.WorkOrderVO;
-import com.ycl.platform.domain.query.WorkOrderQuery;
-
-import java.time.LocalDateTime;
-import java.time.ZoneId;
-import java.time.temporal.ChronoUnit;
-import java.util.Arrays;
-import java.util.List;
-
-import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.ycl.system.model.LoginUser;
 import com.ycl.system.page.PageUtil;
 import com.ycl.utils.DateUtils;
 import com.ycl.utils.SecurityUtils;
-import enumeration.general.FastDistributeTypeEnum;
+import com.ycl.utils.redis.RedisCache;
 import enumeration.general.NotifyTypeEnum;
+import enumeration.general.WorkOrderDistributeWayEnum;
 import enumeration.general.WorkOrderStatusEnum;
+import lombok.RequiredArgsConstructor;
 import org.apache.commons.lang3.ArrayUtils;
-import org.springframework.stereotype.Service;
 import org.springframework.beans.BeanUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.Assert;
-import java.util.ArrayList;
-import java.util.Objects;
-import java.util.stream.Collectors;
-
-import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
-import lombok.RequiredArgsConstructor;
 import org.springframework.util.StringUtils;
+
+import java.time.LocalDateTime;
+import java.time.ZoneId;
+import java.util.List;
+import java.util.Objects;
+import java.util.concurrent.TimeUnit;
+import java.util.stream.Collectors;
 
 /**
  * 宸ュ崟 鏈嶅姟瀹炵幇绫�
@@ -56,7 +55,12 @@
     private final WorkOrderAuditingRecordMapper workOrderAuditingRecordMapper;
     private final WorkOrderYwConditionRecordMapper workOrderYwConditionRecordMapper;
     private final NotifyService notifyService;
+    private final WorkOrderDistributeRecordMapper workOrderDistributeRecordMapper;
 
+    private final String DISTRIBUTE_LOCK_KEY = "distributeLock";
+
+    @Autowired
+    private RedisCache redisCache;
 
     /**
      * 娣诲姞
@@ -110,8 +114,12 @@
         workOrderAuditingRecord.setRemark(form.getAuditingRemark());
         workOrderAuditingRecordMapper.insert(workOrderAuditingRecord);
         // 娣诲姞鏂伴�氱煡
-        Notify notify = new Notify(NotifyTypeEnum.WORK_ORDER, form.getAuditingResult().getDesc(), workOrder.getYwPeopleId(), "0", "0", workOrder.getId());
-        notifyService.save(notify);
+        try {
+            Notify notify = new Notify(NotifyTypeEnum.WORK_ORDER, form.getAuditingResult().getDesc(), workOrder.getYwPeopleId(), "0", "0", workOrder.getId());
+            notifyService.save(notify);
+        } catch (Exception e) {
+            return Result.error("瀹℃牳鎴愬姛锛岄�氱煡杩愮淮浜哄憳澶辫触");
+        }
         return Result.ok("鎿嶄綔鎴愬姛");
     }
 
@@ -285,6 +293,7 @@
                 break;
            }
 
+        // 鏌ヨ绗﹀悎鏉′欢鐨勫伐鍗�
         List<Integer> ids = new LambdaQueryChainWrapper<>(baseMapper)
                 .eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)
                 .eq(Objects.nonNull(data.getUnitId()), WorkOrder::getUnitId, data.getUnitId())
@@ -299,24 +308,89 @@
 
         if (ids.isEmpty()) {
             return Result.error("娌℃湁绗﹀悎鏉′欢鐨勫伐鍗�");
-        } else {
+        }
+        if (!getDistributeLock()) {
+            return Result.error("姝ゅ埢鏈変汉涓嬪彂涓紝涓洪伩鍏嶅啿绐侊紝璇风◢鍚庨噸璇�");
+        }
+        try {
             new LambdaUpdateChainWrapper<>(baseMapper)
                     .set(WorkOrder::getStatus, WorkOrderStatusEnum.DISTRIBUTED)
                     .in(WorkOrder::getId, ids)
                     .update();
+            addDistributeRecord(ids, WorkOrderDistributeWayEnum.FAST_DISTRIBUTE);
+            return Result.ok("鎴愬姛涓嬪彂" + ids.size() + "鏉″伐鍗�");
+        } catch (Exception e) {
+            return Result.error("鎿嶄綔澶辫触");
+        } finally {
+            distributeUnLock();
         }
-        return Result.ok("鎴愬姛涓嬪彂" + ids.size() + "鏉″伐鍗�");
     }
 
     @Override
-    public Result selectedIdsDistribute(List<Integer> ids, Integer unitId) {
-        new LambdaUpdateChainWrapper<>(baseMapper)
-                .set(WorkOrder::getStatus, WorkOrderStatusEnum.DISTRIBUTED)
-                .in(ArrayUtils.isNotEmpty(ids.toArray()), WorkOrder::getId, ids)
-                .eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)
-                .eq(Objects.nonNull(unitId), WorkOrder::getUnitId, unitId)
-                .update();
-        return Result.ok("鎿嶄綔鎴愬姛");
+    public Result selectedIdsDistribute(DistributeWorkOrderQuery query) {
+        WorkOrderDistributeWayEnum distributeWayEnum = WorkOrderDistributeWayEnum.SELECTED_DISTRIBUTE;
+        if (!getDistributeLock()) {
+            return Result.error("姝ゅ埢鏈変汉涓嬪彂涓紝涓洪伩鍏嶅啿绐侊紝璇风◢鍚庨噸璇�");
+        }
+        try {
+            if (query.getIds().isEmpty()) {
+                query.setIds(new LambdaQueryChainWrapper<>(baseMapper)
+                        .eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)
+                        .eq(Objects.nonNull(query.getUnitId()), WorkOrder::getUnitId, query.getUnitId())
+                        .select(WorkOrder::getId)
+                        .list()
+                        .stream()
+                        .map(WorkOrder::getId)
+                        .collect(Collectors.toList()));
+                distributeWayEnum = WorkOrderDistributeWayEnum.ALL_DISTRIBUTE;
+            }
+            if (query.getIds().isEmpty()) {
+                return Result.error("娌℃湁宸ュ崟寰呬笅鍙�");
+            }
+            new LambdaUpdateChainWrapper<>(baseMapper)
+                    .set(WorkOrder::getStatus, WorkOrderStatusEnum.DISTRIBUTED)
+                    .in(WorkOrder::getId, query.getIds())
+                    .update();
+            addDistributeRecord(query.getIds(), distributeWayEnum);
+            return Result.ok("鎴愬姛涓嬪彂" + query.getIds().size() + "鏉″伐鍗�");
+        } catch (Exception e) {
+            return Result.error("鎿嶄綔澶辫触");
+        } finally {
+            distributeUnLock();
+        }
+    }
+
+    /**
+     * 鐢宠宸ュ崟涓嬪彂閿�
+     * @return 宸ュ崟涓嬪彂閿佺敵璇风粨鏋�
+     */
+    public synchronized Boolean getDistributeLock() {
+        if (Objects.isNull(redisCache.getCacheObject(DISTRIBUTE_LOCK_KEY))) {
+            redisCache.setCacheObject(DISTRIBUTE_LOCK_KEY, "1");
+            return true;
+        } else {
+            return false;
+        }
+    }
+
+    /**
+     * 宸ュ崟涓嬪彂閿侀噴鏀�
+     */
+    public synchronized void distributeUnLock() {
+        redisCache.deleteObject(DISTRIBUTE_LOCK_KEY);
+    }
+
+    /**
+     * 娣诲姞宸ュ崟涓嬪彂璁板綍
+     * @param ids 宸ュ崟id
+     */
+    private void addDistributeRecord(List<Integer> ids, WorkOrderDistributeWayEnum distributeWay) {
+        LoginUser loginUser = SecurityUtils.getLoginUser();
+        workOrderDistributeRecordMapper.insertBatch(
+                ids.stream()
+                        .map(id -> new WorkOrderDistributeRecord(id, loginUser.getUserId(), distributeWay))
+                        .toList()
+        );
     }
 
     /**

--
Gitblit v1.8.0