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