Examples with TelemetryWebSocketSessionRef org.thingsboard.server.service.telemetry.TelemetryWebSocketSessionRef used on opensource projects

Example 1 with TelemetryWebSocketSessionRef

use of org.thingsboard.server.service.telemetry.TelemetryWebSocketSessionRef in project thingsboard by thingsboard.

the class TbWebSocketHandler method toRef.

private TelemetryWebSocketSessionRef toRef(WebSocketSession session) throws IOException {
    URI sessionUri = session.getUri();
    String path = sessionUri.getPath();
    path = path.substring(WebSocketConfiguration.WS_PLUGIN_PREFIX.length());
    if (path.length() == 0) {
        throw new IllegalArgumentException("URL should contain plugin token!");
    }
    String[] pathElements = path.split("/");
    String serviceToken = pathElements[0];
    if (!"telemetry".equalsIgnoreCase(serviceToken)) {
        throw new InvalidParameterException("Can't find plugin with specified token!");
    } else {
        SecurityUser currentUser = (SecurityUser) ((Authentication) session.getPrincipal()).getPrincipal();
        return new TelemetryWebSocketSessionRef(UUID.randomUUID().toString(), currentUser, session.getLocalAddress(), session.getRemoteAddress());
    }
}

Example 2 with TelemetryWebSocketSessionRef

use of org.thingsboard.server.service.telemetry.TelemetryWebSocketSessionRef 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()));
    }
}