分支自 ard/ruoyi-api
blame | 历史 | 补丁 | 提交 | 提交对比 | 忽略空白
zhangnaisong
2024-08-01 93cecfaf0fae080ebd0212adea714726290d0d4a 
 ard-work/src/main/java/com/ruoyi/utils/websocket/service/ChatServerEndpoint.java


@@ -1,30 +1,29 @@


package com.ruoyi.utils.websocket.service;




import com.alibaba.fastjson2.JSONObject;


import com.ruoyi.app.position.domain.ArdAppPosition;


import com.ruoyi.app.position.service.IArdAppPositionService;


import com.ruoyi.app.position.service.impl.AppPositionPush;


import com.ruoyi.app.position.service.impl.AppPositionPushService;


import com.ruoyi.app.position.service.impl.ArdAppPositionServiceImpl;


import com.ruoyi.common.core.domain.entity.SysUser;


import com.ruoyi.common.utils.spring.SpringUtils;


import com.ruoyi.system.service.ISysUserService;


import com.ruoyi.call.dto.CallMessage;


import com.ruoyi.common.exception.ServiceException;


import com.ruoyi.utils.websocket.util.WebSocketUtils;


import lombok.extern.slf4j.Slf4j;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.stereotype.Component;




import javax.annotation.Resource;


import javax.websocket.*;


import javax.websocket.server.PathParam;


import javax.websocket.server.ServerEndpoint;


import java.io.IOException;


import java.util.*;


import java.util.stream.Collectors;


import java.util.HashSet;


import java.util.List;


import java.util.Map;


import java.util.Set;


import java.util.regex.Matcher;


import java.util.regex.Pattern;




import static com.ruoyi.utils.websocket.util.WebSocketUtils.ONLINE_USER_SESSIONS;




