Examples with WsSessionMetaData org.thingsboard.server.extensions.api.plugins.ws.WsSessionMetaData used on opensource projects

Example 1 with WsSessionMetaData

use of org.thingsboard.server.extensions.api.plugins.ws.WsSessionMetaData in project thingsboard by thingsboard.

the class TelemetryWebsocketMsgHandler method validateSessionMetadata.

private boolean validateSessionMetadata(PluginContext ctx, PluginWebsocketSessionRef sessionRef, SubscriptionCmd cmd, String sessionId) {
    WsSessionMetaData sessionMD = wsSessionsMap.get(sessionId);
    if (sessionMD == null) {
        log.warn("[{}] Session meta data not found. ", sessionId);
        SubscriptionUpdate update = new SubscriptionUpdate(cmd.getCmdId(), SubscriptionErrorCode.INTERNAL_ERROR, SESSION_META_DATA_NOT_FOUND);
        sendWsMsg(ctx, sessionRef, update);
        return false;
    } else {
        return true;
    }
}

Example 2 with WsSessionMetaData

use of org.thingsboard.server.extensions.api.plugins.ws.WsSessionMetaData in project thingsboard by thingsboard.

the class TelemetryWebsocketMsgHandler method handleWsHistoryCmd.

private void handleWsHistoryCmd(PluginContext ctx, PluginWebsocketSessionRef sessionRef, GetHistoryCmd cmd) {
    String sessionId = sessionRef.getSessionId();
    WsSessionMetaData sessionMD = wsSessionsMap.get(sessionId);
    if (sessionMD == null) {
        log.warn("[{}] Session meta data not found. ", sessionId);
        SubscriptionUpdate update = new SubscriptionUpdate(cmd.getCmdId(), SubscriptionErrorCode.INTERNAL_ERROR, SESSION_META_DATA_NOT_FOUND);
        sendWsMsg(ctx, sessionRef, update);
        return;
    }
    if (cmd.getEntityId() == null || cmd.getEntityId().isEmpty() || cmd.getEntityType() == null || cmd.getEntityType().isEmpty()) {
        SubscriptionUpdate update = new SubscriptionUpdate(cmd.getCmdId(), SubscriptionErrorCode.BAD_REQUEST, "Device id is empty!");
        sendWsMsg(ctx, sessionRef, update);
        return;
    }
    if (cmd.getKeys() == null || cmd.getKeys().isEmpty()) {
        SubscriptionUpdate update = new SubscriptionUpdate(cmd.getCmdId(), SubscriptionErrorCode.BAD_REQUEST, "Keys are empty!");
        sendWsMsg(ctx, sessionRef, update);
        return;
    }
    EntityId entityId = EntityIdFactory.getByTypeAndId(cmd.getEntityType(), cmd.getEntityId());
    List<String> keys = new ArrayList<>(getKeys(cmd).orElse(Collections.emptySet()));
    List<TsKvQuery> queries = keys.stream().map(key -> new BaseTsKvQuery(key, cmd.getStartTs(), cmd.getEndTs(), cmd.getInterval(), getLimit(cmd.getLimit()), getAggregation(cmd.getAgg()))).collect(Collectors.toList());
    ctx.loadTimeseries(entityId, queries, new PluginCallback<List<TsKvEntry>>() {

        @Override
        public void onSuccess(PluginContext ctx, List<TsKvEntry> data) {
            sendWsMsg(ctx, sessionRef, new SubscriptionUpdate(cmd.getCmdId(), data));
        }

        @Override
        public void onFailure(PluginContext ctx, Exception e) {
            SubscriptionUpdate update;
            if (UnauthorizedException.class.isInstance(e)) {
                update = new SubscriptionUpdate(cmd.getCmdId(), SubscriptionErrorCode.UNAUTHORIZED, SubscriptionErrorCode.UNAUTHORIZED.getDefaultMsg());
            } else {
                update = new SubscriptionUpdate(cmd.getCmdId(), SubscriptionErrorCode.INTERNAL_ERROR, FAILED_TO_FETCH_DATA);
            }
            sendWsMsg(ctx, sessionRef, update);
        }
    });
}

Example 3 with WsSessionMetaData

use of org.thingsboard.server.extensions.api.plugins.ws.WsSessionMetaData in project thingsboard by thingsboard.

the class DefaultWebsocketMsgHandler method handleWebSocketSessionEvent.

protected void handleWebSocketSessionEvent(PluginContext ctx, PluginWebsocketSessionRef sessionRef, SessionEventPluginWebSocketMsg wsMsg) {
    String sessionId = sessionRef.getSessionId();
    SessionEvent event = wsMsg.getPayload();
    log.debug(PROCESSING_MSG, sessionId, event);
    switch(event.getEventType()) {
        case ESTABLISHED:
            wsSessionsMap.put(sessionId, new WsSessionMetaData(sessionRef));
            break;
        case ERROR:
            log.debug("[{}] Unknown websocket session error: {}. ", sessionId, event.getError().orElse(null));
            break;
        case CLOSED:
            wsSessionsMap.remove(sessionId);
            cleanupWebSocketSession(ctx, sessionId);
            break;
    }
}

Example 4 with WsSessionMetaData

use of org.thingsboard.server.extensions.api.plugins.ws.WsSessionMetaData in project thingsboard by thingsboard.

the class DefaultWebsocketMsgHandler method handleWebSocketPongEvent.

protected void handleWebSocketPongEvent(PluginContext ctx, PluginWebsocketSessionRef sessionRef) {
    String sessionId = sessionRef.getSessionId();
    WsSessionMetaData sessionMD = wsSessionsMap.get(sessionId);
    if (sessionMD != null) {
        log.debug("[{}] Updating session metadata: {}", sessionId, sessionRef);
        sessionMD.setSessionRef(sessionRef);
        sessionMD.setLastActivityTime(System.currentTimeMillis());
    }
}