Search in sources :

Example 1 with BooleanFilterPredicate

use of org.thingsboard.server.common.data.query.BooleanFilterPredicate in project thingsboard by thingsboard.

the class TbDeviceProfileNodeTest method testConstantKeyFilterInherited.

@Test
public void testConstantKeyFilterInherited() 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, tenantId.getId(), "SERVER_SCOPE", "alarmEnabled");
    AttributeKvEntity attributeKvEntity = new AttributeKvEntity();
    attributeKvEntity.setId(compositeKey);
    attributeKvEntity.setBooleanValue(Boolean.TRUE);
    attributeKvEntity.setLastUpdateTs(System.currentTimeMillis());
    AttributeKvEntry entry = attributeKvEntity.toData();
    ListenableFuture<Optional<AttributeKvEntry>> attrListListenableFuture = Futures.immediateFuture(Optional.of(entry));
    AlarmConditionFilter alarmEnabledFilter = new AlarmConditionFilter();
    alarmEnabledFilter.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.CONSTANT, "alarmEnabled"));
    alarmEnabledFilter.setValue(Boolean.TRUE);
    alarmEnabledFilter.setValueType(EntityKeyValueType.BOOLEAN);
    BooleanFilterPredicate alarmEnabledPredicate = new BooleanFilterPredicate();
    alarmEnabledPredicate.setOperation(BooleanFilterPredicate.BooleanOperation.EQUAL);
    alarmEnabledPredicate.setValue(new FilterPredicateValue<>(Boolean.FALSE, null, new DynamicValue<>(DynamicValueSourceType.CURRENT_DEVICE, "alarmEnabled", true)));
    alarmEnabledFilter.setPredicate(alarmEnabledPredicate);
    AlarmConditionFilter temperatureFilter = new AlarmConditionFilter();
    temperatureFilter.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.TIME_SERIES, "temperature"));
    temperatureFilter.setValueType(EntityKeyValueType.NUMERIC);
    NumericFilterPredicate temperaturePredicate = new NumericFilterPredicate();
    temperaturePredicate.setOperation(NumericFilterPredicate.NumericOperation.GREATER);
    temperaturePredicate.setValue(new FilterPredicateValue<>(20.0, null, null));
    temperatureFilter.setPredicate(temperaturePredicate);
    AlarmCondition alarmCondition = new AlarmCondition();
    alarmCondition.setCondition(Arrays.asList(alarmEnabledFilter, temperatureFilter));
    AlarmRule alarmRule = new AlarmRule();
    alarmRule.setCondition(alarmCondition);
    DeviceProfileAlarm dpa = new DeviceProfileAlarm();
    dpa.setId("alarmEnabledAlarmID");
    dpa.setAlarmType("alarmEnabledAlarm");
    dpa.setCreateRules(new TreeMap<>(Collections.singletonMap(AlarmSeverity.CRITICAL, alarmRule)));
    deviceProfileData.setAlarms(Collections.singletonList(dpa));
    deviceProfile.setProfileData(deviceProfileData);
    Mockito.when(deviceService.findDeviceById(tenantId, deviceId)).thenReturn(device);
    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, "alarmEnabledAlarm")).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(Futures.immediateFuture(Collections.emptyList()));
    Mockito.when(attributesService.find(eq(tenantId), eq(customerId), Mockito.anyString(), Mockito.anyString())).thenReturn(Futures.immediateFuture(Optional.empty()));
    Mockito.when(attributesService.find(eq(tenantId), eq(tenantId), Mockito.anyString(), Mockito.anyString())).thenReturn(attrListListenableFuture);
    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", 21);
    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());
}
Also used : BooleanFilterPredicate(org.thingsboard.server.common.data.query.BooleanFilterPredicate) AttributeKvEntry(org.thingsboard.server.common.data.kv.AttributeKvEntry) Optional(java.util.Optional) ObjectNode(com.fasterxml.jackson.databind.node.ObjectNode) AttributeKvEntity(org.thingsboard.server.dao.model.sql.AttributeKvEntity) NumericFilterPredicate(org.thingsboard.server.common.data.query.NumericFilterPredicate) Device(org.thingsboard.server.common.data.Device) AlarmRule(org.thingsboard.server.common.data.device.profile.AlarmRule) TbMsgMetaData(org.thingsboard.server.common.msg.TbMsgMetaData) AlarmConditionFilter(org.thingsboard.server.common.data.device.profile.AlarmConditionFilter) DeviceProfile(org.thingsboard.server.common.data.DeviceProfile) AlarmCondition(org.thingsboard.server.common.data.device.profile.AlarmCondition) DeviceProfileAlarm(org.thingsboard.server.common.data.device.profile.DeviceProfileAlarm) 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) DynamicValue(org.thingsboard.server.common.data.query.DynamicValue) TbMsg(org.thingsboard.server.common.msg.TbMsg) Test(org.junit.Test)

