|
@@ -20,7 +20,7 @@ public class WebSocketHandler extends AbstractWebSocketHandler {
|
|
|
|
|
|
private static final Logger LOGGER = LoggerFactory.getLogger(WebSocketHandler.class);
|
|
|
|
|
|
- private final Map<String, WebSocketSession> clients = new ConcurrentHashMap<>();
|
|
|
+ public static final Map<String, WebSocketSession> clients = new ConcurrentHashMap<>();
|
|
|
|
|
|
public WebSocketHandler() {
|
|
|
super();
|
|
@@ -28,13 +28,15 @@ public class WebSocketHandler extends AbstractWebSocketHandler {
|
|
|
|
|
|
@Override
|
|
|
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
|
|
|
- LOGGER.info("{}上线", session.getPrincipal().getName());
|
|
|
+ String phone = session.getPrincipal().getName().split(":")[1];
|
|
|
+ LOGGER.info("{}上线", session.getPrincipal().getName().split(":")[1]);
|
|
|
+ clients.put(phone, session);
|
|
|
super.afterConnectionEstablished(session);
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
|
|
|
- LOGGER.info("{}: {}", session.getPrincipal().getName(), message.getPayload());
|
|
|
+ LOGGER.info("{}: {}", session.getPrincipal().getName().split(":")[1], message.getPayload());
|
|
|
super.handleMessage(session, message);
|
|
|
}
|
|
|
|
|
@@ -57,14 +59,14 @@ public class WebSocketHandler extends AbstractWebSocketHandler {
|
|
|
@Override
|
|
|
public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
|
|
|
super.handleTransportError(session, exception);
|
|
|
- LOGGER.info("发生了错误,移除客户端: {}", session.getId());
|
|
|
+ LOGGER.info("发生了错误,移除客户端: {}", session.getPrincipal().getName().split(":")[1]);
|
|
|
session.close();
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
|
|
|
super.afterConnectionClosed(session, status);
|
|
|
- LOGGER.info("{}离线", session.getPrincipal().getName());
|
|
|
+ LOGGER.info("{}离线", session.getPrincipal().getName().split(":")[1]);
|
|
|
}
|
|
|
|
|
|
@Override
|