joylink/rt-sim-training-service
6
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases 10 Wiki Activity

去掉一个token只记录一个ws连接的限制

Browse Source
This commit is contained in:
joylink_zhangsai 2022-05-25 13:05:20 +08:00
parent bd58435a06
commit c937230ed5
1 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/main/java/club/joylink/rtss/simulation/messaging/websocket/DefaultSubscribeManager.java
View File

@ -18,12 +18,12 @@ import java.util.concurrent.ConcurrentHashMap;
@Slf4j
@Component
public class DefaultSubscribeManager {
/**
* 暂时限制一个token只能对应一个wsSessionId
* k - token
* v - wsSessionId
*/
public static final Map<String, String> tokenSessionMap = new ConcurrentHashMap<>();
// /**
// * 暂时限制一个token只能对应一个wsSessionId
// * k - token
// * v - wsSessionId
// */
// public static final Map<String, String> tokenSessionMap = new ConcurrentHashMap<>();
/**
* key-wsSessionId
* val-simulationId
@ -44,11 +44,11 @@ public class DefaultSubscribeManager {
public void disconnect(WebsocketConfig.MyPrincipal user, String wsSessionId) {
if (wsSessionId == null)
return;
String token = user.getToken();
String connectedSession = tokenSessionMap.get(token);
if (wsSessionId.equals(connectedSession)) { //当前的sessionId和要断开的一致
tokenSessionMap.remove(token);
}
// String token = user.getToken();
// String connectedSession = tokenSessionMap.get(token);
// if (wsSessionId.equals(connectedSession)) { //当前的sessionId和要断开的一致
// tokenSessionMap.remove(token);
// }
String sid = wsSidMap.remove(wsSessionId);
if (sid == null) {
return;
@ -66,13 +66,13 @@ public class DefaultSubscribeManager {
}
public synchronized void subscribe(WebsocketConfig.MyPrincipal user, String wsSessionId, String subId, String destination) {
String token = user.getToken();
String oldSession = tokenSessionMap.get(token);
if (!wsSessionId.equals(oldSession)) { //一个token出现多次连接
disconnect(user, oldSession); //断开旧连接
}
tokenSessionMap.put(token, wsSessionId); //记录新连接
// String token = user.getToken();
// String oldSession = tokenSessionMap.get(token);
// if (!wsSessionId.equals(oldSession)) { //一个token出现多次连接
// disconnect(user, oldSession); //断开旧连接
// }
//
// tokenSessionMap.put(token, wsSessionId); //记录新连接
if (destination.startsWith("/user")) {
destination = destination.substring(5);
}