Search in sources :

Example 16 with AlarmConditionFilterKey

use of org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey in project thingsboard by thingsboard.

the class DeviceState method updateProfile.

public void updateProfile(TbContext ctx, DeviceProfile deviceProfile) throws ExecutionException, InterruptedException {
    Set<AlarmConditionFilterKey> oldKeys = Set.copyOf(this.deviceProfile.getEntityKeys());
    this.deviceProfile.updateDeviceProfile(deviceProfile);
    if (latestValues != null) {
        Set<AlarmConditionFilterKey> keysToFetch = new HashSet<>(this.deviceProfile.getEntityKeys());
        keysToFetch.removeAll(oldKeys);
        if (!keysToFetch.isEmpty()) {
            addEntityKeysToSnapshot(ctx, deviceId, keysToFetch, latestValues);
        }
    }
    Set<String> newAlarmStateIds = this.deviceProfile.getAlarmSettings().stream().map(DeviceProfileAlarm::getId).collect(Collectors.toSet());
    alarmStates.keySet().removeIf(id -> !newAlarmStateIds.contains(id));
    for (DeviceProfileAlarm alarm : this.deviceProfile.getAlarmSettings()) {
        if (alarmStates.containsKey(alarm.getId())) {
            alarmStates.get(alarm.getId()).updateState(alarm, getOrInitPersistedAlarmState(alarm));
        } else {
            alarmStates.putIfAbsent(alarm.getId(), new AlarmState(this.deviceProfile, deviceId, alarm, getOrInitPersistedAlarmState(alarm), dynamicPredicateValueCtx));
        }
    }
}
Also used : PersistedAlarmState(org.thingsboard.rule.engine.profile.state.PersistedAlarmState) DeviceProfileAlarm(org.thingsboard.server.common.data.device.profile.DeviceProfileAlarm) AlarmConditionFilterKey(org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey) HashSet(java.util.HashSet)

Example 17 with AlarmConditionFilterKey

use of org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey in project thingsboard by thingsboard.

the class DeviceState method merge.

private SnapshotUpdate merge(DataSnapshot latestValues, Set<AttributeKvEntry> attributes, String scope) {
    long newTs = 0;
    Set<AlarmConditionFilterKey> keys = new HashSet<>();
    for (AttributeKvEntry entry : attributes) {
        newTs = Math.max(newTs, entry.getLastUpdateTs());
        AlarmConditionFilterKey entityKey = new AlarmConditionFilterKey(AlarmConditionKeyType.ATTRIBUTE, entry.getKey());
        if (latestValues.putValue(entityKey, newTs, toEntityValue(entry))) {
            keys.add(entityKey);
        }
    }
    latestValues.setTs(newTs);
    return new SnapshotUpdate(AlarmConditionKeyType.ATTRIBUTE, keys);
}
Also used : AttributeKvEntry(org.thingsboard.server.common.data.kv.AttributeKvEntry) AlarmConditionFilterKey(org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey) HashSet(java.util.HashSet)

Example 18 with AlarmConditionFilterKey

use of org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey in project thingsboard by thingsboard.

the class DeviceState method merge.

private SnapshotUpdate merge(DataSnapshot latestValues, Long newTs, List<KvEntry> data) {
    Set<AlarmConditionFilterKey> keys = new HashSet<>();
    for (KvEntry entry : data) {
        AlarmConditionFilterKey entityKey = new AlarmConditionFilterKey(AlarmConditionKeyType.TIME_SERIES, entry.getKey());
        if (latestValues.putValue(entityKey, newTs, toEntityValue(entry))) {
            keys.add(entityKey);
        }
    }
    latestValues.setTs(newTs);
    return new SnapshotUpdate(AlarmConditionKeyType.TIME_SERIES, keys);
}
Also used : AttributeKvEntry(org.thingsboard.server.common.data.kv.AttributeKvEntry) TsKvEntry(org.thingsboard.server.common.data.kv.TsKvEntry) KvEntry(org.thingsboard.server.common.data.kv.KvEntry) AlarmConditionFilterKey(org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey) HashSet(java.util.HashSet)

Example 19 with AlarmConditionFilterKey

use of org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey in project thingsboard by thingsboard.

the class DeviceState method addEntityKeysToSnapshot.

