~admin/ERP_override.git

New file
			@@ -0,0 +1,76 @@
			package com.example.erp.tools;

			import javax.websocket.Session;
			import java.io.IOException;
			import java.util.Iterator;
			import java.util.Map;
			import java.util.concurrent.ConcurrentHashMap;
			import java.util.concurrent.ConcurrentMap;

			/**
			* @author HFL
			* @date 2022/5/16 9:39
			* Websocket连接池、对连接池内连接操作和数据推送方法
			*/
			public class WebSocketServerPool {

			/**
			* WebSocket连接池
			*/
			private static ConcurrentMap<Session, String> dataConnect = new ConcurrentHashMap<>();

			/**
			* 将websocket连接，放入连接池
			* @param session websocket连接
			* @param screen 场景ID
			*/
			public static void addDataConnect(Session session, String screen){
			dataConnect.put(session, screen);
			Iterator<Map.Entry<Session, String>> iterator = dataConnect.entrySet().iterator();
			synchronized (iterator){
			//移除失效连接
			while(iterator.hasNext()){
			Map.Entry<Session, String> entry = iterator.next();
			Session sessionNew = entry.getKey();
			Map<String, Object> userProperties = sessionNew.getUserProperties();
			if(null != userProperties && null != userProperties.get("ReadyState") && "0" != String.valueOf(userProperties.get("ReadyState"))){
			iterator.remove();
			}
			}
			}
			}

			/**
			* 将websocket连接从连接池中移除
			* @param session websocket连接
			*/
			public static void removeConnect(Session session){
			Iterator<Map.Entry<Session, String>> iterator = dataConnect.entrySet().iterator();
			synchronized (iterator){
			//主动移除连接
			while (iterator.hasNext()){
			if(session.equals(iterator.next().getKey())){
			iterator.remove();
			}
			}
			}
			}

			/**
			* 获取连接池中所有连接
			* @return 连接池所有数据
			*/
			public static ConcurrentMap<Session, String> getDataConnect(){
			return dataConnect;
			}

			/**
			* Websocket消息推送
			* @param session 连接
			* @param message 消息主体
			* @throws IOException I/O异常
			*/
			public static void sendMessage(Session session, String message) throws IOException {
			session.getBasicRemote().sendText(message);
			}
			}