sockeio修改

guoshilong

2023-03-17 ee917d6bc0e9c511e76d1b5c8b5d17896e32ef7d

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

@@ -1,11 +1,16 @@
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.tool.api.R;
import org.springblade.modules.modules.entity.FunctionEntity;
import org.springblade.modules.modules.service.IFunctionService;
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 +24,7 @@
import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import java.net.ServerSocket;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
@@ -37,22 +43,30 @@
   // 用来存已连接的客户端
   private static Map<String, SocketIOClient> clientMap = new ConcurrentHashMap<>();

   // 用来存已连接的视图客户端
   private List<Map<String, SocketIOClient>> view = new ArrayList<>();

   // 用来存已连接的控制客户端
   private List<Map<String, SocketIOClient>> controller = new ArrayList<>();

   @Autowired
   private SocketIOServer socketIOServer;

   @Autowired
   private IFunctionService functionService;

   /**
    * Spring IoC容器创建之后，在加载SocketIOServiceImpl Bean之后启动
    * Spring IoC容器创建之后，在加载SocketIOConfiguration Bean之后启动
    *
    * @throws Exception
    */
   @PostConstruct
   private void autoStartup() throws Exception {
      log.info("启动Socket！！！！！！！！！！");
      start();
   }

   /**
    * Spring IoC容器在销毁SocketIOServiceImpl Bean之前关闭,避免重启项目服务端口占用问题
    * Spring IoC容器在销毁SocketIOConfiguration Bean之前关闭,避免重启项目服务端口占用问题
    *
    * @throws Exception
    */
@@ -62,58 +76,111 @@
   }

   public void start() {
      // 监听客户端连接
      socketIOServer.addConnectListener(client -> {
         String uid = getParamsByClient(client);
         if (uid != null) {
            clientMap.put(uid, client);
            log.info("有新的客户端连接UID:{}", uid);
         }
         // 给客户端发送一条信息 发送ClientReceive事件 需要客户端绑定此事件即可接收到消息
         JSONObject jsonObject = new JSONObject();
         jsonObject.put("name", "goat");
         jsonObject.put("message", "hello client");
         client.sendEvent("ClientReceive", jsonObject);
      });

      // 监听客户端断开连接
      socketIOServer.addDisconnectListener(listener -> {
         String uid = getParamsByClient(listener);
         if (uid != null) {
            clientMap.remove(uid);
            listener.disconnect();
            log.info("一条客户端连接中断");
         }
      });

      socketIOServer.addEventListener("ServerReceive", JSONObject.class, (client, data, ackSender) -> {
         String uid = getParamsByClient(client);
         String ip = getIpByClient(client);
         if (uid != null) {
            log.info("接收到SID:{}发来的消息:{}", uid, data.toJSONString());
            log.debug(ip + " ************ 客户端：" + data);
         }
      });

      socketIOServer.start();
      log.info("socket.io初始化服务完成");
   }

   /**
    * 连接时
    * @param client
    */
   @OnConnect
   public void onConnect(SocketIOClient client){
      log.info(getIpByClient(client));
      HandshakeData handshakeData = client.getHandshakeData();
      Map<String, List<String>> urlParams = handshakeData.getUrlParams();
      FunctionVO functionEntity = getFuncVo(client);

      //判断页面和控制器是否已经连接过了
      if (functionEntity.getIsView()){
         for (int i = 0; i < view.size(); i++) {
            if (view.get(i).containsKey(functionEntity.getModulesId())){
               client.sendEvent("connectError",R.fail("已连接"));
               return;
            }
         }
      }else {
         for (int i = 0; i < controller.size(); i++) {
            if (controller.get(i).containsKey(functionEntity.getModulesId())){
               client.sendEvent("connectError",R.fail("已连接"));
               return;
            }
         }
      }

      List<FunctionEntity> all = functionService.getAll(functionEntity);
      log.info(client.getSessionId().toString());
      if (all.size()>0){
         clientMap.put(functionEntity.getModulesId(),client);
         if (functionEntity.getIsView()){
               view.add(clientMap);
         }else {
               controller.add(clientMap);
         }
         client.sendEvent("connectOk", R.data(all));
      }else {
         client.sendEvent("connectError",R.fail("连接失败"));
      }
   }

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

   /**
    * 监听前端订阅相同事件发送过来的信息
    * @param client
    * @param ackRequest
    * @param data
    */
   @OnEvent(value = "changeImgPage")
   public void OnEventImgChange(SocketIOClient client, AckRequest ackRequest, String data){
      log.info("图册变更：" + data);
      FunctionVO funcVo = getFuncVo(client);
      view.forEach(e->{
         if (data.equals("previous")){
            e.get(funcVo.getModulesId()).sendEvent("previousPage","previous");
         }else if (data.equals("next")){
            e.get(funcVo.getModulesId()).sendEvent("nextPage","next");
         }
      });
   }

   @OnEvent(value = "menuChange")
   public void OnEventMenuChange(SocketIOClient client, AckRequest ackRequest, String data){
      log.info("菜单变更：" + data);
      FunctionVO funcVo = getFuncVo(client);
      view.forEach(e->{
         e.get(funcVo.getModulesId()).sendEvent("menuChange",data);
      });
   }

   /**
    * 断开连接时
    * @param client
    */
   @OnDisconnect()
   public void OnEvent(SocketIOClient client){


      FunctionVO funcVo = getFuncVo(client);
      if (funcVo.getIsView()){
         for (int i = 0; i <=view.size() ; i++) {
            if (view.get(i).containsKey(funcVo.getModulesId())) {
               view.remove(i);
            }
         }
      }else {
         for (int i = 0; i <=controller.size() ; i++) {
            if (controller.get(i).containsKey(funcVo.getModulesId())) {
               controller.remove(i);
            }
         }
      }
      log.info("{}断开连接",client.getSessionId());
   }

@@ -152,5 +219,10 @@
      return sa.substring(1, sa.indexOf(":"));
   }

   private FunctionVO getFuncVo(SocketIOClient client){
      String connectInfo = client.getHandshakeData().getSingleUrlParam("connectInfo");
      return JSON.parseObject(connectInfo, FunctionVO.class);
   }


}