private void addEntityKeysToSnapshot(TbContext ctx, EntityId originator, Set<AlarmConditionFilterKey> entityKeysToFetch, DataSnapshot result) throws InterruptedException, ExecutionException {
    Set<String> attributeKeys = new HashSet<>();
    Set<String> latestTsKeys = new HashSet<>();
    Device device = null;
    for (AlarmConditionFilterKey entityKey : entityKeysToFetch) {
        String key = entityKey.getKey();
        switch(entityKey.getType()) {
            case ATTRIBUTE:
                attributeKeys.add(key);
                break;
            case TIME_SERIES:
                latestTsKeys.add(key);
                break;
            case ENTITY_FIELD:
                if (device == null) {
                    device = ctx.getDeviceService().findDeviceById(ctx.getTenantId(), new DeviceId(originator.getId()));
                }
                if (device != null) {
                    switch(key) {
                        case EntityKeyMapping.NAME:
                            result.putValue(entityKey, device.getCreatedTime(), EntityKeyValue.fromString(device.getName()));
                            break;
                        case EntityKeyMapping.TYPE:
                            result.putValue(entityKey, device.getCreatedTime(), EntityKeyValue.fromString(device.getType()));
                            break;
                        case EntityKeyMapping.CREATED_TIME:
                            result.putValue(entityKey, device.getCreatedTime(), EntityKeyValue.fromLong(device.getCreatedTime()));
                            break;
                        case EntityKeyMapping.LABEL:
                            result.putValue(entityKey, device.getCreatedTime(), EntityKeyValue.fromString(device.getLabel()));
                            break;
                    }
                }
                break;
        }
    }
    if (!latestTsKeys.isEmpty()) {
        List<TsKvEntry> data = ctx.getTimeseriesService().findLatest(ctx.getTenantId(), originator, latestTsKeys).get();
        for (TsKvEntry entry : data) {
            if (entry.getValue() != null) {
                result.putValue(new AlarmConditionFilterKey(AlarmConditionKeyType.TIME_SERIES, entry.getKey()), entry.getTs(), toEntityValue(entry));
            }
        }
    }
    if (!attributeKeys.isEmpty()) {
        addToSnapshot(result, ctx.getAttributesService().find(ctx.getTenantId(), originator, DataConstants.CLIENT_SCOPE, attributeKeys).get());
        addToSnapshot(result, ctx.getAttributesService().find(ctx.getTenantId(), originator, DataConstants.SHARED_SCOPE, attributeKeys).get());
        addToSnapshot(result, ctx.getAttributesService().find(ctx.getTenantId(), originator, DataConstants.SERVER_SCOPE, attributeKeys).get());
    }
}
Also used : TsKvEntry(org.thingsboard.server.common.data.kv.TsKvEntry) Device(org.thingsboard.server.common.data.Device) DeviceId(org.thingsboard.server.common.data.id.DeviceId) AlarmConditionFilterKey(org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey) HashSet(java.util.HashSet)

Example 20 with AlarmConditionFilterKey

use of org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey in project thingsboard by thingsboard.

the class TbDeviceProfileNodeTest method testInheritTenantAttributeForRepeating.

