直播

ruoyi-cai/src/main/java/com/ruoyi/cai/ws/job/CheckTimeOutJob.java

@@ -0,0 +1,75 @@
+package com.ruoyi.cai.ws.job;
+
+import cn.hutool.core.date.DateUtil;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.ruoyi.cai.domain.UserCall;
+import com.ruoyi.cai.service.UserCallService;
+import com.ruoyi.cai.trd.ImDataRes;
+import com.ruoyi.cai.trd.ImMsgGen;
+import com.ruoyi.cai.ws.bean.Room;
+import com.ruoyi.cai.ws.bean.UserData;
+import com.ruoyi.cai.ws.cache.OnlineDataCache;
+import com.ruoyi.cai.ws.cache.RoomCtxCache;
+import com.ruoyi.cai.ws.constant.RoomStatusEnums;
+import com.ruoyi.cai.ws.service.RoomService;
+import com.ruoyi.yunxin.Yunxin;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+
+import java.util.Set;
+import java.util.concurrent.TimeUnit;
+
+@Component
+public class CheckTimeOutJob {
+    @Autowired
+    private OnlineDataCache onlineDataCache;
+    @Autowired
+    private RoomService roomService;
+    @Autowired
+    private UserCallService userCallService;
+    @Autowired
+    private RoomCtxCache roomCtxCache;
+    @Autowired
+    private Yunxin yunxin;
+    /**
+     * 1 分钟执行一次
+     */
+    @Scheduled(fixedDelay = 60,timeUnit = TimeUnit.SECONDS)
+    public void run(){
+        Set<String> roomIdStr = onlineDataCache.getAll();
+        for (String roomIdS : roomIdStr) {
+            Long roomId = Long.valueOf(roomIdS);
+            this.deal(roomId);
+        }
+    }
+
+    public void deal(Long roomId){
+        Room room = roomService.load(roomId);
+        if(!RoomStatusEnums.STATUS_CALLER_CONNECT.getCode().equals(room.getStatus())
+                && !RoomStatusEnums.STATUS_RECEIVER_CONNECT.getCode().equals(room.getStatus())){
+            return;
+        }
+        UserData callUserData = room.getCallUserData();
+        UserData receiverUserData = room.getReceiverUserData();
+        // 检测是不是3分钟没有接听
+        Long connectTimeCaller = callUserData.getConnectTime();
+        Long connectTimeReceiver = receiverUserData.getConnectTime();
+        boolean timeOut = false;
+        if(connectTimeCaller != null && DateUtil.currentSeconds() - connectTimeCaller > 180){
+            timeOut = true;
+        }else if(connectTimeReceiver != null && DateUtil.currentSeconds() - connectTimeReceiver > 180){
+            timeOut = true;
+        }
+        if(timeOut){
+            if(roomService.hangUp(roomId)){
+                userCallService.update(Wrappers.lambdaUpdate(UserCall.class)
+                        .eq(UserCall::getId,roomId)
+                        .set(UserCall::getStatus, RoomStatusEnums.STATUS_TIMEOUT_CANCEL.getCode()));
+                roomService.closeAllFd(roomId);
+                ImDataRes imDataRes = ImMsgGen.callNotice(3, callUserData.getId(), receiverUserData.getId(), 0);
+                yunxin.sendTo(receiverUserData.getId(),callUserData.getId(),imDataRes);
+            }
+        }
+    }
+}

ruoyi-cai/src/main/java/com/ruoyi/cai/ws/job/HeartbeatJob.java

@@ -0,0 +1,99 @@
+package com.ruoyi.cai.ws.job;
+
+import cn.hutool.core.date.DateUtil;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.ruoyi.cai.domain.UserCall;
+import com.ruoyi.cai.service.UserCallService;
+import com.ruoyi.cai.ws.bean.Room;
+import com.ruoyi.cai.ws.bean.UserData;
+import com.ruoyi.cai.ws.cache.OnlineDataCache;
+import com.ruoyi.cai.ws.cache.RoomCtxCache;
+import com.ruoyi.cai.ws.constant.HangUpEnums;
+import com.ruoyi.cai.ws.constant.RoomStatusEnums;
+import com.ruoyi.cai.ws.dto.WsR;
+import com.ruoyi.cai.ws.dto.WsRMsgGen;
+import com.ruoyi.cai.ws.service.RoomService;
+import com.ruoyi.cai.ws.service.SettleService;
+import com.ruoyi.cai.ws.util.RoomWebSocketUtil;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+import java.util.Set;
+import java.util.concurrent.TimeUnit;
+
+@Component
+@Slf4j
+public class HeartbeatJob {
+
+    @Autowired
+    private OnlineDataCache onlineDataCache;
+    @Autowired
+    private RoomService roomService;
+    @Autowired
+    private UserCallService userCallService;
+    @Autowired
+    private SettleService settleService;
+    @Autowired
+    private RoomCtxCache roomCtxCache;
+
+
+    /**
+     * 30秒执行一次
+     */
+    @Scheduled(fixedDelay = 30,timeUnit = TimeUnit.SECONDS)
+    public void run(){
+        Set<String> all = onlineDataCache.getAll();
+        for (String roomIdS : all) {
+            try {
+                Long roomId = Long.valueOf(roomIdS);
+                Room room = roomService.load(roomId);
+                if(!room.isCanCall()){
+                    return;
+                }
+                UserData callUserData = room.getCallUserData();
+                UserData receiverUserData = room.getReceiverUserData();
+                boolean timeOut = false;
+                WsR hangup = null;
+                if(isHeartTimeout(callUserData)){
+                    timeOut = true;
+                    hangup = WsRMsgGen.hangup("呼叫方连接中断", roomId, HangUpEnums.FROM.getCode());
+                }else if(isHeartTimeout(receiverUserData)){
+                    timeOut = true;
+                    hangup = WsRMsgGen.hangup("接听方连接中断", roomId, HangUpEnums.TO.getCode());
+                }
+                if(timeOut){
+                    if(roomService.hangUp(roomId)){
+                        userCallService.update(Wrappers.lambdaUpdate(UserCall.class)
+                                .eq(UserCall::getId,roomId)
+                                .set(UserCall::getStatus, RoomStatusEnums.STATUS_TIMEOUT_CANCEL.getCode()));
+                        settleService.processOn(room);
+                    }
+                    List<String> keys = roomCtxCache.getSessionKeysByRoomId(roomId);
+                    RoomWebSocketUtil.sendSendMessage(keys,  hangup);
+                    roomService.closeAllFd(roomId);
+                }
+            }catch (Exception e){
+                log.error("定时心跳检测失败！",e);
+            }
+
+        }
+    }
+
+    /**
+     * 是否心跳超时
+     * @param userData
+     * @return
+     */
+    private boolean isHeartTimeout(UserData userData){
+        if(userData.getConnectTime() == null){
+            return false;
+        }
+        if(DateUtil.currentSeconds() - userData.getHeartTime() < 120){
+            return false;
+        }
+        return true;
+    }
+}

ruoyi-cai/src/main/java/com/ruoyi/cai/ws/job/SettleJob.java

@@ -0,0 +1,44 @@
+package com.ruoyi.cai.ws.job;
+
+import com.ruoyi.cai.ws.bean.Room;
+import com.ruoyi.cai.ws.cache.OnlineDataCache;
+import com.ruoyi.cai.ws.service.RoomService;
+import com.ruoyi.cai.ws.service.SettleService;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
+
+import java.util.Set;
+import java.util.concurrent.TimeUnit;
+
+@Component
+@Slf4j
+public class SettleJob {
+    @Autowired
+    private OnlineDataCache onlineDataCache;
+    @Autowired
+    private SettleService settleService;
+    @Autowired
+    private RoomService roomService;
+
+    /**
+     * 每 1 分钟执行一次
+     */
+    @Scheduled(fixedDelay = 60,timeUnit = TimeUnit.SECONDS)
+    public void run(){
+        Set<String> all = onlineDataCache.getAll();
+        for (String roomIdS : all) {
+            try {
+                Long roomId = Long.valueOf(roomIdS);
+                Room room = roomService.load(roomId);
+                if(room == null){
+                    return;
+                }
+                settleService.processOn(room);
+            }catch (Exception e){
+                log.info("定时任务结算失败！",e);
+            }
+        }
+    }
+}