sockeio连接调整,连接与获取菜单数据进行分离

guoshilong

2023-03-20 eb70b82c29aa08cd3e94b1f6c0ced5f1cd7135dd

 src/main/java/org/springblade/common/handler/SocketIOService.java

@@ -1,11 +1,20 @@
package org.springblade.common.handler;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.corundumstudio.socketio.*;
import com.corundumstudio.socketio.annotation.OnConnect;
import com.corundumstudio.socketio.annotation.OnDisconnect;
import com.corundumstudio.socketio.annotation.OnEvent;
import com.corundumstudio.socketio.protocol.Packet;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.StringUtil;
import org.springblade.modules.modules.entity.FunctionEntity;
import org.springblade.modules.modules.entity.ModulesEntity;
import org.springblade.modules.modules.service.IFunctionService;
import org.springblade.modules.modules.service.IModulesService;
import org.springblade.modules.modules.vo.FunctionVO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.core.annotation.Order;
@@ -19,6 +28,8 @@
import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import java.net.ServerSocket;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
@@ -34,7 +45,6 @@

   private final Logger log = LoggerFactory.getLogger(this.getClass());

   // 用来存已连接的客户端
   private static Map<String, SocketIOClient> clientMap = new ConcurrentHashMap<>();

   @Autowired
@@ -42,8 +52,6 @@

   /**
    * Spring IoC容器创建之后，在加载SocketIOConfiguration Bean之后启动
    *
    * @throws Exception
    */
   @PostConstruct
   private void autoStartup() throws Exception {
@@ -52,8 +60,6 @@

   /**
    * Spring IoC容器在销毁SocketIOConfiguration Bean之前关闭,避免重启项目服务端口占用问题
    *
    * @throws Exception
    */
   @PreDestroy
   private void autoStop() throws Exception {
@@ -67,45 +73,76 @@

   /**
    * 连接时
    * @param client
    */
   @OnConnect
   public void onConnect(SocketIOClient client){
      log.info(getIpByClient(client));
      HandshakeData handshakeData = client.getHandshakeData();
      Map<String, List<String>> urlParams = handshakeData.getUrlParams();
   public void onConnect(SocketIOClient client) {
      String connectInfo = client.getHandshakeData().getSingleUrlParam("connectInfo");
      SocketEntity socketEntity = parseMsg(connectInfo);

      if (clientMap.containsKey(socketEntity.getCurrent())){
         client.sendEvent("connectError",R.fail("连接已存在"));
         log.info("设备{}连接失败",socketEntity.getCurrent());
      }else {
         clientMap.put(socketEntity.getCurrent(),client);
         client.sendEvent("connectOk",R.success("连接成功"));
         log.info("设备{}连接成功",socketEntity.getCurrent());
      }
   }

   /**
    * 监听前端订阅相同事件发送过来的信息
    *
    * @param client
    * @param ackRequest
    * @param data
    */
   @OnEvent(value = "msg")
   public void OnEvent(SocketIOClient client, AckRequest ackRequest, String data){
   public void OnEvent(SocketIOClient client, AckRequest ackRequest, String data) {
      log.info("发来消息：" + data);
   }

   /**
    * 监听前端订阅相同事件发送过来的信息
    * @param client
    * @param ackRequest
    * @param data
    * 图册上下页控制
    */
   @OnEvent(value = "msg2")
   public void OnEventMsg2(SocketIOClient client, AckRequest ackRequest, String data){
      log.info("发来消息：" + data);
      client.sendEvent("ClientReceive","copy"+data);
   @OnEvent(value = "changeImgPage")
   public void OnEventImgChange(SocketIOClient client, AckRequest ackRequest, String data) {
      log.info("图册变更：" + data);
      SocketEntity socketEntity = parseMsg(data);
      if (socketEntity.getMsg().equals("previous")) {
         //上一页
         clientMap.get(socketEntity.getTarget()).sendEvent("previousPage", "previous");
      } else {
         //下一页
         clientMap.get(socketEntity.getTarget()).sendEvent("nextPage", "next");
      }
   }

   /**
    * 菜单控制
    */
   @OnEvent(value = "menuChange")
   public void OnEventMenuChange(SocketIOClient client, AckRequest ackRequest, String data) {
      log.info("菜单变更：" + data);
      SocketEntity socketEntity = parseMsg(data);
      log.info("控制的设备是：{}", parseMsg(client.getHandshakeData().getSingleUrlParam("connectInfo")).getTarget());
      clientMap.get(socketEntity.getTarget()).sendEvent("menuChange", socketEntity.getMsg());
   }

   /**
    * 断开连接时
    *
    * @param client
    */
   @OnDisconnect()
   public void OnEvent(SocketIOClient client){
      log.info("{}断开连接",client.getSessionId());
   public void OnEvent(SocketIOClient client) {
      String connectInfo = client.getHandshakeData().getSingleUrlParam("connectInfo");
      SocketEntity socketEntity = parseMsg(connectInfo);
      clientMap.forEach((key,value)->{
         if (value.getSessionId().equals(client.getSessionId())){
            clientMap.remove(key);
         }
      });
      log.info("设备{}断开连接",socketEntity.getCurrent());
   }

   public void stop() {
@@ -143,5 +180,12 @@
      return sa.substring(1, sa.indexOf(":"));
   }

   /**
    * 解析参数，转化为实体类
    */
   private SocketEntity parseMsg(String msg) {
      return JSON.parseObject(msg, SocketEntity.class);
   }


}