Example 2 with BooleanFilterPredicate

use of org.thingsboard.server.common.data.query.BooleanFilterPredicate in project thingsboard by thingsboard.

the class DefaultSystemDataLoaderService method loadDemoData.

@Override
public void loadDemoData() throws Exception {
    Tenant demoTenant = new Tenant();
    demoTenant.setRegion("Global");
    demoTenant.setTitle("Tenant");
    demoTenant = tenantService.saveTenant(demoTenant);
    installScripts.loadDemoRuleChains(demoTenant.getId());
    createUser(Authority.TENANT_ADMIN, demoTenant.getId(), null, "tenant@thingsboard.org", "tenant");
    Customer customerA = new Customer();
    customerA.setTenantId(demoTenant.getId());
    customerA.setTitle("Customer A");
    customerA = customerService.saveCustomer(customerA);
    Customer customerB = new Customer();
    customerB.setTenantId(demoTenant.getId());
    customerB.setTitle("Customer B");
    customerB = customerService.saveCustomer(customerB);
    Customer customerC = new Customer();
    customerC.setTenantId(demoTenant.getId());
    customerC.setTitle("Customer C");
    customerC = customerService.saveCustomer(customerC);
    createUser(Authority.CUSTOMER_USER, demoTenant.getId(), customerA.getId(), "customer@thingsboard.org", CUSTOMER_CRED);
    createUser(Authority.CUSTOMER_USER, demoTenant.getId(), customerA.getId(), "customerA@thingsboard.org", CUSTOMER_CRED);
    createUser(Authority.CUSTOMER_USER, demoTenant.getId(), customerB.getId(), "customerB@thingsboard.org", CUSTOMER_CRED);
    createUser(Authority.CUSTOMER_USER, demoTenant.getId(), customerC.getId(), "customerC@thingsboard.org", CUSTOMER_CRED);
    DeviceProfile defaultDeviceProfile = this.deviceProfileService.findOrCreateDeviceProfile(demoTenant.getId(), DEFAULT_DEVICE_TYPE);
    createDevice(demoTenant.getId(), customerA.getId(), defaultDeviceProfile.getId(), "Test Device A1", "A1_TEST_TOKEN", null);
    createDevice(demoTenant.getId(), customerA.getId(), defaultDeviceProfile.getId(), "Test Device A2", "A2_TEST_TOKEN", null);
    createDevice(demoTenant.getId(), customerA.getId(), defaultDeviceProfile.getId(), "Test Device A3", "A3_TEST_TOKEN", null);
    createDevice(demoTenant.getId(), customerB.getId(), defaultDeviceProfile.getId(), "Test Device B1", "B1_TEST_TOKEN", null);
    createDevice(demoTenant.getId(), customerC.getId(), defaultDeviceProfile.getId(), "Test Device C1", "C1_TEST_TOKEN", null);
    createDevice(demoTenant.getId(), null, defaultDeviceProfile.getId(), "DHT11 Demo Device", "DHT11_DEMO_TOKEN", "Demo device that is used in sample " + "applications that upload data from DHT11 temperature and humidity sensor");
    createDevice(demoTenant.getId(), null, defaultDeviceProfile.getId(), "Raspberry Pi Demo Device", "RASPBERRY_PI_DEMO_TOKEN", "Demo device that is used in " + "Raspberry Pi GPIO control sample application");
    DeviceProfile thermostatDeviceProfile = new DeviceProfile();
    thermostatDeviceProfile.setTenantId(demoTenant.getId());
    thermostatDeviceProfile.setDefault(false);
    thermostatDeviceProfile.setName("thermostat");
    thermostatDeviceProfile.setType(DeviceProfileType.DEFAULT);
    thermostatDeviceProfile.setTransportType(DeviceTransportType.DEFAULT);
    thermostatDeviceProfile.setProvisionType(DeviceProfileProvisionType.DISABLED);
    thermostatDeviceProfile.setDescription("Thermostat device profile");
    thermostatDeviceProfile.setDefaultRuleChainId(ruleChainService.findTenantRuleChainsByType(demoTenant.getId(), RuleChainType.CORE, new PageLink(1, 0, "Thermostat")).getData().get(0).getId());
    DeviceProfileData deviceProfileData = new DeviceProfileData();
    DefaultDeviceProfileConfiguration configuration = new DefaultDeviceProfileConfiguration();
    DefaultDeviceProfileTransportConfiguration transportConfiguration = new DefaultDeviceProfileTransportConfiguration();
    DisabledDeviceProfileProvisionConfiguration provisionConfiguration = new DisabledDeviceProfileProvisionConfiguration(null);
    deviceProfileData.setConfiguration(configuration);
    deviceProfileData.setTransportConfiguration(transportConfiguration);
    deviceProfileData.setProvisionConfiguration(provisionConfiguration);
    thermostatDeviceProfile.setProfileData(deviceProfileData);
    DeviceProfileAlarm highTemperature = new DeviceProfileAlarm();
    highTemperature.setId("highTemperatureAlarmID");
    highTemperature.setAlarmType("High Temperature");
    AlarmRule temperatureRule = new AlarmRule();
    AlarmCondition temperatureCondition = new AlarmCondition();
    temperatureCondition.setSpec(new SimpleAlarmConditionSpec());
    AlarmConditionFilter temperatureAlarmFlagAttributeFilter = new AlarmConditionFilter();
    temperatureAlarmFlagAttributeFilter.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.ATTRIBUTE, "temperatureAlarmFlag"));
    temperatureAlarmFlagAttributeFilter.setValueType(EntityKeyValueType.BOOLEAN);
    BooleanFilterPredicate temperatureAlarmFlagAttributePredicate = new BooleanFilterPredicate();
    temperatureAlarmFlagAttributePredicate.setOperation(BooleanFilterPredicate.BooleanOperation.EQUAL);
    temperatureAlarmFlagAttributePredicate.setValue(new FilterPredicateValue<>(Boolean.TRUE));
    temperatureAlarmFlagAttributeFilter.setPredicate(temperatureAlarmFlagAttributePredicate);
    AlarmConditionFilter temperatureTimeseriesFilter = new AlarmConditionFilter();
    temperatureTimeseriesFilter.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.TIME_SERIES, "temperature"));
    temperatureTimeseriesFilter.setValueType(EntityKeyValueType.NUMERIC);
    NumericFilterPredicate temperatureTimeseriesFilterPredicate = new NumericFilterPredicate();
    temperatureTimeseriesFilterPredicate.setOperation(NumericFilterPredicate.NumericOperation.GREATER);
    FilterPredicateValue<Double> temperatureTimeseriesPredicateValue = new FilterPredicateValue<>(25.0, null, new DynamicValue<>(DynamicValueSourceType.CURRENT_DEVICE, "temperatureAlarmThreshold"));
    temperatureTimeseriesFilterPredicate.setValue(temperatureTimeseriesPredicateValue);
    temperatureTimeseriesFilter.setPredicate(temperatureTimeseriesFilterPredicate);
    temperatureCondition.setCondition(Arrays.asList(temperatureAlarmFlagAttributeFilter, temperatureTimeseriesFilter));
    temperatureRule.setAlarmDetails("Current temperature = ${temperature}");
    temperatureRule.setCondition(temperatureCondition);
    highTemperature.setCreateRules(new TreeMap<>(Collections.singletonMap(AlarmSeverity.MAJOR, temperatureRule)));
    AlarmRule clearTemperatureRule = new AlarmRule();
    AlarmCondition clearTemperatureCondition = new AlarmCondition();
    clearTemperatureCondition.setSpec(new SimpleAlarmConditionSpec());
    AlarmConditionFilter clearTemperatureTimeseriesFilter = new AlarmConditionFilter();
    clearTemperatureTimeseriesFilter.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.TIME_SERIES, "temperature"));
    clearTemperatureTimeseriesFilter.setValueType(EntityKeyValueType.NUMERIC);
    NumericFilterPredicate clearTemperatureTimeseriesFilterPredicate = new NumericFilterPredicate();
    clearTemperatureTimeseriesFilterPredicate.setOperation(NumericFilterPredicate.NumericOperation.LESS_OR_EQUAL);
    FilterPredicateValue<Double> clearTemperatureTimeseriesPredicateValue = new FilterPredicateValue<>(25.0, null, new DynamicValue<>(DynamicValueSourceType.CURRENT_DEVICE, "temperatureAlarmThreshold"));
    clearTemperatureTimeseriesFilterPredicate.setValue(clearTemperatureTimeseriesPredicateValue);
    clearTemperatureTimeseriesFilter.setPredicate(clearTemperatureTimeseriesFilterPredicate);
    clearTemperatureCondition.setCondition(Collections.singletonList(clearTemperatureTimeseriesFilter));
    clearTemperatureRule.setCondition(clearTemperatureCondition);
    clearTemperatureRule.setAlarmDetails("Current temperature = ${temperature}");
    highTemperature.setClearRule(clearTemperatureRule);
    DeviceProfileAlarm lowHumidity = new DeviceProfileAlarm();
    lowHumidity.setId("lowHumidityAlarmID");
    lowHumidity.setAlarmType("Low Humidity");
    AlarmRule humidityRule = new AlarmRule();
    AlarmCondition humidityCondition = new AlarmCondition();
    humidityCondition.setSpec(new SimpleAlarmConditionSpec());
    AlarmConditionFilter humidityAlarmFlagAttributeFilter = new AlarmConditionFilter();
    humidityAlarmFlagAttributeFilter.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.ATTRIBUTE, "humidityAlarmFlag"));
    humidityAlarmFlagAttributeFilter.setValueType(EntityKeyValueType.BOOLEAN);
    BooleanFilterPredicate humidityAlarmFlagAttributePredicate = new BooleanFilterPredicate();
    humidityAlarmFlagAttributePredicate.setOperation(BooleanFilterPredicate.BooleanOperation.EQUAL);
    humidityAlarmFlagAttributePredicate.setValue(new FilterPredicateValue<>(Boolean.TRUE));
    humidityAlarmFlagAttributeFilter.setPredicate(humidityAlarmFlagAttributePredicate);
    AlarmConditionFilter humidityTimeseriesFilter = new AlarmConditionFilter();
    humidityTimeseriesFilter.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.TIME_SERIES, "humidity"));
    humidityTimeseriesFilter.setValueType(EntityKeyValueType.NUMERIC);
    NumericFilterPredicate humidityTimeseriesFilterPredicate = new NumericFilterPredicate();
    humidityTimeseriesFilterPredicate.setOperation(NumericFilterPredicate.NumericOperation.LESS);
    FilterPredicateValue<Double> humidityTimeseriesPredicateValue = new FilterPredicateValue<>(60.0, null, new DynamicValue<>(DynamicValueSourceType.CURRENT_DEVICE, "humidityAlarmThreshold"));
    humidityTimeseriesFilterPredicate.setValue(humidityTimeseriesPredicateValue);
    humidityTimeseriesFilter.setPredicate(humidityTimeseriesFilterPredicate);
    humidityCondition.setCondition(Arrays.asList(humidityAlarmFlagAttributeFilter, humidityTimeseriesFilter));
    humidityRule.setCondition(humidityCondition);
    humidityRule.setAlarmDetails("Current humidity = ${humidity}");
    lowHumidity.setCreateRules(new TreeMap<>(Collections.singletonMap(AlarmSeverity.MINOR, humidityRule)));
    AlarmRule clearHumidityRule = new AlarmRule();
    AlarmCondition clearHumidityCondition = new AlarmCondition();
    clearHumidityCondition.setSpec(new SimpleAlarmConditionSpec());
    AlarmConditionFilter clearHumidityTimeseriesFilter = new AlarmConditionFilter();
    clearHumidityTimeseriesFilter.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.TIME_SERIES, "humidity"));
    clearHumidityTimeseriesFilter.setValueType(EntityKeyValueType.NUMERIC);
    NumericFilterPredicate clearHumidityTimeseriesFilterPredicate = new NumericFilterPredicate();
    clearHumidityTimeseriesFilterPredicate.setOperation(NumericFilterPredicate.NumericOperation.GREATER_OR_EQUAL);
    FilterPredicateValue<Double> clearHumidityTimeseriesPredicateValue = new FilterPredicateValue<>(60.0, null, new DynamicValue<>(DynamicValueSourceType.CURRENT_DEVICE, "humidityAlarmThreshold"));
    clearHumidityTimeseriesFilterPredicate.setValue(clearHumidityTimeseriesPredicateValue);
    clearHumidityTimeseriesFilter.setPredicate(clearHumidityTimeseriesFilterPredicate);
    clearHumidityCondition.setCondition(Collections.singletonList(clearHumidityTimeseriesFilter));
    clearHumidityRule.setCondition(clearHumidityCondition);
    clearHumidityRule.setAlarmDetails("Current humidity = ${humidity}");
    lowHumidity.setClearRule(clearHumidityRule);
    deviceProfileData.setAlarms(Arrays.asList(highTemperature, lowHumidity));
    DeviceProfile savedThermostatDeviceProfile = deviceProfileService.saveDeviceProfile(thermostatDeviceProfile);
    DeviceId t1Id = createDevice(demoTenant.getId(), null, savedThermostatDeviceProfile.getId(), "Thermostat T1", "T1_TEST_TOKEN", "Demo device for Thermostats dashboard").getId();
    DeviceId t2Id = createDevice(demoTenant.getId(), null, savedThermostatDeviceProfile.getId(), "Thermostat T2", "T2_TEST_TOKEN", "Demo device for Thermostats dashboard").getId();
    attributesService.save(demoTenant.getId(), t1Id, DataConstants.SERVER_SCOPE, Arrays.asList(new BaseAttributeKvEntry(System.currentTimeMillis(), new DoubleDataEntry("latitude", 37.3948)), new BaseAttributeKvEntry(System.currentTimeMillis(), new DoubleDataEntry("longitude", -122.1503)), new BaseAttributeKvEntry(System.currentTimeMillis(), new BooleanDataEntry("temperatureAlarmFlag", true)), new BaseAttributeKvEntry(System.currentTimeMillis(), new BooleanDataEntry("humidityAlarmFlag", true)), new BaseAttributeKvEntry(System.currentTimeMillis(), new LongDataEntry("temperatureAlarmThreshold", (long) 20)), new BaseAttributeKvEntry(System.currentTimeMillis(), new LongDataEntry("humidityAlarmThreshold", (long) 50))));
    attributesService.save(demoTenant.getId(), t2Id, DataConstants.SERVER_SCOPE, Arrays.asList(new BaseAttributeKvEntry(System.currentTimeMillis(), new DoubleDataEntry("latitude", 37.493801)), new BaseAttributeKvEntry(System.currentTimeMillis(), new DoubleDataEntry("longitude", -121.948769)), new BaseAttributeKvEntry(System.currentTimeMillis(), new BooleanDataEntry("temperatureAlarmFlag", true)), new BaseAttributeKvEntry(System.currentTimeMillis(), new BooleanDataEntry("humidityAlarmFlag", true)), new BaseAttributeKvEntry(System.currentTimeMillis(), new LongDataEntry("temperatureAlarmThreshold", (long) 25)), new BaseAttributeKvEntry(System.currentTimeMillis(), new LongDataEntry("humidityAlarmThreshold", (long) 30))));
    installScripts.loadDashboards(demoTenant.getId(), null);
}
Also used : BooleanFilterPredicate(org.thingsboard.server.common.data.query.BooleanFilterPredicate) Customer(org.thingsboard.server.common.data.Customer) AlarmRule(org.thingsboard.server.common.data.device.profile.AlarmRule) DeviceId(org.thingsboard.server.common.data.id.DeviceId) FilterPredicateValue(org.thingsboard.server.common.data.query.FilterPredicateValue) DisabledDeviceProfileProvisionConfiguration(org.thingsboard.server.common.data.device.profile.DisabledDeviceProfileProvisionConfiguration) DeviceProfile(org.thingsboard.server.common.data.DeviceProfile) Tenant(org.thingsboard.server.common.data.Tenant) DoubleDataEntry(org.thingsboard.server.common.data.kv.DoubleDataEntry) BooleanDataEntry(org.thingsboard.server.common.data.kv.BooleanDataEntry) PageLink(org.thingsboard.server.common.data.page.PageLink) BaseAttributeKvEntry(org.thingsboard.server.common.data.kv.BaseAttributeKvEntry) DeviceProfileData(org.thingsboard.server.common.data.device.profile.DeviceProfileData) AlarmConditionFilterKey(org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey) NumericFilterPredicate(org.thingsboard.server.common.data.query.NumericFilterPredicate) SimpleAlarmConditionSpec(org.thingsboard.server.common.data.device.profile.SimpleAlarmConditionSpec) AlarmConditionFilter(org.thingsboard.server.common.data.device.profile.AlarmConditionFilter) DefaultDeviceProfileConfiguration(org.thingsboard.server.common.data.device.profile.DefaultDeviceProfileConfiguration) AlarmCondition(org.thingsboard.server.common.data.device.profile.AlarmCondition) DefaultDeviceProfileTransportConfiguration(org.thingsboard.server.common.data.device.profile.DefaultDeviceProfileTransportConfiguration) DeviceProfileAlarm(org.thingsboard.server.common.data.device.profile.DeviceProfileAlarm) LongDataEntry(org.thingsboard.server.common.data.kv.LongDataEntry)