/**


 * @ClassName ChatServerEndpoint


 * @Description:


 * @Description: websocket操作类


 * @Author 刘苏义


 * @Date 2023/1/27 7:42


 * @Version 1.0


@@ -40,114 +39,67 @@


        String message = "用户[" + userId + "] 成功连接！";


        log.info("用户登录：" + message);


        WebSocketUtils.sendMessage(session, message);


        //region 旧方法


//        //每个用户连接时，启动初始所有app用户位置发送


//        AppPositionPush.initDataMap.put(userId,AppPositionPush.getAppPositionList());


//        AppPositionPush.initPushTask(userId,session,3000);


        //endregion


    }




    @OnMessage


    public void onMessage(@PathParam("userId") String userId, String message) {


        log.info("收到消息：" + message);


        Session session = WebSocketUtils.ONLINE_USER_SESSIONS.get(userId);


        WebSocketUtils.sendMessage(session, message);


//        sendMessageAll("用户[" + userid + "] : " + message);


        // 根据用户新的频率重新调整定时任务


        Map<String, Object> map = JSONObject.parseObject(message, Map.class);


        if (map.size() > 0) {


            Boolean enabled = (Boolean) map.get("enabled");


            if (enabled) {


                //启动全局推送


                List<ArdAppPosition> newList = new ArrayList<>(AppPositionPushService.initDataList);


                AppPositionPushService.dataMap.put(userId, newList);


                if (!AppPositionPushService.userTimerMap.containsKey(userId)) {


                    Timer timer = AppPositionPushService.startTimerTask(userId, 3000);


                    AppPositionPushService.userTimerMap.put(userId, timer);//将定时器关联用户


                }


            } else {


                //停止全局推送


                if (AppPositionPushService.userTimerMap.containsKey(userId)) {


                    Timer timer = AppPositionPushService.userTimerMap.get(userId);


                    AppPositionPushService.stopTimerTask(timer);


                    AppPositionPushService.userTimerMap.remove(userId);


                }


                if (AppPositionPushService.tempUserTimerMap.containsKey(userId)) {


                    Timer temptimer = AppPositionPushService.tempUserTimerMap.get(userId);


                    AppPositionPushService.stopTimerTask(temptimer);


                    AppPositionPushService.tempUserTimerMap.remove(userId);


                }


                AppPositionPushService.dataMap.remove(userId);


                AppPositionPushService.tempDataMap.remove(userId);


        log.debug("收到消息：" + message);


        Map<String, Object> messageMap = JSONObject.parseObject(message, Map.class);


        if (messageMap.get("type")!=null &&messageMap.get("type").equals("callMessage")) {


            String msg = (String) messageMap.get("message");


            CallMessage callMessage = JSONObject.parseObject(msg, CallMessage.class);


            //单聊


            if (callMessage != null && callMessage.getType().equals(0)) {


                WebSocketUtils.sendMessagePrefix(callMessage.getTargetId(), message);


            }


            //群聊


            if (callMessage != null && callMessage.getType().equals(1)) {


                String step = callMessage.getStep();


                switch (step) {


                    case "joinRoom":


                        Set<String> userSet = WebSocketUtils.ROOM_USER_SET.getOrDefault(callMessage.getRoomId(), new HashSet<>());


                        userSet.add(callMessage.getUserId());


                        WebSocketUtils.ROOM_USER_SET.put(callMessage.getRoomId(), userSet);


                        WebSocketUtils.sendRoomMessage(callMessage.getRoomId(), message);


                        break;


                    case "inviteRoom":


                        List<String> targetIds = callMessage.getTargetIds();


                        targetIds.stream().forEach(targetId -> {


                            WebSocketUtils.sendMessagePrefix(targetId, message);


                        });


                        break;


                    case "leaveRoom":


                        userSet = WebSocketUtils.ROOM_USER_SET.getOrDefault(callMessage.getRoomId(), new HashSet<>());


                        if (userSet.size() > 0) {


                            userSet.remove(callMessage.getUserId());


                            WebSocketUtils.ROOM_USER_SET.put(callMessage.getRoomId(), userSet);


                            WebSocketUtils.sendRoomMessage(callMessage.getRoomId(), message);


                        }


                        break;


                    default:


                        WebSocketUtils.sendMessagePrefix(callMessage.getTargetId(), message);


                        break;


                }




            Integer frequency = (Integer) map.get("frequency");


            String uId = (String) map.get("userId");


            if (frequency > 0) {


                List<ArdAppPosition> List = AppPositionPushService.initDataList.stream()


                        .filter(obj -> !obj.getUserId().equals(uId))


                        .collect(Collectors.toList());//去除引导后的list


                AppPositionPushService.dataMap.put(userId,List);




                List<ArdAppPosition> tempList = AppPositionPushService.initDataList.stream()


                        .filter(obj -> obj.getUserId().equals(uId))


                        .collect(Collectors.toList());//获取引导的list


                AppPositionPushService.tempDataMap.put(userId, tempList);


                Timer timer = AppPositionPushService.startTempTimerTask(userId, frequency);


                AppPositionPushService.tempUserTimerMap.put(userId, timer);


            } else {


                List<ArdAppPosition> newList = new ArrayList<>(AppPositionPushService.initDataList);


                AppPositionPushService.dataMap.put(userId, newList);


                Timer timer = AppPositionPushService.tempUserTimerMap.get(userId);


                AppPositionPushService.stopTimerTask(timer);


                AppPositionPushService.tempUserTimerMap.remove(userId);


                AppPositionPushService.tempDataMap.remove(userId);


            }


            //region 旧方法


//            Map<String, List<ArdAppPosition>> dataMap = new HashMap<>();


//            List<ArdAppPosition> positionList = AppPositionPush.initDataMap.get(userId);//获取当前用户的默认发送的数据map


//            if(frequency>0)


//            {


//                //接收到频率大于0，属于申请引导，获取除引导外的其他app用户的位置List


//                List<ArdAppPosition> appPositionList = positionList.stream()


//                        .filter(obj -> !obj.getUserId().equals(uId))


//                        .collect(Collectors.toList());


//                dataMap.put(userId,appPositionList);


//            }


//           else {


//                //接收到频率等于0，属于释放引导，获取到全部app用户的位置List


//                dataMap.put(userId,AppPositionPush.getAppPositionList());


//            }


//            AppPositionPush.initDataMap.putAll(dataMap);//更新初始会话默认数据map


//            AppPositionPush.startLocationPushTask(userId,session,uId, frequency);//开启新任务


            //endregion


        }


        // 根据用户新的频率重新调整定时任务


        AppPositionPushService.messageHandler(userId, message);




    }




    @OnClose


    public void onClose(@PathParam("userId") String userId, Session session) {


        //当前的Session 移除


        WebSocketUtils.ONLINE_USER_SESSIONS.remove(userId);


//        //并且通知其他人当前用户已经离开聊天室了


        String message = "用户[" + userId + "] 断开连接！";


//        sendMessage(session,message);


        log.info("消息：" + message);


        //从房间中移除用户


        WebSocketUtils.ROOM_USER_SET.values().stream().forEach(userSet -> userSet.remove(userId));


        try {


            session.close();


            //停止全局推送


            if (AppPositionPushService.userTimerMap.containsKey(userId)) {


                Timer timer = AppPositionPushService.userTimerMap.get(userId);


                AppPositionPushService.stopTimerTask(timer);


                AppPositionPushService.userTimerMap.remove(userId);


            }


            if (AppPositionPushService.tempUserTimerMap.containsKey(userId)) {


                Timer temptimer = AppPositionPushService.tempUserTimerMap.get(userId);


                AppPositionPushService.stopTimerTask(temptimer);


                AppPositionPushService.tempUserTimerMap.remove(userId);


            }


            AppPositionPushService.dataMap.remove(userId);


            AppPositionPushService.tempDataMap.remove(userId);


            // AppPositionPush.stopLocationPushTask(session);//旧方法


            AppPositionPushService.stopHandler(userId);


        } catch (IOException e) {


            log.error("onClose error", e);


        }


@@ -162,5 +114,4 @@


        }


        log.info("Throwable msg " + throwable.getMessage());


    }




}