| | |
| | | @Resource |
| | | private TokenService tokenService; |
| | | private final Map<Long, Long> lastDisconnectTime = new ConcurrentHashMap<>(); |
| | | |
| | | @Override |
| | | public void handleTextMessage(WebSocketSession session, TextMessage message) { |
| | | try { |
| | | String userIdStr = (String) session.getAttributes().get("userId"); |
| | | Long userId = Long.parseLong(userIdStr); |
| | | String payload = message.getPayload(); |
| | | log.info("【cmd】Received【{}】message: {} size: {}", userId, payload, |
| | | log.info("【ws】【{}】Received message: {}, size: {}", userId, payload, |
| | | WebSocketSessionManager.getOnlineUserCount()); |
| | | BaseMessage<?> baseMessage = JSON.parseObject(payload, BaseMessage.class); |
| | | MessageType messageType = MessageType.fromString(baseMessage.getType()); |
| | |
| | | |
| | | @Override |
| | | public void afterConnectionEstablished(WebSocketSession session) { |
| | | log.info("【cmd】Connected: {}", session.getId()); |
| | | String userIdStr = (String) session.getAttributes().get("userId"); |
| | | if (userIdStr == null || userIdStr.isBlank()) { |
| | | return; |
| | |
| | | Long userId = Long.parseLong(userIdStr); |
| | | lastDisconnectTime.remove(userId); // 清除断开标记 |
| | | WebSocketSessionManager.addSession(userId, session); |
| | | log.info("【cmd】size: {}", WebSocketSessionManager.getOnlineUserCount()); |
| | | log.info("【ws】【{}】Connected,size: {}",userId, WebSocketSessionManager.getOnlineUserCount()); |
| | | } |
| | | |
| | | @Override |
| | | public void afterConnectionClosed(WebSocketSession session, CloseStatus status) { |
| | | log.info("【cmd】Disconnected: {}", session.getId()); |
| | | |
| | | String userIdStr = (String) session.getAttributes().get("userId"); |
| | | if (userIdStr == null || userIdStr.isBlank()) { |
| | | return; |
| | | } |
| | | Long userId = Long.parseLong(userIdStr); |
| | | WebSocketSessionManager.removeSession(userId); |
| | | log.info("【cmd】size: {}", WebSocketSessionManager.getOnlineUserCount()); |
| | | log.info("【ws】【{}】Disconnected,size: {}",userId, WebSocketSessionManager.getOnlineUserCount()); |
| | | long version = System.currentTimeMillis(); |
| | | lastDisconnectTime.put(userId, version); |
| | | // 释放该用户所有 UAV 控制权 |