Example 3 with BooleanFilterPredicate

use of org.thingsboard.server.common.data.query.BooleanFilterPredicate in project thingsboard by thingsboard.

the class TbDeviceProfileNodeTest method testConstantKeyFilterSimple.

@Test
public void testConstantKeyFilterSimple() 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", "alarmEnabled");
    AttributeKvEntity attributeKvEntity = new AttributeKvEntity();
    attributeKvEntity.setId(compositeKey);
    attributeKvEntity.setBooleanValue(Boolean.TRUE);
    attributeKvEntity.setLastUpdateTs(System.currentTimeMillis());
    AttributeKvEntry entry = attributeKvEntity.toData();
    ListenableFuture<List<AttributeKvEntry>> attrListListenableFuture = Futures.immediateFuture(Collections.singletonList(entry));
    AlarmConditionFilter alarmEnabledFilter = new AlarmConditionFilter();
    alarmEnabledFilter.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.CONSTANT, "alarmEnabled"));
    alarmEnabledFilter.setValue(Boolean.TRUE);
    alarmEnabledFilter.setValueType(EntityKeyValueType.BOOLEAN);
    BooleanFilterPredicate alarmEnabledPredicate = new BooleanFilterPredicate();
    alarmEnabledPredicate.setOperation(BooleanFilterPredicate.BooleanOperation.EQUAL);
    alarmEnabledPredicate.setValue(new FilterPredicateValue<>(Boolean.FALSE, null, new DynamicValue<>(DynamicValueSourceType.CURRENT_DEVICE, "alarmEnabled")));
    alarmEnabledFilter.setPredicate(alarmEnabledPredicate);
    AlarmConditionFilter temperatureFilter = new AlarmConditionFilter();
    temperatureFilter.setKey(new AlarmConditionFilterKey(AlarmConditionKeyType.TIME_SERIES, "temperature"));
    temperatureFilter.setValueType(EntityKeyValueType.NUMERIC);
    NumericFilterPredicate temperaturePredicate = new NumericFilterPredicate();
    temperaturePredicate.setOperation(NumericFilterPredicate.NumericOperation.GREATER);
    temperaturePredicate.setValue(new FilterPredicateValue<>(20.0, null, null));
    temperatureFilter.setPredicate(temperaturePredicate);
    AlarmCondition alarmCondition = new AlarmCondition();
    alarmCondition.setCondition(Arrays.asList(alarmEnabledFilter, temperatureFilter));
    AlarmRule alarmRule = new AlarmRule();
    alarmRule.setCondition(alarmCondition);
    DeviceProfileAlarm dpa = new DeviceProfileAlarm();
    dpa.setId("alarmEnabledAlarmID");
    dpa.setAlarmType("alarmEnabledAlarm");
    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, "alarmEnabledAlarm")).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(attrListListenableFuture);
    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", 21);
    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());
}
Also used : BooleanFilterPredicate(org.thingsboard.server.common.data.query.BooleanFilterPredicate) AttributeKvEntry(org.thingsboard.server.common.data.kv.AttributeKvEntry) ObjectNode(com.fasterxml.jackson.databind.node.ObjectNode) AttributeKvEntity(org.thingsboard.server.dao.model.sql.AttributeKvEntity) NumericFilterPredicate(org.thingsboard.server.common.data.query.NumericFilterPredicate) Device(org.thingsboard.server.common.data.Device) AlarmRule(org.thingsboard.server.common.data.device.profile.AlarmRule) TbMsgMetaData(org.thingsboard.server.common.msg.TbMsgMetaData) AlarmConditionFilter(org.thingsboard.server.common.data.device.profile.AlarmConditionFilter) DeviceProfile(org.thingsboard.server.common.data.DeviceProfile) AlarmCondition(org.thingsboard.server.common.data.device.profile.AlarmCondition) DeviceProfileAlarm(org.thingsboard.server.common.data.device.profile.DeviceProfileAlarm) 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) DynamicValue(org.thingsboard.server.common.data.query.DynamicValue) TbMsg(org.thingsboard.server.common.msg.TbMsg) Test(org.junit.Test)

