电磁锁手动操控提交

zhangnaisong

2024-07-11 cbbea55a92b178b75f58e7c468f7e8d8ae417c9d

 ard-work/src/main/java/com/ruoyi/utils/websocket/service/ChatServerEndpoint.java

@@ -1,8 +1,8 @@
package com.ruoyi.utils.websocket.service;

import com.alibaba.fastjson2.JSONObject;
import com.ruoyi.app.position.domain.ArdAppPosition;
import com.ruoyi.app.position.service.impl.AppPositionPush;
import com.ruoyi.app.position.service.impl.AppPositionPushService;
import com.ruoyi.call.dto.CallMessage;
import com.ruoyi.utils.websocket.util.WebSocketUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
@@ -11,12 +11,15 @@
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
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
@@ -26,58 +29,51 @@
@Slf4j(topic = "websocket")
@ServerEndpoint("/websocket/{userId}")
public class ChatServerEndpoint {

    @OnOpen
    public void openSession(@PathParam("userId") String userId, Session session) {
        WebSocketUtils.ONLINE_USER_SESSIONS.put(userId, session);
        String message = "用户[" + userId + "] 成功连接！";
        log.info("用户登录：" + message);
        WebSocketUtils.sendMessage(session, message);
        //每个用户连接时，启动初始所有app用户位置发送
        Map<String, ArdAppPosition> appPositionMap = AppPositionPush.getAppPositionList();
        AppPositionPush.defaultDataMap.put(session,appPositionMap);
        AppPositionPush.initPushTask(session,3000);
    }

    @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) {
            Integer frequency = (Integer)map.get("frequency");
            String uId = (String) map.get("userId");
            Map allMap = AppPositionPush.defaultDataMap.get(session);//获取当前session的默认发送的数据map
            if(frequency>0)
            {
                //接收到频率大于0，属于申请引导，将app从默认map中移除
                allMap.remove(uId);
                AppPositionPush.defaultDataMap.putAll(allMap);
//        Session session = WebSocketUtils.ONLINE_USER_SESSIONS.get(userId);
//        WebSocketUtils.sendMessage(session, message);
        Map<String, Object> messageMap = JSONObject.parseObject(message, Map.class);
        if (((String) 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)) {
                String regex = "^" + Pattern.quote(callMessage.getTargetId()) + "_\\d+$";
                Pattern pattern = Pattern.compile(regex);
                ONLINE_USER_SESSIONS.entrySet().stream()
                        .filter(entry -> {
                            Matcher matcher = pattern.matcher(entry.getKey());
                            return matcher.matches();
                        })
                        .map(Map.Entry::getValue)
                        .forEach(session -> WebSocketUtils.sendMessage(session, message));
            }
           else {
                //接收到频率等于0，属于释放引导，将app恢复到初始map
                Map<String, ArdAppPosition> appPositionList = AppPositionPush.getAppPositionList();
                allMap.put(uId,appPositionList.get(uId));
                AppPositionPush.defaultDataMap.putAll(allMap);
            }
            AppPositionPush.startLocationPushTask(uId, session, frequency);

        }
        // 根据用户新的频率重新调整定时任务
        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);
        //并且通知其他人当前用户已经离开聊天室了
        String message = "用户[" + userId + "] 断开连接！";
        //sendMessage(session,message);
        log.info("消息：" + message);
        try {
            session.close();
            AppPositionPush.stopLocationPushTask(session);
            AppPositionPushService.stopHandler(userId);
        } catch (IOException e) {
            log.error("onClose error", e);
        }