Examples with AlarmDataUpdate org.thingsboard.server.service.telemetry.cmd.v2.AlarmDataUpdate used on opensource projects

Example 1 with AlarmDataUpdate

use of org.thingsboard.server.service.telemetry.cmd.v2.AlarmDataUpdate in project thingsboard by thingsboard.

the class TbAlarmDataSubCtx method sendWsMsg.

private void sendWsMsg(String sessionId, AlarmSubscriptionUpdate subscriptionUpdate) {
    Alarm alarm = subscriptionUpdate.getAlarm();
    AlarmId alarmId = alarm.getId();
    if (subscriptionUpdate.isAlarmDeleted()) {
        Alarm deleted = alarmsMap.remove(alarmId);
        if (deleted != null) {
            fetchAlarms();
        }
    } else {
        AlarmData current = alarmsMap.get(alarmId);
        boolean onCurrentPage = current != null;
        boolean matchesFilter = filter(alarm);
        if (onCurrentPage) {
            if (matchesFilter) {
                AlarmData updated = new AlarmData(alarm, current.getOriginatorName(), current.getEntityId());
                updated.getLatest().putAll(current.getLatest());
                alarmsMap.put(alarmId, updated);
                wsService.sendWsMsg(sessionId, new AlarmDataUpdate(cmdId, null, Collections.singletonList(updated), maxEntitiesPerAlarmSubscription, data.getTotalElements()));
            } else {
                fetchAlarms();
            }
        } else if (matchesFilter && query.getPageLink().getPage() == 0) {
            fetchAlarms();
        }
    }
}

Example 2 with AlarmDataUpdate

use of org.thingsboard.server.service.telemetry.cmd.v2.AlarmDataUpdate in project thingsboard by thingsboard.

the class TbAlarmDataSubCtx method doFetchAlarms.

private void doFetchAlarms() {
    AlarmDataUpdate update;
    if (!entitiesMap.isEmpty()) {
        long start = System.currentTimeMillis();
        PageData<AlarmData> alarms = alarmService.findAlarmDataByQueryForEntities(getTenantId(), query, getOrderedEntityIds());
        long end = System.currentTimeMillis();
        stats.getAlarmQueryInvocationCnt().incrementAndGet();
        stats.getAlarmQueryTimeSpent().addAndGet(end - start);
        alarms = setAndMergeAlarmsData(alarms);
        update = new AlarmDataUpdate(cmdId, alarms, null, maxEntitiesPerAlarmSubscription, data.getTotalElements());
    } else {
        update = new AlarmDataUpdate(cmdId, new PageData<>(), null, maxEntitiesPerAlarmSubscription, data.getTotalElements());
    }
    wsService.sendWsMsg(getSessionId(), update);
}

Example 3 with AlarmDataUpdate

use of org.thingsboard.server.service.telemetry.cmd.v2.AlarmDataUpdate in project thingsboard by thingsboard.

the class DefaultTbEntityDataSubscriptionService method handleCmd.

@Override
public void handleCmd(TelemetryWebSocketSessionRef session, AlarmDataCmd cmd) {
    TbAlarmDataSubCtx ctx = getSubCtx(session.getSessionId(), cmd.getCmdId());
    if (ctx == null) {
        log.debug("[{}][{}] Creating new alarm subscription using: {}", session.getSessionId(), cmd.getCmdId(), cmd);
        ctx = createSubCtx(session, cmd);
    }
    ctx.setAndResolveQuery(cmd.getQuery());
    AlarmDataQuery adq = ctx.getQuery();
    long start = System.currentTimeMillis();
    ctx.fetchData();
    long end = System.currentTimeMillis();
    stats.getRegularQueryInvocationCnt().incrementAndGet();
    stats.getRegularQueryTimeSpent().addAndGet(end - start);
    List<EntityData> entities = ctx.getEntitiesData();
    ctx.cancelTasks();
    ctx.clearEntitySubscriptions();
    if (entities.isEmpty()) {
        AlarmDataUpdate update = new AlarmDataUpdate(cmd.getCmdId(), new PageData<>(), null, 0, 0);
        wsService.sendWsMsg(ctx.getSessionId(), update);
    } else {
        ctx.fetchAlarms();
        ctx.createLatestValuesSubscriptions(cmd.getQuery().getLatestValues());
        if (adq.getPageLink().getTimeWindow() > 0) {
            TbAlarmDataSubCtx finalCtx = ctx;
            ScheduledFuture<?> task = scheduler.scheduleWithFixedDelay(finalCtx::checkAndResetInvocationCounter, dynamicPageLinkRefreshInterval, dynamicPageLinkRefreshInterval, TimeUnit.SECONDS);
            finalCtx.setRefreshTask(task);
        }
    }
}

Example 4 with AlarmDataUpdate

use of org.thingsboard.server.service.telemetry.cmd.v2.AlarmDataUpdate in project thingsboard by thingsboard.

the class TbAlarmDataSubCtx method sendWsMsg.

@Override
void sendWsMsg(String sessionId, TelemetrySubscriptionUpdate subscriptionUpdate, EntityKeyType keyType, boolean resultToLatestValues) {
    EntityId entityId = subToEntityIdMap.get(subscriptionUpdate.getSubscriptionId());
    if (entityId != null) {
        Map<String, TsValue> latestUpdate = new HashMap<>();
        subscriptionUpdate.getData().forEach((k, v) -> {
            Object[] data = (Object[]) v.get(0);
            latestUpdate.put(k, new TsValue((Long) data[0], (String) data[1]));
        });
        EntityData entityData = entitiesMap.get(entityId);
        entityData.getLatest().computeIfAbsent(keyType, tmp -> new HashMap<>()).putAll(latestUpdate);
        log.trace("[{}][{}][{}][{}] Received subscription update: {}", sessionId, cmdId, subscriptionUpdate.getSubscriptionId(), keyType, subscriptionUpdate);
        List<AlarmData> update = alarmsMap.values().stream().filter(alarm -> entityId.equals(alarm.getEntityId())).map(alarm -> {
            alarm.getLatest().computeIfAbsent(keyType, tmp -> new HashMap<>()).putAll(latestUpdate);
            return alarm;
        }).collect(Collectors.toList());
        if (!update.isEmpty()) {
            wsService.sendWsMsg(sessionId, new AlarmDataUpdate(cmdId, null, update, maxEntitiesPerAlarmSubscription, data.getTotalElements()));
        }
    } else {
        log.trace("[{}][{}][{}][{}] Received stale subscription update: {}", sessionId, cmdId, subscriptionUpdate.getSubscriptionId(), keyType, subscriptionUpdate);
    }
}