Examples with AlarmConditionFilterKey org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey used on opensource projects

Example 21 with AlarmConditionFilterKey

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

the class TbDeviceProfileNodeTest method testCurrentDeviceAttributeForUseDefaultRepeatingWhenDynamicDurationValueIsNull.

@Test
public void testCurrentDeviceAttributeForUseDefaultRepeatingWhenDynamicDurationValueIsNull() throws Exception {
    init();
    DeviceProfile deviceProfile = new DeviceProfile();
    deviceProfile.setId(deviceProfileId);
    DeviceProfileData deviceProfileData = new DeviceProfileData();
    Device device = new Device();
    device.setId(deviceId);
    device.setCustomerId(customerId);
    AttributeKvCompositeKey compositeKey = new AttributeKvCompositeKey(EntityType.TENANT, deviceId.getId(), "SERVER_SCOPE", "greaterAttribute");
    AttributeKvEntity attributeKvEntity = new AttributeKvEntity();
    attributeKvEntity.setId(compositeKey);
    attributeKvEntity.setLongValue(30L);
    attributeKvEntity.setLastUpdateTs(0L);
    AttributeKvEntry entry = attributeKvEntity.toData();
    ListenableFuture<List<AttributeKvEntry>> listListenableFuture = Futures.immediateFuture(Collections.singletonList(entry));
    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")));
    highTempFilter.setPredicate(highTemperaturePredicate);
    AlarmCondition alarmCondition = new AlarmCondition();
    alarmCondition.setCondition(Collections.singletonList(highTempFilter));
    RepeatingAlarmConditionSpec repeating = new RepeatingAlarmConditionSpec();
    repeating.setPredicate(new FilterPredicateValue<>(0, null, new DynamicValue<>(DynamicValueSourceType.CURRENT_DEVICE, "alarm_rule", false)));
    alarmCondition.setSpec(repeating);
    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(deviceId), Mockito.anyString(), Mockito.anySet())).thenReturn(listListenableFuture);
    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", 35);
    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).enqueueForTellNext(theMsg, "Alarm Created");
    verify(ctx, Mockito.never()).tellFailure(Mockito.any(), Mockito.any());
}

Example 22 with AlarmConditionFilterKey

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

the class ProfileState method updateDeviceProfile.

void updateDeviceProfile(DeviceProfile deviceProfile) {
    this.deviceProfile = deviceProfile;
    alarmSettings.clear();
    alarmCreateKeys.clear();
    alarmClearKeys.clear();
    entityKeys.clear();
    if (deviceProfile.getProfileData().getAlarms() != null) {
        alarmSettings.addAll(deviceProfile.getProfileData().getAlarms());
        for (DeviceProfileAlarm alarm : deviceProfile.getProfileData().getAlarms()) {
            Map<AlarmSeverity, Set<AlarmConditionFilterKey>> createAlarmKeys = alarmCreateKeys.computeIfAbsent(alarm.getId(), id -> new HashMap<>());
            alarm.getCreateRules().forEach(((severity, alarmRule) -> {
                var ruleKeys = createAlarmKeys.computeIfAbsent(severity, id -> new HashSet<>());
                for (var keyFilter : alarmRule.getCondition().getCondition()) {
                    entityKeys.add(keyFilter.getKey());
                    ruleKeys.add(keyFilter.getKey());
                    addDynamicValuesRecursively(keyFilter.getPredicate(), entityKeys, ruleKeys);
                }
                addEntityKeysFromAlarmConditionSpec(alarmRule);
            }));
            if (alarm.getClearRule() != null) {
                var clearAlarmKeys = alarmClearKeys.computeIfAbsent(alarm.getId(), id -> new HashSet<>());
                for (var keyFilter : alarm.getClearRule().getCondition().getCondition()) {
                    entityKeys.add(keyFilter.getKey());
                    clearAlarmKeys.add(keyFilter.getKey());
                    addDynamicValuesRecursively(keyFilter.getPredicate(), entityKeys, clearAlarmKeys);
                }
                addEntityKeysFromAlarmConditionSpec(alarm.getClearRule());
            }
        }
    }
}

Example 23 with AlarmConditionFilterKey

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

the class ProfileState method addEntityKeysFromAlarmConditionSpec.

private void addEntityKeysFromAlarmConditionSpec(AlarmRule alarmRule) {
    AlarmConditionSpec spec = alarmRule.getCondition().getSpec();
    if (spec == null) {
        return;
    }
    AlarmConditionSpecType specType = spec.getType();
    switch(specType) {
        case DURATION:
            DurationAlarmConditionSpec duration = (DurationAlarmConditionSpec) spec;
            if (duration.getPredicate().getDynamicValue() != null && duration.getPredicate().getDynamicValue().getSourceAttribute() != null) {
                entityKeys.add(new AlarmConditionFilterKey(AlarmConditionKeyType.ATTRIBUTE, duration.getPredicate().getDynamicValue().getSourceAttribute()));
            }
            break;
        case REPEATING:
            RepeatingAlarmConditionSpec repeating = (RepeatingAlarmConditionSpec) spec;
            if (repeating.getPredicate().getDynamicValue() != null && repeating.getPredicate().getDynamicValue().getSourceAttribute() != null) {
                entityKeys.add(new AlarmConditionFilterKey(AlarmConditionKeyType.ATTRIBUTE, repeating.getPredicate().getDynamicValue().getSourceAttribute()));
            }
            break;
    }
}

Example 24 with AlarmConditionFilterKey

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

the class DeviceState method addToSnapshot.

private void addToSnapshot(DataSnapshot snapshot, List<AttributeKvEntry> data) {
    for (AttributeKvEntry entry : data) {
        if (entry.getValue() != null) {
            EntityKeyValue value = toEntityValue(entry);
            snapshot.putValue(new AlarmConditionFilterKey(AlarmConditionKeyType.ATTRIBUTE, entry.getKey()), entry.getLastUpdateTs(), value);
        }
    }
}