@Test
public void testInheritTenantAttributeForRepeating() throws Exception {
    init();
    DeviceProfile deviceProfile = new DeviceProfile();
    deviceProfile.setId(deviceProfileId);
    DeviceProfileData deviceProfileData = new DeviceProfileData();
    AttributeKvCompositeKey compositeKey = new AttributeKvCompositeKey(EntityType.TENANT, deviceId.getId(), "SERVER_SCOPE", "greaterAttribute");
    Device device = new Device();
    device.setId(deviceId);
    device.setCustomerId(customerId);
    AttributeKvEntity attributeKvEntity = new AttributeKvEntity();
    attributeKvEntity.setId(compositeKey);
    attributeKvEntity.setLongValue(30L);
    attributeKvEntity.setLastUpdateTs(0L);
    AttributeKvCompositeKey alarmDelayCompositeKey = new AttributeKvCompositeKey(EntityType.TENANT, deviceId.getId(), "SERVER_SCOPE", "alarm_delay");
    AttributeKvEntity alarmDelayAttributeKvEntity = new AttributeKvEntity();
    alarmDelayAttributeKvEntity.setId(alarmDelayCompositeKey);
    long repeatingCondition = 2;
    alarmDelayAttributeKvEntity.setLongValue(repeatingCondition);
    alarmDelayAttributeKvEntity.setLastUpdateTs(0L);
    AttributeKvEntry entry = attributeKvEntity.toData();
    AttributeKvEntry alarmDelayAttributeKvEntry = alarmDelayAttributeKvEntity.toData();
    ListenableFuture<Optional<AttributeKvEntry>> optionalDurationAttribute = Futures.immediateFuture(Optional.of(alarmDelayAttributeKvEntry));
    ListenableFuture<List<AttributeKvEntry>> listNoDurationAttribute = Futures.immediateFuture(Collections.singletonList(entry));
    ListenableFuture<Optional<AttributeKvEntry>> emptyOptional = Futures.immediateFuture(Optional.empty());
    AlarmConditionFilter highTempFilter = new AlarmConditionFilter();
    highTempFilter.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.TIME_SERIES, "temperature"));
    highTempFilter.setValueType(EntityKeyValueType.NUMERIC);
    NumericFilterPredicate highTemperaturePredicate = new NumericFilterPredicate();
    highTemperaturePredicate.setOperation(NumericFilterPredicate.NumericOperation.GREATER);
    highTemperaturePredicate.setValue(new FilterPredicateValue<>(0.0, null, new DynamicValue<>(DynamicValueSourceType.CURRENT_DEVICE, "greaterAttribute", false)));
    highTempFilter.setPredicate(highTemperaturePredicate);
    AlarmCondition alarmCondition = new AlarmCondition();
    alarmCondition.setCondition(Collections.singletonList(highTempFilter));
    FilterPredicateValue<Integer> filterPredicateValue = new FilterPredicateValue<>(10, null, new DynamicValue<>(DynamicValueSourceType.CURRENT_DEVICE, "alarm_delay", true));
    RepeatingAlarmConditionSpec repeatingSpec = new RepeatingAlarmConditionSpec();
    repeatingSpec.setPredicate(filterPredicateValue);
    alarmCondition.setSpec(repeatingSpec);
    AlarmRule alarmRule = new AlarmRule();
    alarmRule.setCondition(alarmCondition);
    DeviceProfileAlarm dpa = new DeviceProfileAlarm();
    dpa.setId("highTemperatureAlarmID");
    dpa.setAlarmType("highTemperatureAlarm");
    dpa.setCreateRules(new TreeMap<>(Collections.singletonMap(AlarmSeverity.CRITICAL, alarmRule)));
    deviceProfileData.setAlarms(Collections.singletonList(dpa));
    deviceProfile.setProfileData(deviceProfileData);
    Mockito.when(cache.get(tenantId, deviceId)).thenReturn(deviceProfile);
    Mockito.when(timeseriesService.findLatest(tenantId, deviceId, Collections.singleton("temperature"))).thenReturn(Futures.immediateFuture(Collections.emptyList()));
    Mockito.when(alarmService.findLatestByOriginatorAndType(tenantId, deviceId, "highTemperatureAlarm")).thenReturn(Futures.immediateFuture(null));
    Mockito.when(alarmService.createOrUpdateAlarm(Mockito.any())).thenAnswer(AdditionalAnswers.returnsFirstArg());
    Mockito.when(ctx.getAttributesService()).thenReturn(attributesService);
    Mockito.when(attributesService.find(eq(tenantId), eq(tenantId), Mockito.anyString(), Mockito.anyString())).thenReturn(optionalDurationAttribute);
    Mockito.when(ctx.getDeviceService().findDeviceById(tenantId, deviceId)).thenReturn(device);
    Mockito.when(attributesService.find(eq(tenantId), eq(customerId), eq(DataConstants.SERVER_SCOPE), Mockito.anyString())).thenReturn(emptyOptional);
    Mockito.when(attributesService.find(eq(tenantId), eq(deviceId), Mockito.anyString(), Mockito.anySet())).thenReturn(listNoDurationAttribute);
    TbMsg theMsg = TbMsg.newMsg("ALARM", deviceId, new TbMsgMetaData(), "");
    Mockito.when(ctx.newMsg(Mockito.anyString(), Mockito.anyString(), Mockito.any(), Mockito.any(), Mockito.any(), Mockito.anyString())).thenReturn(theMsg);
    ObjectNode data = mapper.createObjectNode();
    data.put("temperature", 150);
    TbMsg msg = TbMsg.newMsg(SessionMsgType.POST_TELEMETRY_REQUEST.name(), deviceId, new TbMsgMetaData(), TbMsgDataType.JSON, mapper.writeValueAsString(data), null, null);
    node.onMsg(ctx, msg);
    verify(ctx).tellSuccess(msg);
    verify(ctx, Mockito.never()).tellNext(theMsg, "Alarm Created");
    data.put("temperature", 151);
    TbMsg msg2 = TbMsg.newMsg(SessionMsgType.POST_TELEMETRY_REQUEST.name(), deviceId, new TbMsgMetaData(), TbMsgDataType.JSON, mapper.writeValueAsString(data), null, null);
    node.onMsg(ctx, msg2);
    verify(ctx).tellSuccess(msg2);
    verify(ctx).enqueueForTellNext(theMsg, "Alarm Created");
    verify(ctx, Mockito.never()).tellFailure(Mockito.any(), Mockito.any());
}
Also used : AttributeKvEntry(org.thingsboard.server.common.data.kv.AttributeKvEntry) AttributeKvEntity(org.thingsboard.server.dao.model.sql.AttributeKvEntity) AlarmRule(org.thingsboard.server.common.data.device.profile.AlarmRule) FilterPredicateValue(org.thingsboard.server.common.data.query.FilterPredicateValue) TbMsgMetaData(org.thingsboard.server.common.msg.TbMsgMetaData) DeviceProfile(org.thingsboard.server.common.data.DeviceProfile) List(java.util.List) DeviceProfileData(org.thingsboard.server.common.data.device.profile.DeviceProfileData) AlarmConditionFilterKey(org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey) AttributeKvCompositeKey(org.thingsboard.server.dao.model.sql.AttributeKvCompositeKey) TbMsg(org.thingsboard.server.common.msg.TbMsg) Optional(java.util.Optional) ObjectNode(com.fasterxml.jackson.databind.node.ObjectNode) NumericFilterPredicate(org.thingsboard.server.common.data.query.NumericFilterPredicate) Device(org.thingsboard.server.common.data.Device) AlarmConditionFilter(org.thingsboard.server.common.data.device.profile.AlarmConditionFilter) AlarmCondition(org.thingsboard.server.common.data.device.profile.AlarmCondition) DeviceProfileAlarm(org.thingsboard.server.common.data.device.profile.DeviceProfileAlarm) RepeatingAlarmConditionSpec(org.thingsboard.server.common.data.device.profile.RepeatingAlarmConditionSpec) DynamicValue(org.thingsboard.server.common.data.query.DynamicValue) Test(org.junit.Test)

