From 9b18e467d407c66ecc3c46a197aca45dcc3f8056 Mon Sep 17 00:00:00 2001 From: eight <641137800@qq.com> Date: 星期二, 27 八月 2024 09:11:02 +0800 Subject: [PATCH] update --- jh-module-ecg/jh-module-ecg-biz/src/main/java/cn/lihu/jh/module/ecg/service/queue/queueServiceImpl.java | 434 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 433 insertions(+), 1 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 98a3b40..fd31e8c 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 @@ -1,5 +1,11 @@ package cn.lihu.jh.module.ecg.service.queue; +import cn.lihu.jh.module.ecg.dal.dataobject.queue.BedQueueStatisticDO; +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.mysql.room.RoomMapper; +import cn.lihu.jh.module.ecg.enums.BedStatusEnum; +import cn.lihu.jh.module.ecg.enums.QueueStatusEnum; import org.springframework.stereotype.Service; import org.springframework.validation.annotation.Validated; @@ -11,6 +17,10 @@ import cn.lihu.jh.module.ecg.dal.mysql.queue.queueMapper; import javax.annotation.Resource; + +import java.util.*; +import java.util.concurrent.*; +import java.util.concurrent.atomic.AtomicInteger; import static cn.lihu.jh.framework.common.exception.util.ServiceExceptionUtil.exception; import static cn.lihu.jh.module.ecg.enums.ErrorCodeConstants.*; @@ -26,6 +36,19 @@ @Resource private queueMapper queueMapper; + + @Resource + private RoomMapper roomMapper; + + AtomicInteger openingFlag = new AtomicInteger(0); + + AtomicInteger curSeqNum = new AtomicInteger(0); + + PriorityBlockingQueue<BedQueueBO> priorityQueue = new PriorityBlockingQueue<>(); + ConcurrentHashMap<String, BedQueueBO > mapBedVsQueue = new ConcurrentHashMap<>(); + + ExecutorService singleThreadExecutor = Executors.newSingleThreadExecutor(); + @Override public Integer createqueue(QueueSaveReqVO createReqVO) { @@ -53,6 +76,195 @@ queueMapper.deleteById(id); } + @Override + public Integer startBedOpen(Long roomId, String bedNo) { + Future<Integer> future = singleThreadExecutor.submit( new BedOpenCallable(this, roomId, bedNo)); + + try { + Integer ret = future.get(); + return ret; + } catch (InterruptedException e) { + e.printStackTrace(); + } catch (ExecutionException e) { + e.printStackTrace(); + } + + return 100; + } + + @Override + public Integer startBedClose(Long roomId, String bedNo) { + Future<Integer> future = singleThreadExecutor.submit( new BedCloseCallable(this, roomId, bedNo)); + + try { + Integer ret = future.get(); + return ret; + } catch (InterruptedException e) { + e.printStackTrace(); + } catch (ExecutionException e) { + e.printStackTrace(); + } + + return 100; + } + + @Override + public Integer startBedDoctorPause(Long roomId, String bedNo, Long docId, String docName) { + Future<Integer> future = singleThreadExecutor.submit( + new BedDoctorPauseCallable(this, roomId, bedNo, docId, docName) + ); + + try { + Integer ret = future.get(); + return ret; + } catch (InterruptedException e) { + e.printStackTrace(); + } catch (ExecutionException e) { + e.printStackTrace(); + } + + return 100; + } + + @Override + public Integer startBedDoctorOn(Long roomId, String bedNo, Long docId, String docName) { + Future<Integer> future = singleThreadExecutor.submit( + new BedDoctorOnCallable(this, roomId, bedNo, docId, docName) + ); + + try { + Integer ret = future.get(); + return ret; + } catch (InterruptedException e) { + e.printStackTrace(); + } catch (ExecutionException e) { + e.printStackTrace(); + } + + return 100; + } + + @Override + public Integer startBedDoctorOff(Long roomId, String bedNo, Long docId, String docName) { + Future<Integer> future = singleThreadExecutor.submit( + new BedDoctorOffCallable(this, roomId, bedNo, docId, docName) + ); + + try { + Integer ret = future.get(); + return ret; + } catch (InterruptedException e) { + e.printStackTrace(); + } catch (ExecutionException e) { + e.printStackTrace(); + } + + return 100; + } + + + /** + * 宸插叧闂� 鎴栬�� 鍏抽棴涓紝鍙互寮�閫氬伐浣� + * @param roomId + * @param bedNo + * @return + */ + @Override + public Integer bedOpen(Long roomId, String bedNo) { + // DB update + List statusList = new ArrayList<BedStatusEnum>(); + statusList.add(BedStatusEnum.CLOSED); + statusList.add(BedStatusEnum.CLOSING); + Integer updateNum = roomMapper.setBedStatus(roomId, bedNo, + BedStatusEnum.OPENING, statusList); + if ( null==updateNum || 0 == updateNum ) + return 310; + + List<Byte> queueStatusList = new ArrayList<>(); + queueStatusList.add(QueueStatusEnum.READY.getStatus()); + List<QueueDO> queueDOList = queueMapper.getDoctorQueueByStatus(roomId, bedNo, queueStatusList); + + // 鏂板宸ヤ綅 浼樺厛闃熷垪 + BedQueueBO bedQueueBO = new BedQueueBO(); + bedQueueBO.setRoomId(roomId); + bedQueueBO.setBedNo(bedNo); + bedQueueBO.setMaxQueueNum(MAX_QUEUE_NUM); + bedQueueBO.setQueueNum(new AtomicInteger(queueDOList.size())); + priorityQueue.offer(bedQueueBO); + mapBedVsQueue.put(String.format("%09d%s", roomId, bedNo), bedQueueBO); + return 0; + } + + @Override + public Integer bedClose(Long roomId, String bedNo) { + BedQueueBO bedQueueBO = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo)); + if (null == bedQueueBO) + return 320; + + // DB update + List statusList = new ArrayList<BedStatusEnum>(); + statusList.add(BedStatusEnum.OPENING); + statusList.add(BedStatusEnum.DOCTOR_ON); + statusList.add(BedStatusEnum.PAUSE); + Integer updateNum = roomMapper.setBedStatus(roomId, bedNo, + BedStatusEnum.CLOSING, statusList); + if ( null==updateNum || 0 == updateNum ) + return 321; + + return 0; + } + + @Override + public Integer bedDoctorPause(Long roomId, String bedNo, Long docId, String docName) { + BedQueueBO bedQueueBO = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo)); + if (null == bedQueueBO) + return 320; + + // DB update + List statusList = new ArrayList<BedStatusEnum>(); + statusList.add(BedStatusEnum.DOCTOR_ON); + Integer updateNum = roomMapper.setBedDoctorPause(roomId, bedNo, docId, docName, + BedStatusEnum.PAUSE, statusList); + if ( null==updateNum || 0 == updateNum ) + return 331; + + return 0; + } + + @Override + public Integer bedDoctorOn(Long roomId, String bedNo, Long docId, String docName) { + BedQueueBO bedQueueBO = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo)); + if (null == bedQueueBO) + return 320; + + // DB update + List statusList = new ArrayList<BedStatusEnum>(); + statusList.add(BedStatusEnum.OPENING); + Integer updateNum = roomMapper.setBedDoctorOn(roomId, bedNo, docId, docName, + BedStatusEnum.DOCTOR_ON, statusList); + if ( null==updateNum || 0 == updateNum ) + return 341; + + return 0; + } + + @Override + public Integer bedDoctorOff(Long roomId, String bedNo, Long docId, String docName) { + BedQueueBO bedQueueBO = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo)); + if (null == bedQueueBO) + return 320; + + // DB update + List statusList = new ArrayList<BedStatusEnum>(); + statusList.add(BedStatusEnum.DOCTOR_ON); + Integer updateNum = roomMapper.setBedDoctorOff(roomId, bedNo, docId, docName, + BedStatusEnum.OPENING, statusList); + if ( null==updateNum || 0 == updateNum ) + return 351; + + return 0; + } + private void validatequeueExists(Integer id) { if (queueMapper.selectById(id) == null) { throw exception(QUEUE_NOT_EXISTS); @@ -69,4 +281,224 @@ return queueMapper.selectPage(pageReqVO); } -} \ No newline at end of file + /** + * 锛侊紒寮�璇婃湡闂达紝涓嶈兘鎵ц杩欎釜鏂规硶锛屽惁鍒欎細鏈� P0 闂 + * 1. 姣忓ぉ寮�璇婂墠 浠嶥B鍚屾宸ヤ綅鐨勬偅鑰呴槦鍒楁暟鎹埌 宸ヤ綅浼樺厛闃熷垪 + * 2. 鏈嶅姟杩愮淮閲嶅惎鏃� + */ + public void initBedQueueAndSeqNumFromDB() { + priorityQueue.clear(); + mapBedVsQueue.clear(); + + // 浠嶥B 鑾峰彇 宸ヤ綅鍒楄〃 + List<RoomDO> roomDOList = roomMapper.simpleRoomList(); + List<BedQueueBO> bedQueueBOList = roomDOList.stream().map(item -> BeanUtils.toBean(item, BedQueueBO.class)).toList(); + + // 浠嶥B 鑾峰彇 闃熷垪涓� 灏辫瘖鍑嗗涓汉鍛樼粺璁� 鍒楄〃 + List<Byte> queueStatusList = new ArrayList<>(); + queueStatusList.add(QueueStatusEnum.READY.getStatus()); + List<QueueStatisticDO> queueStatisticDOList = queueMapper.queueStatistic(queueStatusList); + + bedQueueBOList.forEach(item -> { + item.maxQueueNum = MAX_QUEUE_NUM; + Optional<QueueStatisticDO> queueStatisticDOOptional = queueStatisticDOList.stream().filter(it->it.getRoomId()==item.roomId && it.getBedNo().equals(item.getBedNo())).findFirst(); + int queueNum = queueStatisticDOOptional.isPresent() ? queueStatisticDOOptional.get().getTotalInStatus() : 0; + if ( MAX_QUEUE_NUM < queueNum ) + throw new RuntimeException("init: exceed max queue number!"); + + item.queueNum.set( queueNum ); + priorityQueue.offer(item); + mapBedVsQueue.put(String.format("%09d%s", item.roomId, item.bedNo), item); + }); + + Integer num = queueMapper.getMaxSeqNum(); + curSeqNum = new AtomicInteger(null == num ? 0 : num); + } + + /** + * TODO 鏂板紑闃熷垪鏃讹紝闇�瑕佹妸鎺掗槦涓殑浜� 杞埌 灏辫瘖鍑嗗 鐘舵�� + * 绛夊埌鍙栦笅涓�涓� 鎺掗槦涓汉鍛� 鐨勯�昏緫瀹屾垚鍚庯紝鍐嶅洖鏉ヤ笉閿� + */ + public void hurryup() { + if (0 == openingFlag.get()) + return; + + // 澶勭悊 杩囧彿-鍥炴潵 鐨勪汉 + for (BedQueueBO bedQueueBO : mapBedVsQueue.values()) { + while (bedQueueBO.queueNum.get() < bedQueueBO.maxQueueNum) { + // 鏌ョ湅 褰撳墠宸ヤ綅 鏄惁鏈夎繃鍙�-鍥炴潵鐨勬偅鑰� + Integer updateNum = queueMapper.procPassedReturnPatient( + bedQueueBO.getRoomId(), + bedQueueBO.getRoomName(), + bedQueueBO.getBedNo(), + curSeqNum.get() + 1, + QueueStatusEnum.PASSED_RETURN.getStatus(), + QueueStatusEnum.READY.getStatus()); + if (null == updateNum || 0 == updateNum) + break; + + curSeqNum.getAndIncrement(); + + // 鍙兘宸茬粡銆愬苟鍙戠殑銆戝湪 nextPatient 涓敼鍙樹簡鍊� + bedQueueBO.queueNum.incrementAndGet(); + + // 鍙兘宸茬粡銆愬苟鍙戠殑銆戝湪 nextPatient 涓敼鍙樹簡浼樺厛闃熷垪椤哄簭 + priorityQueue.remove(bedQueueBO); + priorityQueue.offer(bedQueueBO); + } + } + + // 澶勭悊 鎺掗槦涓� 鎮h�� + while (true) { + BedQueueBO bedQueueBO = priorityQueue.peek(); + if (null == bedQueueBO) + return; + + int curQueueNum = bedQueueBO.queueNum.get(); + if (curQueueNum > bedQueueBO.maxQueueNum) + throw new RuntimeException("hurryup: exceed max queue number!"); + + if (curQueueNum == bedQueueBO.maxQueueNum) + return; + + // 鏌ョ湅 鏄惁鏈夋帓闃熶腑鐨勬偅鑰� + Integer updateNum = queueMapper.preemptPatient( + bedQueueBO.getRoomId(), + bedQueueBO.getRoomName(), + bedQueueBO.getBedNo(), + curSeqNum.get() + 1, + QueueStatusEnum.WAITING.getStatus(), + QueueStatusEnum.READY.getStatus()); + + // 娌℃湁鎶㈠埌鎺掗槦鎮h�� + if (null == updateNum || 0 == updateNum) { + return; + } + + curSeqNum.getAndIncrement(); + + // 鍙兘宸茬粡銆愬苟鍙戠殑銆戝湪 nextPatient 涓敼鍙樹簡鍊� + bedQueueBO.queueNum.incrementAndGet(); + + // 鍙兘宸茬粡銆愬苟鍙戠殑銆戝湪 nextPatient 涓敼鍙樹簡浼樺厛闃熷垪椤哄簭 + priorityQueue.remove(bedQueueBO); + priorityQueue.offer(bedQueueBO); + } + } + + /** + * 棰勭害纭鍚庣殑鎺掗槦 + * @param queueSaveReqVO + */ + @Override + public void queue(QueueSaveReqVO queueSaveReqVO) { + queueSaveReqVO.setStatus(QueueStatusEnum.WAITING.getStatus()); //鎺掗槦涓� + QueueDO queue = BeanUtils.toBean(queueSaveReqVO, QueueDO.class); + queueMapper.insert(queue); + + if (0 == openingFlag.get()) + return; + + startHurryUp(); + } + + private void nextPatient(Long roomId, String bedNo) { + // 浠� DB 鎶� 搴忓彿鏈�灏忕殑 灏辫瘖鍑嗗涓殑浜� 璁剧疆涓哄氨璇婁腑 + Integer updateNum = queueMapper.updateQueueStatus(roomId, bedNo, + QueueStatusEnum.READY.getStatus(), QueueStatusEnum.ONSTAGE.getStatus()); + + // 璇ュ伐浣� 娌℃湁 灏辫瘖鍑嗗涓� 浜哄憳 + if (null == updateNum || 0 == updateNum) { + return; + } + + // 浼樺厛闃熷垪涓� 璇ュ伐浣� 灏辫瘖鍑嗗涓汉鐨勬暟閲� 鍑忎竴 + BedQueueBO bo = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo)); + bo.queueNum.getAndDecrement(); // 鍙兘宸茬粡銆愬苟鍙戠殑銆戝湪 hurry-up 涓敼鍙樹簡鍊� + priorityQueue.remove(bo); + priorityQueue.offer(bo); + + startHurryUp(); + } + + 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; + } + + public PatientStatisticVO getPatientStatistic(Long roomId, String bedNo) { + PatientStatisticVO patientStatisticVO = new PatientStatisticVO(); + List<BedQueueStatisticDO> bedQueueStatisticDOList = queueMapper.bedQueueStatistic(roomId, bedNo); + bedQueueStatisticDOList.forEach(item -> { + if (QueueStatusEnum.READY.getStatus() == item.getStatus()) { + patientStatisticVO.setReadyNum(item.getTotalInStatus()); + } else if (QueueStatusEnum.FINISH.getStatus() == item.getStatus()) { + patientStatisticVO.setFinishedNum(item.getTotalInStatus()); + } else if (QueueStatusEnum.PASSED.getStatus() == item.getStatus()) { + patientStatisticVO.setPassedNum(item.getTotalInStatus()); + } + }); + + List<Byte> statusList = new ArrayList<>(); + statusList.add(QueueStatusEnum.WAITING.getStatus()); + Integer num = queueMapper.statusStatistic(statusList); + patientStatisticVO.setQueuingNum(num); + + return patientStatisticVO; + } + + public void startBiz() { + if (1 == openingFlag.get()) + return; + + // 娓呴櫎闈炲綋澶╃殑鎺掗槦浜哄憳 + queueMapper.clearQueue(); + + initBedQueueAndSeqNumFromDB(); + + openingFlag.set(1); + hurryup(); + } + + public void closeBiz() { + openingFlag.set(0); + } + + @Override + public Integer recallPatient(Long roomId, String bedNo, String patId) { + Integer updateNum = queueMapper.passedPatientReturn(roomId, bedNo, patId, + QueueStatusEnum.PASSED.getStatus(), QueueStatusEnum.PASSED_RETURN.getStatus()); + startHurryUp(); + return updateNum; + } + + @Override + public Integer patientJump(String patId, Byte jumped) { + Integer updateNum = queueMapper.queueJump(patId, QueueStatusEnum.WAITING.getStatus(), jumped); + startHurryUp(); + return updateNum; + } + + private void startHurryUp() { + singleThreadExecutor.execute( () -> { + hurryup(); + }); + } + +} -- Gitblit v1.9.3