From ac53ebf1b33930f175cf59c26c2a92bbb25366db Mon Sep 17 00:00:00 2001
From: guoshilong <123456>
Date: Sat, 18 Mar 2023 10:45:46 +0800
Subject: [PATCH] sockeio修改
---
src/main/java/org/springblade/common/handler/SocketIOService.java | 181 ++++++++++++++++++++++++++++++++++-----------
1 files changed, 137 insertions(+), 44 deletions(-)
diff --git a/src/main/java/org/springblade/common/handler/SocketIOService.java b/src/main/java/org/springblade/common/handler/SocketIOService.java
index b6c88ec..320d0ae 100644
--- a/src/main/java/org/springblade/common/handler/SocketIOService.java
+++ b/src/main/java/org/springblade/common/handler/SocketIOService.java
@@ -1,11 +1,19 @@
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.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 +27,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;
@@ -35,24 +44,38 @@
private final Logger log = LoggerFactory.getLogger(this.getClass());
// 用来存已连接的客户端
- private static Map<String, SocketIOClient> clientMap = new ConcurrentHashMap<>();
+ private static Map<String, SocketIOClient> viewMap = new ConcurrentHashMap<>();
+
+ // 用来存已连接的客户端
+ private static Map<String, SocketIOClient> controllerMap = 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;
+
+ @Autowired
+ private IModulesService modulesService;
+
/**
- * 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,59 +85,124 @@
}
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();
+ public void onConnect(SocketIOClient client) {
+ FunctionVO functionEntity = getFuncVo(client);
+ //判断页面和控制器是否已经连接过了
+ if (functionEntity.getIsView()) {
+ for (int i = 0; i < viewMap.size(); i++) {
+ if (viewMap.containsKey(functionEntity.getModulesId())) {
+ client.sendEvent("connectError", R.fail("已连接"));
+ return;
+ }
+ }
+ } else {
+ for (int i = 0; i < controllerMap.size(); i++) {
+ if (controllerMap.containsKey(functionEntity.getModulesId())) {
+ client.sendEvent("connectError", R.fail("已连接"));
+ return;
+ }
+ }
+ }
+
+ List<FunctionEntity> all = functionService.getAll(functionEntity);
+ if (all.size() > 0) {
+ if (functionEntity.getIsView()) {
+ log.info("预览页{}", client.getSessionId().toString());
+ viewMap.put(functionEntity.getModulesId(), client);
+
+ } else {
+ log.info("控制页{}", client.getSessionId().toString());
+ controllerMap.put(functionEntity.getModulesId(), client);
+ }
+ 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){
+ public void OnEvent(SocketIOClient client, AckRequest ackRequest, String data) {
log.info("发来消息：" + data);
}
+ /**
+ * 图册上下页控制
+ */
+ @OnEvent(value = "changeImgPage")
+ public void OnEventImgChange(SocketIOClient client, AckRequest ackRequest, String data) {
+ log.info("图册变更：" + data);
+ FunctionVO funcVo = getFuncVo(client);
+
+
+ viewMap.forEach((e,socketIO)->{
+ if (data.equals("previous")){
+ if (e.equals(funcVo.getModulesId())){
+ socketIO.sendEvent("previousPage","previous");
+ }
+ }else if (data.equals("next")){
+ if (e.equals(funcVo.getModulesId())){
+ socketIO.sendEvent("previousPage","next");
+ }
+ }
+ });
+ }
+
+ /**
+ * 菜单控制
+ */
+ @OnEvent(value = "menuChange")
+ public void OnEventMenuChange(SocketIOClient client, AckRequest ackRequest, String data) {
+ log.info("菜单变更：" + data);
+ FunctionVO funcVo = getFuncVo(client);
+ viewMap.forEach((e,socketIOClient)->{
+ if (e.equals(funcVo.getModulesId())){
+ log.info("目标客户端：{}", socketIOClient.getSessionId().toString());
+ socketIOClient.sendEvent("menuChange", data);
+ }
+ });
+ }
+
+ /**
+ * 断开连接时
+ *
+ * @param client
+ */
@OnDisconnect()
- public void OnEvent(SocketIOClient client){
- log.info("{}断开连接",client.getSessionId());
+ public void OnEvent(SocketIOClient client) {
+ FunctionVO funcVo = getFuncVo(client);
+ if (funcVo.getIsView()) {
+ viewMap.forEach((e,socketIOClient)->{
+ if (e.equals(funcVo.getModulesId())){
+ viewMap.remove(e);
+ log.info("预览页：{}断开连接", socketIOClient.getSessionId());
+ }
+ });
+ } else {
+ controllerMap.forEach((e,socketIOClient)->{
+ if (e.equals(funcVo.getModulesId())){
+ controllerMap.remove(e);
+ log.info("控制页：{}断开连接", socketIOClient.getSessionId());
+ }
+ });
+ }
}
public void stop() {
@@ -152,5 +240,10 @@
return sa.substring(1, sa.indexOf(":"));
}
+ private FunctionVO getFuncVo(SocketIOClient client) {
+ String connectInfo = client.getHandshakeData().getSingleUrlParam("connectInfo");
+ return JSON.parseObject(connectInfo, FunctionVO.class);
+ }
+
}
--
Gitblit v1.9.3