Aggregations

AlarmConditionFilterKey (org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey)24 DeviceProfileAlarm (org.thingsboard.server.common.data.device.profile.DeviceProfileAlarm)18 AlarmRule (org.thingsboard.server.common.data.device.profile.AlarmRule)17 DeviceProfile (org.thingsboard.server.common.data.DeviceProfile)16 AlarmCondition (org.thingsboard.server.common.data.device.profile.AlarmCondition)16 AlarmConditionFilter (org.thingsboard.server.common.data.device.profile.AlarmConditionFilter)16 DeviceProfileData (org.thingsboard.server.common.data.device.profile.DeviceProfileData)16 AttributeKvEntry (org.thingsboard.server.common.data.kv.AttributeKvEntry)16 NumericFilterPredicate (org.thingsboard.server.common.data.query.NumericFilterPredicate)16 DynamicValue (org.thingsboard.server.common.data.query.DynamicValue)15 ObjectNode (com.fasterxml.jackson.databind.node.ObjectNode)14 Test (org.junit.Test)14 TbMsg (org.thingsboard.server.common.msg.TbMsg)14 TbMsgMetaData (org.thingsboard.server.common.msg.TbMsgMetaData)14 List (java.util.List)13 AttributeKvCompositeKey (org.thingsboard.server.dao.model.sql.AttributeKvCompositeKey)13 AttributeKvEntity (org.thingsboard.server.dao.model.sql.AttributeKvEntity)13 Device (org.thingsboard.server.common.data.Device)10 Optional (java.util.Optional)7 FilterPredicateValue (org.thingsboard.server.common.data.query.FilterPredicateValue)6