Aggregations

DeviceProfile (org.thingsboard.server.common.data.DeviceProfile)3 AlarmCondition (org.thingsboard.server.common.data.device.profile.AlarmCondition)3 AlarmConditionFilter (org.thingsboard.server.common.data.device.profile.AlarmConditionFilter)3 AlarmConditionFilterKey (org.thingsboard.server.common.data.device.profile.AlarmConditionFilterKey)3 AlarmRule (org.thingsboard.server.common.data.device.profile.AlarmRule)3 DeviceProfileAlarm (org.thingsboard.server.common.data.device.profile.DeviceProfileAlarm)3 DeviceProfileData (org.thingsboard.server.common.data.device.profile.DeviceProfileData)3 BooleanFilterPredicate (org.thingsboard.server.common.data.query.BooleanFilterPredicate)3 NumericFilterPredicate (org.thingsboard.server.common.data.query.NumericFilterPredicate)3 ObjectNode (com.fasterxml.jackson.databind.node.ObjectNode)2 Test (org.junit.Test)2 Device (org.thingsboard.server.common.data.Device)2 AttributeKvEntry (org.thingsboard.server.common.data.kv.AttributeKvEntry)2 DynamicValue (org.thingsboard.server.common.data.query.DynamicValue)2 TbMsg (org.thingsboard.server.common.msg.TbMsg)2 TbMsgMetaData (org.thingsboard.server.common.msg.TbMsgMetaData)2 AttributeKvCompositeKey (org.thingsboard.server.dao.model.sql.AttributeKvCompositeKey)2 AttributeKvEntity (org.thingsboard.server.dao.model.sql.AttributeKvEntity)2 List (java.util.List)1 Optional (java.util.Optional)1