Examples with NativeWebSocketSession org.springframework.web.socket.adapter.NativeWebSocketSession used on opensource projects

Example 1 with NativeWebSocketSession

use of org.springframework.web.socket.adapter.NativeWebSocketSession in project thingsboard by thingsboard.

the class TbWebSocketHandler method afterConnectionEstablished.

@Override
public void afterConnectionEstablished(WebSocketSession session) throws Exception {
    super.afterConnectionEstablished(session);
    try {
        if (session instanceof NativeWebSocketSession) {
            Session nativeSession = ((NativeWebSocketSession) session).getNativeSession(Session.class);
            if (nativeSession != null) {
                nativeSession.getAsyncRemote().setSendTimeout(sendTimeout);
            }
        }
        String internalSessionId = session.getId();
        TelemetryWebSocketSessionRef sessionRef = toRef(session);
        String externalSessionId = sessionRef.getSessionId();
        if (!checkLimits(session, sessionRef)) {
            return;
        }
        internalSessionMap.put(internalSessionId, new SessionMetaData(session, sessionRef, maxMsgQueuePerSession));
        externalSessionMap.put(externalSessionId, internalSessionId);
        processInWebSocketService(sessionRef, SessionEvent.onEstablished());
        log.info("[{}][{}][{}] Session is opened", sessionRef.getSecurityCtx().getTenantId(), externalSessionId, session.getId());
    } catch (InvalidParameterException e) {
        log.warn("[{}] Failed to start session", session.getId(), e);
        session.close(CloseStatus.BAD_DATA.withReason(e.getMessage()));
    } catch (Exception e) {
        log.warn("[{}] Failed to start session", session.getId(), e);
        session.close(CloseStatus.SERVER_ERROR.withReason(e.getMessage()));
    }
}