From 042936ba89e122bece0e4ce595c68f859b9e9bed Mon Sep 17 00:00:00 2001
From: 龚焕茏 <2842157468@qq.com>
Date: 星期三, 17 四月 2024 17:53:43 +0800
Subject: [PATCH] 工单下发：新增互斥锁、下发记录
---
ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java | 122 +++++++++++++++++++++++++++++++---------
1 files changed, 94 insertions(+), 28 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 f136b80..670853e 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,42 @@
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.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 +54,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;
/**
* 娣诲姞
@@ -289,6 +292,7 @@
break;
}
+ // 鏌ヨ绗﹀悎鏉′欢鐨勫伐鍗�
List<Integer> ids = new LambdaQueryChainWrapper<>(baseMapper)
.eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)
.eq(Objects.nonNull(data.getUnitId()), WorkOrder::getUnitId, data.getUnitId())
@@ -303,24 +307,86 @@
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);
+ return Result.ok("鎴愬姛涓嬪彂" + ids.size() + "鏉″伐鍗�");
+ } catch (Exception e) {
+ return Result.ok("鎿嶄綔澶辫触");
+ } 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("鎿嶄綔鎴愬姛");
+ if (!getDistributeLock()) {
+ return Result.error("宸ュ崟涓嬪彂涓紝璇风◢鍚庨噸璇�");
+ }
+ try {
+ if (ids.isEmpty()) {
+ ids = new LambdaQueryChainWrapper<>(baseMapper)
+ .eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)
+ .eq(Objects.nonNull(unitId), WorkOrder::getUnitId, unitId)
+ .select(WorkOrder::getId)
+ .list()
+ .stream()
+ .map(WorkOrder::getId)
+ .collect(Collectors.toList());
+ }
+ new LambdaUpdateChainWrapper<>(baseMapper)
+ .set(WorkOrder::getStatus, WorkOrderStatusEnum.DISTRIBUTED)
+ .in(WorkOrder::getId, ids)
+ .eq(WorkOrder::getStatus, WorkOrderStatusEnum.WAIT_DISTRIBUTE)
+ .eq(Objects.nonNull(unitId), WorkOrder::getUnitId, unitId)
+ .update();
+ addDistributeRecord(ids);
+ return Result.ok("鎿嶄綔鎴愬姛");
+ } catch (Exception e) {
+ return Result.ok("鎿嶄綔澶辫触");
+ } 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) {
+ LoginUser loginUser = SecurityUtils.getLoginUser();
+ workOrderDistributeRecordMapper.insertBatch(
+ ids.stream()
+ .map(id -> new WorkOrderDistributeRecord(id, loginUser.getUserId()))
+ .toList()
+ );
}
/**
--
Gitblit v1.8.0