center/dataease-dm
center/dataease-dm
13
0
Fork 0
forked from github/dataease
Code Pull Requests Activity

Merge branch 'dev' of github.com:dataease/dataease into dev

Browse Source
This commit is contained in:
taojinlong 2022-04-18 11:59:30 +08:00
commit b9847a2c4a
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
backend/src/main/java/io/dataease/websocket/factory/DeWebSocketHandlerDecorator.java
View File

@ -6,6 +6,8 @@ import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.WebSocketSession; import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.WebSocketHandlerDecorator; import org.springframework.web.socket.handler.WebSocketHandlerDecorator;
import java.util.Optional;
public class DeWebSocketHandlerDecorator extends WebSocketHandlerDecorator { public class DeWebSocketHandlerDecorator extends WebSocketHandlerDecorator {
@ -16,17 +18,22 @@ public class DeWebSocketHandlerDecorator extends WebSocketHandlerDecorator {
@Override @Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception { public void afterConnectionEstablished(WebSocketSession session) throws Exception {
String name = session.getPrincipal().getName(); Optional.ofNullable(session.getPrincipal()).ifPresent(principal -> {
Long userId = Long.parseLong(name); String name = principal.getName();
WsUtil.onLine(userId); Long userId = Long.parseLong(name);
WsUtil.onLine(userId);
});
super.afterConnectionEstablished(session); super.afterConnectionEstablished(session);
} }
@Override @Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception { public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
String name = session.getPrincipal().getName(); Optional.ofNullable(session.getPrincipal()).ifPresent(principal -> {
Long userId = Long.parseLong(name); String name = principal.getName();
WsUtil.offLine(userId); Long userId = Long.parseLong(name);
WsUtil.offLine(userId);
});
super.afterConnectionClosed(session, closeStatus); super.afterConnectionClosed(session, closeStatus);
} }