package cn.lihu.jh.module.ecg.service.queue;
|
|
import cn.lihu.jh.framework.common.exception.ErrorCode;
|
import cn.lihu.jh.framework.common.exception.enums.GlobalErrorCodeConstants;
|
import cn.lihu.jh.framework.common.util.object.BeanUtils;
|
import cn.lihu.jh.module.ecg.controller.admin.room.vo.MonitorInfoVO;
|
import cn.lihu.jh.module.ecg.dal.dataobject.queue.QueueDO;
|
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.call.CallMapper;
|
import cn.lihu.jh.module.ecg.dal.mysql.devrent.DevRentMapper;
|
import cn.lihu.jh.module.ecg.dal.mysql.queue.queueMapper;
|
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 cn.lihu.jh.module.system.api.oauth2.OAuth2TokenApi;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Component;
|
import org.springframework.validation.annotation.Validated;
|
|
import javax.annotation.Resource;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Optional;
|
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.framework.common.pojo.CommonResult.error;
|
import static cn.lihu.jh.module.ecg.enums.ErrorCodeConstants.*;
|
|
/**
|
* 排队 Service 数据库事务相关的方法
|
*
|
* @author 芋道源码
|
*/
|
@Component
|
@Validated
|
@Slf4j
|
public class QueueServiceTxFunctions {
|
|
@Resource
|
private OAuth2TokenApi oAuth2TokenApi;
|
|
@Resource
|
private queueMapper queueMapper;
|
|
@Resource
|
private RoomMapper roomMapper;
|
|
@Resource
|
private CallMapper callMapper;
|
|
@Resource
|
private DevRentMapper devRentMapper;
|
|
AtomicInteger openingFlag = new AtomicInteger(0);
|
AtomicInteger curSeqNum = new AtomicInteger(0);
|
|
PriorityBlockingQueue<BedQueueBO> priorityQueue = new PriorityBlockingQueue<>();
|
ConcurrentHashMap<String, BedQueueBO > mapBedVsQueue = new ConcurrentHashMap<>();
|
|
Map<Integer, Integer> mapCheckTypeVsReadyMax = null;
|
|
/**
|
* 已关闭 或者 关闭中，可以开通工位
|
* @param roomId
|
* @param bedNo
|
* @return
|
*/
|
public ErrorCode bedOpen(Long roomId, String roomName, String bedNo) {
|
BedQueueBO bedQueueBO2 = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo));
|
if (null != bedQueueBO2) {
|
log.error("bedOpen mapBedVsQueue has existed. " + roomId + " " + bedNo);
|
return QUEUE_BED_EXIST;
|
}
|
|
// DB update
|
List statusList = new ArrayList<BedStatusEnum>();
|
statusList.add(BedStatusEnum.CLOSED);
|
Integer updateNum = roomMapper.setBedOpeningOpening(roomId, bedNo, BedStatusEnum.OPENING, statusList);
|
if ( null==updateNum || 0 == updateNum ) {
|
log.error("bedOpen DB invalid status. " + roomId + " " + bedNo);
|
return ROOM_INVALID_STATUS;
|
}
|
|
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.setRoomName(roomName);
|
bedQueueBO.setBedNo(bedNo);
|
bedQueueBO.setMaxQueueNum(getBedReadyMax(roomId, bedNo));
|
bedQueueBO.setQueueNum(new AtomicInteger(queueDOList.size()));
|
bedQueueBO.setStatus(BedStatusEnum.OPENING.getStatus());
|
mapBedVsQueue.put(String.format("%09d%s", roomId, bedNo), bedQueueBO);
|
|
return GlobalErrorCodeConstants.SUCCESS;
|
}
|
|
public ErrorCode bedClose(Long roomId, String bedNo) {
|
BedQueueBO bedQueueBO = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo));
|
if (null == bedQueueBO) {
|
log.error("bedClose mapBedVsQueue DONOT existed. " + roomId + " " + bedNo);
|
return QUEUE_BED_NOT_EXIST;
|
}
|
|
// 除了判断 准备候诊中 的人数，还需要 判断 过号的人数
|
//if (bedQueueBO.getQueueNum().get() >0)
|
// return QUEUE_HAVE_PATIENT;
|
List<Byte> queueStatusList = new ArrayList<>();
|
queueStatusList.add(QueueStatusEnum.READY.getStatus());
|
queueStatusList.add(QueueStatusEnum.PASSED.getStatus());
|
queueStatusList.add(QueueStatusEnum.RECALLED.getStatus());
|
List<QueueDO> queueDOList = queueMapper.getDoctorQueueByStatus(roomId, bedNo, queueStatusList);
|
if (queueDOList.size() > 0)
|
return QUEUE_HAVE_PATIENT;
|
|
// DB update
|
List statusList = new ArrayList<BedStatusEnum>();
|
statusList.add(BedStatusEnum.OPENING);
|
statusList.add(BedStatusEnum.DOCTOR_ON);
|
statusList.add(BedStatusEnum.PAUSE);
|
Integer updateNum = roomMapper.setBedOpeningClosed(roomId, bedNo,
|
BedStatusEnum.CLOSED, statusList);
|
if ( null==updateNum || 0 == updateNum ) {
|
log.error("bedClose DB invalid status. " + roomId + " " + bedNo);
|
return ROOM_INVALID_STATUS;
|
}
|
|
mapBedVsQueue.remove(String.format("%09d%s", roomId, bedNo));
|
priorityQueue.remove(bedQueueBO);
|
return GlobalErrorCodeConstants.SUCCESS;
|
}
|
|
public ErrorCode bedDoctorPause(Long roomId, String bedNo, Long docId, String docName) {
|
BedQueueBO bedQueueBO = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo));
|
if (null == bedQueueBO) {
|
log.error("bedDoctorPause mapBedVsQueue DONOT existed. " + roomId + " " + bedNo);
|
return QUEUE_BED_NOT_EXIST;
|
}
|
|
// DB update
|
List statusList = new ArrayList<BedStatusEnum>();
|
statusList.add(BedStatusEnum.DOCTOR_ON);
|
Integer updateNum = roomMapper.setBedDoctorStatus(roomId, bedNo, docId, BedStatusEnum.PAUSE, statusList);
|
if ( null==updateNum || 0 == updateNum ) {
|
log.error("bedDoctorPause DB invalid status. " + roomId + " " + bedNo);
|
return ROOM_INVALID_STATUS;
|
}
|
|
bedQueueBO.setStatus(BedStatusEnum.PAUSE.getStatus());
|
priorityQueue.remove(bedQueueBO);
|
return GlobalErrorCodeConstants.SUCCESS;
|
}
|
|
public ErrorCode bedDoctorResume(Long roomId, String bedNo, Long docId, String docName) {
|
BedQueueBO bedQueueBO = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo));
|
if (null == bedQueueBO || !bedQueueBO.getStatus().equals(BedStatusEnum.PAUSE.getStatus())) {
|
log.error("bedDoctorResume mapBedVsQueue DONOT existed OR NOT Paused. " + roomId + " " + bedNo);
|
return QUEUE_BED_NOT_EXIST;
|
}
|
|
// DB update
|
List statusList = new ArrayList<BedStatusEnum>();
|
statusList.add(BedStatusEnum.PAUSE);
|
Integer updateNum = roomMapper.setBedDoctorStatus(roomId, bedNo, docId,
|
BedStatusEnum.DOCTOR_ON, statusList);
|
if ( null==updateNum || 0 == updateNum ) {
|
log.error("bedDoctorResume DB invalid status. " + roomId + " " + bedNo);
|
return ROOM_INVALID_STATUS;
|
}
|
|
bedQueueBO.setStatus(BedStatusEnum.DOCTOR_ON.getStatus());
|
if (!priorityQueue.contains(bedQueueBO)) {
|
priorityQueue.offer(bedQueueBO);
|
} else {
|
log.error("bedDoctorResume priorityQueue still exist!. " + roomId + " " + bedNo);
|
return ECG_INNER_ERROR;
|
}
|
|
return GlobalErrorCodeConstants.SUCCESS;
|
}
|
|
public ErrorCode bedDoctorOn(Long roomId, String bedNo, Long docId, String docName) {
|
BedQueueBO bedQueueBO = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo));
|
if (null == bedQueueBO) {
|
log.error("bedDoctorOn mapBedVsQueue DONOT existed. " + roomId + " " + bedNo);
|
return QUEUE_BED_NOT_EXIST;
|
}
|
|
// 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 ) {
|
log.error("bedDoctorOn DB invalid status. " + roomId + " " + bedNo);
|
return ROOM_INVALID_STATUS;
|
}
|
|
bedQueueBO.setStatus(BedStatusEnum.DOCTOR_ON.getStatus());
|
if (!priorityQueue.contains(bedQueueBO)) {
|
priorityQueue.offer(bedQueueBO);
|
} else {
|
log.error("bedDoctorOn priorityQueue still exist!. " + roomId + " " + bedNo);
|
return ECG_INNER_ERROR;
|
}
|
|
return GlobalErrorCodeConstants.SUCCESS;
|
}
|
|
public ErrorCode bedDoctorOff(Long roomId, String bedNo, Long docId, String docName) {
|
BedQueueBO bedQueueBO = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo));
|
if (null == bedQueueBO) {
|
log.error("bedDoctorOff mapBedVsQueue DONOT existed. " + roomId + " " + bedNo);
|
return QUEUE_BED_NOT_EXIST;
|
}
|
|
// DB update
|
List statusList = new ArrayList<BedStatusEnum>();
|
statusList.add(BedStatusEnum.DOCTOR_ON);
|
statusList.add(BedStatusEnum.PAUSE);
|
Integer updateNum = roomMapper.setBedDoctorOff(roomId, bedNo, docId, BedStatusEnum.OPENING, statusList);
|
if ( null==updateNum || 0 == updateNum ) {
|
log.error("bedDoctorOff DB invalid status. " + roomId + " " + bedNo);
|
return ROOM_INVALID_STATUS;
|
}
|
|
priorityQueue.remove(bedQueueBO);
|
|
bedQueueBO.setStatus(BedStatusEnum.OPENING.getStatus());
|
return GlobalErrorCodeConstants.SUCCESS;
|
}
|
|
public ErrorCode nextPatient(Long roomId, String bedNo) {
|
// 从 DB 把 序号最小的 就诊准备中的人 设置为就诊中
|
Integer updateNum = queueMapper.updateQueueStatus(roomId, bedNo,
|
QueueStatusEnum.READY.getStatus(), QueueStatusEnum.ONSTAGE.getStatus());
|
// 该工位 没有 就诊准备中 人员
|
if (null == updateNum || 0 == updateNum) {
|
return QUEUE_NOT_READY_PATIENT;
|
}
|
|
// 优先队列中 该工位 就诊准备中人的数量 减一
|
BedQueueBO bo = mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo));
|
bo.queueNum.getAndDecrement(); // 可能已经【并发的】在 hurry-up 中改变了值
|
priorityQueue.remove(bo);
|
priorityQueue.offer(bo);
|
|
return GlobalErrorCodeConstants.SUCCESS;
|
}
|
|
/**
|
* 1. 每天开诊前 从DB同步工位的患者队列数据到 工位优先队列
|
* 2. 服务运维重启时
|
*/
|
public ErrorCode bedReload() {
|
priorityQueue.clear();
|
mapBedVsQueue.clear();
|
|
// 清除非当天的排队人员
|
queueMapper.clearQueue();
|
// 清除非当天的语音叫号记录
|
callMapper.clearCall();
|
|
// 从DB 获取 工位列表
|
List<BedStatusEnum> bedStatusEnumList = new ArrayList<BedStatusEnum>();
|
bedStatusEnumList.add(BedStatusEnum.OPENING);
|
bedStatusEnumList.add(BedStatusEnum.DOCTOR_ON);
|
bedStatusEnumList.add(BedStatusEnum.PAUSE);
|
List<RoomDO> roomDOList = roomMapper.simpleRoomList(bedStatusEnumList);
|
List<BedQueueBO> bedQueueBOList = roomDOList.stream().map(item -> BeanUtils.toBean(item, BedQueueBO.class)).toList();
|
|
// 从DB 获取 队列中 就诊准备中人员统计 列表
|
List<Byte> queueStatusList = new ArrayList<>();
|
queueStatusList.add(QueueStatusEnum.READY.getStatus());
|
List<QueueStatisticDO> queueStatisticDOList = queueMapper.queueStatistic(queueStatusList);
|
|
bedQueueBOList.forEach(item -> {
|
item.maxQueueNum = getBedReadyMax(item.roomId, item.bedNo);
|
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 ( item.maxQueueNum < queueNum )
|
throw new RuntimeException("init: exceed max queue number!");
|
|
item.queueNum.set( queueNum );
|
mapBedVsQueue.put(String.format("%09d%s", item.roomId, item.bedNo), item);
|
if (BedStatusEnum.DOCTOR_ON.getStatus() == item.getStatus()) {
|
priorityQueue.offer(item);
|
}
|
});
|
|
Integer num = queueMapper.getMaxSeqNum();
|
curSeqNum = new AtomicInteger(null == num ? 0 : num);
|
|
return GlobalErrorCodeConstants.SUCCESS;
|
}
|
|
public void resetRoom(Boolean needCloseBed) {
|
// 踢出在座的医生
|
List<BedStatusEnum> bedStatusEnumList = new ArrayList<BedStatusEnum>();
|
bedStatusEnumList.add(BedStatusEnum.DOCTOR_ON);
|
bedStatusEnumList.add(BedStatusEnum.PAUSE);
|
List<RoomDO> roomDOList = roomMapper.simpleRoomList(bedStatusEnumList);
|
List<Long> userIdList = roomDOList.stream().map(roomDO -> roomDO.getDocId()).toList();
|
oAuth2TokenApi.tick(userIdList);
|
|
// 关闭所有工位
|
if (needCloseBed) {
|
Integer ret = roomMapper.resetRoom(BedStatusEnum.CLOSED.getStatus());
|
}
|
}
|
|
/**
|
* 把医生候诊的队列塞满
|
*/
|
public void hurryup() {
|
if (0 == openingFlag.get())
|
return;
|
|
// 处理 过号-回来 的人
|
for (BedQueueBO bedQueueBO : mapBedVsQueue.values()) {
|
while (bedQueueBO.queueNum.get() < bedQueueBO.maxQueueNum) {
|
// 查看 当前工位 是否有过号-回来的患者
|
Integer updateNum = queueMapper.queueRecalledPatient(
|
bedQueueBO.getRoomId(),
|
bedQueueBO.getRoomName(),
|
bedQueueBO.getBedNo(),
|
curSeqNum.get() + 1,
|
QueueStatusEnum.RECALLED.getStatus(),
|
QueueStatusEnum.READY.getStatus());
|
if (null == updateNum || 0 == updateNum)
|
break;
|
|
curSeqNum.getAndIncrement();
|
|
// 可能已经【并发的】在 nextPatient 中改变了值
|
bedQueueBO.queueNum.incrementAndGet();
|
|
// 可能已经【并发的】在 nextPatient 中改变了优先队列顺序
|
priorityQueue.remove(bedQueueBO);
|
priorityQueue.offer(bedQueueBO);
|
}
|
}
|
|
// 处理 排队中 患者
|
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());
|
|
// 没有抢到排队患者
|
if (null == updateNum || 0 == updateNum) {
|
return;
|
}
|
|
curSeqNum.getAndIncrement();
|
|
// 可能已经【并发的】在 nextPatient 中改变了值
|
bedQueueBO.queueNum.incrementAndGet();
|
|
// 可能已经【并发的】在 nextPatient 中改变了优先队列顺序
|
priorityQueue.remove(bedQueueBO);
|
priorityQueue.offer(bedQueueBO);
|
}
|
}
|
|
public Integer getOpeningFlag() {
|
return openingFlag.get();
|
}
|
|
public void setOpeningFlag(Integer flag) {
|
openingFlag.set(flag);
|
}
|
|
public void setCheckTypeReadyMax(Map<Integer, Integer> max) {
|
mapCheckTypeVsReadyMax = max;
|
}
|
|
public BedQueueBO getBedQueueBO(Long roomId, String bedNo) {
|
return mapBedVsQueue.get(String.format("%09d%s", roomId, bedNo));
|
}
|
|
public MonitorInfoVO getMonitorInfo() {
|
MonitorInfoVO monitorInfoVO = new MonitorInfoVO();
|
monitorInfoVO.setOpeningFlag( openingFlag.get() );
|
monitorInfoVO.setQueueNum( mapBedVsQueue.size() );
|
monitorInfoVO.setActiveQueueNum( priorityQueue.size() );
|
return monitorInfoVO;
|
}
|
|
public void monitorInfo() {
|
log.info("map " + mapBedVsQueue.size() + " priority " + priorityQueue.size()
|
+ " opening " + openingFlag.get());
|
}
|
|
private Integer getBedReadyMax(Long roomId, String bedNo) {
|
RoomDO roomDO = roomMapper.getRoom(roomId, bedNo);
|
Integer[] checkTypes = roomDO.getCheckTypes();
|
Integer ret = 0;
|
for (int i=0; i<checkTypes.length; i++) {
|
Integer tmp = mapCheckTypeVsReadyMax.get(checkTypes[i]);
|
ret = ret < tmp ? tmp : ret; //取大值
|
}
|
return ret;
|
}
|
}
|
