Andrii Shvaika
5 years ago
7 changed files with 520 additions and 240 deletions
@ -0,0 +1,252 @@ |
|||||
|
/** |
||||
|
* 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.Data; |
||||
|
import org.thingsboard.server.common.data.alarm.AlarmSeverity; |
||||
|
import org.thingsboard.server.common.data.device.profile.AlarmCondition; |
||||
|
import org.thingsboard.server.common.data.device.profile.AlarmRule; |
||||
|
import org.thingsboard.server.common.data.query.BooleanFilterPredicate; |
||||
|
import org.thingsboard.server.common.data.query.ComplexFilterPredicate; |
||||
|
import org.thingsboard.server.common.data.query.KeyFilter; |
||||
|
import org.thingsboard.server.common.data.query.KeyFilterPredicate; |
||||
|
import org.thingsboard.server.common.data.query.NumericFilterPredicate; |
||||
|
import org.thingsboard.server.common.data.query.StringFilterPredicate; |
||||
|
|
||||
|
@Data |
||||
|
public class AlarmRuleState { |
||||
|
|
||||
|
private final AlarmSeverity severity; |
||||
|
private final AlarmRule alarmRule; |
||||
|
private final long requiredDurationInMs; |
||||
|
private long lastEventTs; |
||||
|
private long duration; |
||||
|
|
||||
|
public AlarmRuleState(AlarmSeverity severity, AlarmRule alarmRule) { |
||||
|
this.severity = severity; |
||||
|
this.alarmRule = alarmRule; |
||||
|
if (alarmRule.getCondition().getDurationValue() > 0) { |
||||
|
requiredDurationInMs = alarmRule.getCondition().getDurationUnit().toMillis(alarmRule.getCondition().getDurationValue()); |
||||
|
} else { |
||||
|
requiredDurationInMs = 0; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public boolean eval(DeviceDataSnapshot data) { |
||||
|
if (requiredDurationInMs > 0) { |
||||
|
boolean eval = eval(alarmRule.getCondition(), data); |
||||
|
if (eval) { |
||||
|
if (lastEventTs > 0) { |
||||
|
if (data.getTs() > lastEventTs) { |
||||
|
duration += data.getTs() - lastEventTs; |
||||
|
lastEventTs = data.getTs(); |
||||
|
} |
||||
|
} else { |
||||
|
lastEventTs = data.getTs(); |
||||
|
duration = 0; |
||||
|
} |
||||
|
return duration > requiredDurationInMs; |
||||
|
} else { |
||||
|
lastEventTs = 0; |
||||
|
duration = 0; |
||||
|
return false; |
||||
|
} |
||||
|
} else { |
||||
|
return eval(alarmRule.getCondition(), data); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public boolean eval(long ts) { |
||||
|
if (requiredDurationInMs > 0 && lastEventTs > 0 && ts > lastEventTs) { |
||||
|
duration += ts - lastEventTs; |
||||
|
return duration > requiredDurationInMs; |
||||
|
} else { |
||||
|
return false; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private boolean eval(AlarmCondition condition, DeviceDataSnapshot data) { |
||||
|
boolean eval = true; |
||||
|
for (KeyFilter keyFilter : condition.getCondition()) { |
||||
|
EntityKeyValue value = data.getValue(keyFilter.getKey()); |
||||
|
if (value == null) { |
||||
|
return false; |
||||
|
} |
||||
|
eval = eval && eval(value, keyFilter.getPredicate()); |
||||
|
} |
||||
|
//TODO: use condition duration;
|
||||
|
return eval; |
||||
|
} |
||||
|
|
||||
|
private boolean eval(EntityKeyValue value, KeyFilterPredicate predicate) { |
||||
|
switch (predicate.getType()) { |
||||
|
case STRING: |
||||
|
return evalStrPredicate(value, (StringFilterPredicate) predicate); |
||||
|
case NUMERIC: |
||||
|
return evalNumPredicate(value, (NumericFilterPredicate) predicate); |
||||
|
case COMPLEX: |
||||
|
return evalComplexPredicate(value, (ComplexFilterPredicate) predicate); |
||||
|
case BOOLEAN: |
||||
|
return evalBoolPredicate(value, (BooleanFilterPredicate) predicate); |
||||
|
default: |
||||
|
return false; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private boolean evalComplexPredicate(EntityKeyValue ekv, ComplexFilterPredicate predicate) { |
||||
|
switch (predicate.getOperation()) { |
||||
|
case OR: |
||||
|
for (KeyFilterPredicate kfp : predicate.getPredicates()) { |
||||
|
if (eval(ekv, kfp)) { |
||||
|
return true; |
||||
|
} |
||||
|
} |
||||
|
return false; |
||||
|
case AND: |
||||
|
for (KeyFilterPredicate kfp : predicate.getPredicates()) { |
||||
|
if (!eval(ekv, kfp)) { |
||||
|
return false; |
||||
|
} |
||||
|
} |
||||
|
return true; |
||||
|
default: |
||||
|
throw new RuntimeException("Operation not supported: " + predicate.getOperation()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
|
||||
|
private boolean evalBoolPredicate(EntityKeyValue ekv, BooleanFilterPredicate predicate) { |
||||
|
Boolean value; |
||||
|
switch (ekv.getDataType()) { |
||||
|
case LONG: |
||||
|
value = ekv.getLngValue() > 0; |
||||
|
break; |
||||
|
case DOUBLE: |
||||
|
value = ekv.getDblValue() > 0; |
||||
|
break; |
||||
|
case BOOLEAN: |
||||
|
value = ekv.getBoolValue(); |
||||
|
break; |
||||
|
case STRING: |
||||
|
try { |
||||
|
value = Boolean.parseBoolean(ekv.getStrValue()); |
||||
|
break; |
||||
|
} catch (RuntimeException e) { |
||||
|
return false; |
||||
|
} |
||||
|
case JSON: |
||||
|
try { |
||||
|
value = Boolean.parseBoolean(ekv.getJsonValue()); |
||||
|
break; |
||||
|
} catch (RuntimeException e) { |
||||
|
return false; |
||||
|
} |
||||
|
default: |
||||
|
return false; |
||||
|
} |
||||
|
if (value == null) { |
||||
|
return false; |
||||
|
} |
||||
|
switch (predicate.getOperation()) { |
||||
|
case EQUAL: |
||||
|
return value.equals(predicate.getValue().getDefaultValue()); |
||||
|
case NOT_EQUAL: |
||||
|
return !value.equals(predicate.getValue().getDefaultValue()); |
||||
|
default: |
||||
|
throw new RuntimeException("Operation not supported: " + predicate.getOperation()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private boolean evalNumPredicate(EntityKeyValue ekv, NumericFilterPredicate predicate) { |
||||
|
Double value; |
||||
|
switch (ekv.getDataType()) { |
||||
|
case LONG: |
||||
|
value = ekv.getLngValue().doubleValue(); |
||||
|
break; |
||||
|
case DOUBLE: |
||||
|
value = ekv.getDblValue(); |
||||
|
break; |
||||
|
case BOOLEAN: |
||||
|
value = ekv.getBoolValue() ? 1.0 : 0.0; |
||||
|
break; |
||||
|
case STRING: |
||||
|
try { |
||||
|
value = Double.parseDouble(ekv.getStrValue()); |
||||
|
break; |
||||
|
} catch (RuntimeException e) { |
||||
|
return false; |
||||
|
} |
||||
|
case JSON: |
||||
|
try { |
||||
|
value = Double.parseDouble(ekv.getJsonValue()); |
||||
|
break; |
||||
|
} catch (RuntimeException e) { |
||||
|
return false; |
||||
|
} |
||||
|
default: |
||||
|
return false; |
||||
|
} |
||||
|
if (value == null) { |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
Double predicateValue = predicate.getValue().getDefaultValue(); |
||||
|
switch (predicate.getOperation()) { |
||||
|
case NOT_EQUAL: |
||||
|
return !value.equals(predicateValue); |
||||
|
case EQUAL: |
||||
|
return value.equals(predicateValue); |
||||
|
case GREATER: |
||||
|
return value > predicateValue; |
||||
|
case GREATER_OR_EQUAL: |
||||
|
return value >= predicateValue; |
||||
|
case LESS: |
||||
|
return value < predicateValue; |
||||
|
case LESS_OR_EQUAL: |
||||
|
return value <= predicateValue; |
||||
|
default: |
||||
|
throw new RuntimeException("Operation not supported: " + predicate.getOperation()); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private boolean evalStrPredicate(EntityKeyValue ekv, StringFilterPredicate predicate) { |
||||
|
String val; |
||||
|
String predicateValue; |
||||
|
if (predicate.isIgnoreCase()) { |
||||
|
val = ekv.getStrValue().toLowerCase(); |
||||
|
predicateValue = predicate.getValue().getDefaultValue().toLowerCase(); |
||||
|
} else { |
||||
|
val = ekv.getStrValue(); |
||||
|
predicateValue = predicate.getValue().getDefaultValue(); |
||||
|
} |
||||
|
switch (predicate.getOperation()) { |
||||
|
case CONTAINS: |
||||
|
return val.contains(predicateValue); |
||||
|
case EQUAL: |
||||
|
return val.equals(predicateValue); |
||||
|
case STARTS_WITH: |
||||
|
return val.startsWith(predicateValue); |
||||
|
case ENDS_WITH: |
||||
|
return val.endsWith(predicateValue); |
||||
|
case NOT_EQUAL: |
||||
|
return !val.equals(predicateValue); |
||||
|
case NOT_CONTAINS: |
||||
|
return !val.contains(predicateValue); |
||||
|
default: |
||||
|
throw new RuntimeException("Operation not supported: " + predicate.getOperation()); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue