Igor Kulikov
6 years ago
20 changed files with 519 additions and 237 deletions
@ -1,63 +0,0 @@ |
|||
/** |
|||
* Copyright © 2016-2020 The Thingsboard Authors |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package org.thingsboard.rule.engine.profile; |
|||
|
|||
import lombok.AccessLevel; |
|||
import lombok.Getter; |
|||
import org.thingsboard.server.common.data.DeviceProfile; |
|||
import org.thingsboard.server.common.data.device.profile.AlarmRule; |
|||
import org.thingsboard.server.common.data.device.profile.DeviceProfileAlarm; |
|||
import org.thingsboard.server.common.data.id.DeviceProfileId; |
|||
import org.thingsboard.server.common.data.query.EntityKey; |
|||
import org.thingsboard.server.common.data.query.KeyFilter; |
|||
|
|||
import java.util.List; |
|||
import java.util.Set; |
|||
import java.util.concurrent.ConcurrentHashMap; |
|||
import java.util.concurrent.CopyOnWriteArrayList; |
|||
|
|||
|
|||
class DeviceProfileState { |
|||
|
|||
private DeviceProfile deviceProfile; |
|||
@Getter(AccessLevel.PACKAGE) |
|||
private final List<DeviceProfileAlarm> alarmSettings = new CopyOnWriteArrayList<>(); |
|||
@Getter(AccessLevel.PACKAGE) |
|||
private final Set<EntityKey> entityKeys = ConcurrentHashMap.newKeySet(); |
|||
|
|||
DeviceProfileState(DeviceProfile deviceProfile) { |
|||
updateDeviceProfile(deviceProfile); |
|||
} |
|||
|
|||
void updateDeviceProfile(DeviceProfile deviceProfile) { |
|||
this.deviceProfile = deviceProfile; |
|||
alarmSettings.clear(); |
|||
if (deviceProfile.getProfileData().getAlarms() != null) { |
|||
alarmSettings.addAll(deviceProfile.getProfileData().getAlarms()); |
|||
for (DeviceProfileAlarm alarm : deviceProfile.getProfileData().getAlarms()) { |
|||
for (AlarmRule alarmRule : alarm.getCreateRules().values()) { |
|||
for (KeyFilter keyFilter : alarmRule.getCondition().getCondition()) { |
|||
entityKeys.add(keyFilter.getKey()); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
public DeviceProfileId getProfileId() { |
|||
return deviceProfile.getId(); |
|||
} |
|||
} |
|||
@ -0,0 +1,137 @@ |
|||
/** |
|||
* Copyright © 2016-2020 The Thingsboard Authors |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package org.thingsboard.rule.engine.profile; |
|||
|
|||
import lombok.AccessLevel; |
|||
import lombok.Getter; |
|||
import org.thingsboard.server.common.data.DeviceProfile; |
|||
import org.thingsboard.server.common.data.alarm.AlarmSeverity; |
|||
import org.thingsboard.server.common.data.device.profile.AlarmRule; |
|||
import org.thingsboard.server.common.data.device.profile.DeviceProfileAlarm; |
|||
import org.thingsboard.server.common.data.id.DeviceProfileId; |
|||
import org.thingsboard.server.common.data.query.ComplexFilterPredicate; |
|||
import org.thingsboard.server.common.data.query.DynamicValue; |
|||
import org.thingsboard.server.common.data.query.DynamicValueSourceType; |
|||
import org.thingsboard.server.common.data.query.EntityKey; |
|||
import org.thingsboard.server.common.data.query.EntityKeyType; |
|||
import org.thingsboard.server.common.data.query.FilterPredicateValue; |
|||
import org.thingsboard.server.common.data.query.KeyFilter; |
|||
import org.thingsboard.server.common.data.query.KeyFilterPredicate; |
|||
import org.thingsboard.server.common.data.query.SimpleKeyFilterPredicate; |
|||
import org.thingsboard.server.common.data.query.StringFilterPredicate; |
|||
|
|||
import javax.print.attribute.standard.Severity; |
|||
import java.util.Collections; |
|||
import java.util.HashMap; |
|||
import java.util.HashSet; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
import java.util.Set; |
|||
import java.util.concurrent.ConcurrentHashMap; |
|||
import java.util.concurrent.CopyOnWriteArrayList; |
|||
|
|||
|
|||
class ProfileState { |
|||
|
|||
private DeviceProfile deviceProfile; |
|||
@Getter(AccessLevel.PACKAGE) |
|||
private final List<DeviceProfileAlarm> alarmSettings = new CopyOnWriteArrayList<>(); |
|||
@Getter(AccessLevel.PACKAGE) |
|||
private final Set<EntityKey> entityKeys = ConcurrentHashMap.newKeySet(); |
|||
|
|||
private final Map<String, Map<AlarmSeverity, Set<EntityKey>>> alarmCreateKeys = new HashMap<>(); |
|||
private final Map<String, Set<EntityKey>> alarmClearKeys = new HashMap<>(); |
|||
|
|||
ProfileState(DeviceProfile deviceProfile) { |
|||
updateDeviceProfile(deviceProfile); |
|||
} |
|||
|
|||
void updateDeviceProfile(DeviceProfile deviceProfile) { |
|||
this.deviceProfile = deviceProfile; |
|||
alarmSettings.clear(); |
|||
alarmCreateKeys.clear(); |
|||
alarmClearKeys.clear(); |
|||
if (deviceProfile.getProfileData().getAlarms() != null) { |
|||
alarmSettings.addAll(deviceProfile.getProfileData().getAlarms()); |
|||
for (DeviceProfileAlarm alarm : deviceProfile.getProfileData().getAlarms()) { |
|||
Map<AlarmSeverity, Set<EntityKey>> createAlarmKeys = alarmCreateKeys.computeIfAbsent(alarm.getId(), id -> new HashMap<>()); |
|||
alarm.getCreateRules().forEach(((severity, alarmRule) -> { |
|||
Set<EntityKey> ruleKeys = createAlarmKeys.computeIfAbsent(severity, id -> new HashSet<>()); |
|||
for (KeyFilter keyFilter : alarmRule.getCondition().getCondition()) { |
|||
entityKeys.add(keyFilter.getKey()); |
|||
ruleKeys.add(keyFilter.getKey()); |
|||
addDynamicValuesRecursively(keyFilter.getPredicate(), entityKeys, ruleKeys); |
|||
} |
|||
})); |
|||
if (alarm.getClearRule() != null) { |
|||
Set<EntityKey> clearAlarmKeys = alarmClearKeys.computeIfAbsent(alarm.getId(), id -> new HashSet<>()); |
|||
for (KeyFilter keyFilter : alarm.getClearRule().getCondition().getCondition()) { |
|||
entityKeys.add(keyFilter.getKey()); |
|||
clearAlarmKeys.add(keyFilter.getKey()); |
|||
addDynamicValuesRecursively(keyFilter.getPredicate(), entityKeys, clearAlarmKeys); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
private void addDynamicValuesRecursively(KeyFilterPredicate predicate, Set<EntityKey> entityKeys, Set<EntityKey> ruleKeys) { |
|||
switch (predicate.getType()) { |
|||
case STRING: |
|||
case NUMERIC: |
|||
case BOOLEAN: |
|||
DynamicValue value = ((SimpleKeyFilterPredicate) predicate).getValue().getDynamicValue(); |
|||
if (value != null && value.getSourceType() == DynamicValueSourceType.CURRENT_DEVICE) { |
|||
EntityKey entityKey = new EntityKey(EntityKeyType.ATTRIBUTE, value.getSourceAttribute()); |
|||
entityKeys.add(entityKey); |
|||
ruleKeys.add(entityKey); |
|||
} |
|||
break; |
|||
case COMPLEX: |
|||
for (KeyFilterPredicate child : ((ComplexFilterPredicate) predicate).getPredicates()) { |
|||
addDynamicValuesRecursively(child, entityKeys, ruleKeys); |
|||
} |
|||
break; |
|||
} |
|||
} |
|||
|
|||
DeviceProfileId getProfileId() { |
|||
return deviceProfile.getId(); |
|||
} |
|||
|
|||
Set<EntityKey> getCreateAlarmKeys(String id, AlarmSeverity severity) { |
|||
Map<AlarmSeverity, Set<EntityKey>> sKeys = alarmCreateKeys.get(id); |
|||
if (sKeys == null) { |
|||
return Collections.emptySet(); |
|||
} else { |
|||
Set<EntityKey> keys = sKeys.get(severity); |
|||
if (keys == null) { |
|||
return Collections.emptySet(); |
|||
} else { |
|||
return keys; |
|||
} |
|||
} |
|||
} |
|||
|
|||
Set<EntityKey> getClearAlarmKeys(String id) { |
|||
Set<EntityKey> keys = alarmClearKeys.get(id); |
|||
if (keys == null) { |
|||
return Collections.emptySet(); |
|||
} else { |
|||
return keys; |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue