From 3c2ee852994cc3c4d861484f12d972b42cb0ffee Mon Sep 17 00:00:00 2001 From: eight <641137800@qq.com> Date: 星期三, 21 八月 2024 16:05:01 +0800 Subject: [PATCH] 并发问题 修改 --- jh-module-ecg/jh-module-ecg-biz/src/main/java/cn/lihu/jh/module/ecg/service/queue/queueServiceImpl.java | 134 ++++++++++++++++++++++++++------------------ 1 files changed, 80 insertions(+), 54 deletions(-) diff --git a/jh-module-ecg/jh-module-ecg-biz/src/main/java/cn/lihu/jh/module/ecg/service/queue/queueServiceImpl.java b/jh-module-ecg/jh-module-ecg-biz/src/main/java/cn/lihu/jh/module/ecg/service/queue/queueServiceImpl.java index 69d68d9..ceb0e55 100644 --- a/jh-module-ecg/jh-module-ecg-biz/src/main/java/cn/lihu/jh/module/ecg/service/queue/queueServiceImpl.java +++ b/jh-module-ecg/jh-module-ecg-biz/src/main/java/cn/lihu/jh/module/ecg/service/queue/queueServiceImpl.java @@ -2,11 +2,8 @@ import cn.lihu.jh.module.ecg.dal.dataobject.queue.QueueStatisticDO; import cn.lihu.jh.module.ecg.dal.dataobject.room.RoomDO; -import cn.lihu.jh.module.ecg.dal.dataobject.room.RoomStatisticsDO; import cn.lihu.jh.module.ecg.dal.mysql.room.RoomMapper; import cn.lihu.jh.module.ecg.enums.QueueStatusEnum; -import lombok.Data; -import org.jetbrains.annotations.NotNull; import org.springframework.stereotype.Service; import org.springframework.validation.annotation.Validated; @@ -17,12 +14,10 @@ import cn.lihu.jh.module.ecg.dal.mysql.queue.queueMapper; -import javax.annotation.PostConstruct; import javax.annotation.Resource; import java.util.*; -import java.util.concurrent.ConcurrentHashMap; -import java.util.concurrent.PriorityBlockingQueue; +import java.util.concurrent.*; import java.util.concurrent.atomic.AtomicInteger; import static cn.lihu.jh.framework.common.exception.util.ServiceExceptionUtil.exception; @@ -43,6 +38,8 @@ PriorityBlockingQueue<BedQueueBO> priorityQueue = new PriorityBlockingQueue<>(); ConcurrentHashMap<String, BedQueueBO > mapBedVsQueue = new ConcurrentHashMap<>(); + + ExecutorService singleThreadExecutor = Executors.newSingleThreadExecutor(); @Resource private queueMapper queueMapper; @@ -97,34 +94,6 @@ return queueMapper.queueStatistic(statusList); } - @Override - public void queue(QueueSaveReqVO queueSaveReqVO) { - BedQueueBO bedQueueBO = priorityQueue.peek(); - if (null == bedQueueBO) - return; - - if (bedQueueBO.queueNum.get() == bedQueueBO.maxQueueNum) { - queueSaveReqVO.setStatus(QueueStatusEnum.WAITING.getStatus()); //鎺掗槦涓� - QueueDO queue = BeanUtils.toBean(queueSaveReqVO, QueueDO.class); - queueMapper.insert(queue); // queue.getId(); - } else if (bedQueueBO.queueNum.get() < bedQueueBO.maxQueueNum) { - queueSaveReqVO.setStatus(QueueStatusEnum.READY.getStatus()); //鍊欒瘖鍑嗗涓� - queueSaveReqVO.setRoomId(bedQueueBO.getRoomId()); - queueSaveReqVO.setRoomName(bedQueueBO.getRoomName()); - queueSaveReqVO.setBedNo(bedQueueBO.getBedNo()); - queueSaveReqVO.setSeqNum(curSeqNum.get()); - QueueDO queue = BeanUtils.toBean(queueSaveReqVO, QueueDO.class); - queueMapper.insert(queue); // queue.getId(); - - curSeqNum.getAndIncrement(); - bedQueueBO.queueNum.getAndIncrement(); - BedQueueBO bedQueueBO2 = priorityQueue.poll(); - priorityQueue.offer(bedQueueBO2); - } - - - } - public void initQueue() { priorityQueue.clear(); mapBedVsQueue.clear(); @@ -133,7 +102,7 @@ List<RoomDO> roomDOList = roomMapper.simpleRoomList(); List<BedQueueBO> bedQueueBOList = roomDOList.stream().map(item -> BeanUtils.toBean(item, BedQueueBO.class)).toList(); - // 浠嶥B 鑾峰彇 闃熷垪涓� 灏辫瘖鍑嗗涓汉鍛� 鍒楄〃 + // 浠嶥B 鑾峰彇 闃熷垪涓� 灏辫瘖鍑嗗涓汉鍛樼粺璁� 鍒楄〃 List<Byte> queueStatusList = new ArrayList<>(); queueStatusList.add(QueueStatusEnum.READY.getStatus()); List<QueueStatisticDO> queueStatisticDOList = queueMapper.queueStatistic(queueStatusList); @@ -147,9 +116,22 @@ }); Integer num = queueMapper.getMaxSeqNum(); - curSeqNum = new AtomicInteger(null == num ? 1 : ++num); + curSeqNum = new AtomicInteger(null == num ? 0 : num); + } - hurryup(); + public void reorderQueue() { + // 鏍规嵁棰勭害鍓嶅悗锛屼粠DB 鑾峰彇 闃熷垪涓� 灏辫瘖鍑嗗涓汉鍛� 鍒楄〃 + List<Byte> queueStatusList = new ArrayList<>(); + queueStatusList.add(QueueStatusEnum.READY.getStatus()); + List<QueueDO> queueDOList = queueMapper.getOrderedQueue(queueStatusList); + if (queueDOList.isEmpty()) + return; + + AtomicInteger seqNum = new AtomicInteger(1); + queueDOList.forEach(item -> {item.setSeqNum(seqNum.getAndIncrement());}); + queueMapper.updateBatch(queueDOList); + + curSeqNum.set( seqNum.get() ); } /** @@ -162,43 +144,87 @@ if (null == bedQueueBO) return; - if (bedQueueBO.queueNum.get() == bedQueueBO.maxQueueNum) + int curQueueNum = bedQueueBO.queueNum.get(); + if (curQueueNum > bedQueueBO.maxQueueNum) + throw new RuntimeException("exceed max queue number!"); + + if (curQueueNum == bedQueueBO.maxQueueNum) return; - QueueDO queue = queueMapper.getFirstInQueueByStatus(QueueStatusEnum.WAITING.getStatus()); - if (null == queue) - return; + Integer updateNum = queueMapper.preemptPatient( + bedQueueBO.getRoomId(), + bedQueueBO.getRoomName(), + bedQueueBO.getBedNo(), + curSeqNum.get() + 1, + QueueStatusEnum.WAITING.getStatus(), + QueueStatusEnum.READY.getStatus()); - queue.setStatus(QueueStatusEnum.READY.getStatus()); //鍊欒瘖鍑嗗涓� - queue.setRoomId(bedQueueBO.getRoomId()); - queue.setRoomName(bedQueueBO.getRoomName()); - queue.setBedNo(bedQueueBO.getBedNo()); - queue.setSeqNum(curSeqNum.get()); - queueMapper.updateById(queue); // queue.getId(); + // 娌℃湁鎶㈠埌鎺掗槦鎮h�� + if (null == updateNum || 0 == updateNum) { + return; + } curSeqNum.getAndIncrement(); - bedQueueBO.queueNum.getAndIncrement(); - BedQueueBO bedQueueBO2 = priorityQueue.poll(); - priorityQueue.offer(bedQueueBO2); + + // 鍙兘宸茬粡銆愬苟鍙戠殑銆戝湪 nextPatient 涓敼鍙樹簡鍊� + bedQueueBO.queueNum.incrementAndGet(); + + // 鍙兘宸茬粡銆愬苟鍙戠殑銆戝湪 nextPatient 涓敼鍙樹簡浼樺厛闃熷垪椤哄簭 + priorityQueue.remove(bedQueueBO); + priorityQueue.offer(bedQueueBO); } } - public void nextPatient(Long roomId, String bedNo) { + /** + * 棰勭害纭鍚庣殑鎺掗槦 + * @param queueSaveReqVO + */ + @Override + public void queue(QueueSaveReqVO queueSaveReqVO) { + queueSaveReqVO.setStatus(QueueStatusEnum.WAITING.getStatus()); //鎺掗槦涓� + QueueDO queue = BeanUtils.toBean(queueSaveReqVO, QueueDO.class); + queueMapper.insert(queue); + + singleThreadExecutor.submit( () -> { + hurryup(); + }); + } + + private void nextPatient(Long roomId, String bedNo) { // 浠� DB 鎶� 搴忓彿鏈�灏忕殑 灏辫瘖鍑嗗涓殑浜� 璁剧疆涓哄氨璇婁腑 queueMapper.updateQueueStatus(roomId, bedNo, QueueStatusEnum.READY.getStatus(), QueueStatusEnum.ONSTAGE.getStatus()); // 浼樺厛闃熷垪涓� 璇ュ伐浣� 灏辫瘖鍑嗗涓汉鐨勬暟閲� 鍑忎竴 BedQueueBO bo = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo)); - boolean breturn = priorityQueue.remove(bo); - bo.queueNum.getAndDecrement(); + bo.queueNum.getAndDecrement(); // 鍙兘宸茬粡銆愬苟鍙戠殑銆戝湪 hurry-up 涓敼鍙樹簡鍊� + priorityQueue.remove(bo); priorityQueue.offer(bo); - hurryup(); + singleThreadExecutor.submit( () -> { + hurryup(); + }); + } + + public void finishNextPatient(Long roomId, String bedNo) { + // 浠� DB 鎶� 灏辫瘖涓殑浜� 璁剧疆涓哄氨璇婂畬鎴� + Integer ret = queueMapper.updateQueueStatus(roomId, bedNo, + QueueStatusEnum.ONSTAGE.getStatus(), QueueStatusEnum.FINISH.getStatus()); + + nextPatient(roomId, bedNo); + } + + public void passNextPatient(Long roomId, String bedNo) { + // 浠� DB 鎶� 灏辫瘖涓殑浜� 璁剧疆涓鸿繃鍙� + Integer ret = queueMapper.updateQueueStatus(roomId, bedNo, + QueueStatusEnum.ONSTAGE.getStatus(), QueueStatusEnum.PASSED.getStatus()); + + nextPatient(roomId, bedNo); } public List<QueueDO> getDoctorQueueByStatus(Long roomId, String bedNo, List<Byte> statusList) { List<QueueDO> queueDOList = queueMapper.getDoctorQueueByStatus(roomId, bedNo, statusList); return queueDOList; } + } -- Gitblit v1.9.3