Igor Kulikov
6 years ago
45 changed files with 708 additions and 121 deletions
@ -0,0 +1,42 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2021 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.server.actors.ruleChain; |
||||
|
|
||||
|
import lombok.EqualsAndHashCode; |
||||
|
import lombok.Getter; |
||||
|
import org.thingsboard.rule.engine.api.TbContext; |
||||
|
import org.thingsboard.server.common.msg.TbActorStopReason; |
||||
|
import org.thingsboard.server.common.msg.TbMsg; |
||||
|
import org.thingsboard.server.common.msg.TbRuleEngineActorMsg; |
||||
|
import org.thingsboard.server.common.msg.queue.RuleNodeException; |
||||
|
|
||||
|
@EqualsAndHashCode(callSuper = true) |
||||
|
public abstract class TbToRuleNodeActorMsg extends TbRuleEngineActorMsg { |
||||
|
|
||||
|
@Getter |
||||
|
private final TbContext ctx; |
||||
|
|
||||
|
public TbToRuleNodeActorMsg(TbContext ctx, TbMsg tbMsg) { |
||||
|
super(tbMsg); |
||||
|
this.ctx = ctx; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void onTbActorStopped(TbActorStopReason reason) { |
||||
|
String message = reason == TbActorStopReason.STOPPED ? "Rule node stopped" : "Failed to initialize rule node!"; |
||||
|
msg.getCallback().onFailure(new RuleNodeException(message, ctx.getRuleChainName(), ctx.getSelf())); |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,22 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2021 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.server.common.msg; |
||||
|
|
||||
|
public enum TbActorStopReason { |
||||
|
|
||||
|
INIT_FAILED, STOPPED |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,25 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2021 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; |
||||
|
|
||||
|
public interface DynamicPredicateValueCtx { |
||||
|
|
||||
|
EntityKeyValue getTenantValue(String key); |
||||
|
|
||||
|
EntityKeyValue getCustomerValue(String key); |
||||
|
|
||||
|
void resetCustomer(); |
||||
|
} |
||||
@ -0,0 +1,74 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2021 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.extern.slf4j.Slf4j; |
||||
|
import org.thingsboard.rule.engine.api.TbContext; |
||||
|
import org.thingsboard.server.common.data.DataConstants; |
||||
|
import org.thingsboard.server.common.data.Device; |
||||
|
import org.thingsboard.server.common.data.id.CustomerId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.id.EntityId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.kv.AttributeKvEntry; |
||||
|
|
||||
|
import java.util.Optional; |
||||
|
import java.util.concurrent.ExecutionException; |
||||
|
|
||||
|
@Slf4j |
||||
|
public class DynamicPredicateValueCtxImpl implements DynamicPredicateValueCtx { |
||||
|
private final TenantId tenantId; |
||||
|
private CustomerId customerId; |
||||
|
private final DeviceId deviceId; |
||||
|
private final TbContext ctx; |
||||
|
|
||||
|
public DynamicPredicateValueCtxImpl(TenantId tenantId, DeviceId deviceId, TbContext ctx) { |
||||
|
this.tenantId = tenantId; |
||||
|
this.deviceId = deviceId; |
||||
|
this.ctx = ctx; |
||||
|
resetCustomer(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public EntityKeyValue getTenantValue(String key) { |
||||
|
return getValue(tenantId, key); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public EntityKeyValue getCustomerValue(String key) { |
||||
|
return customerId == null || customerId.isNullUid() ? null : getValue(customerId, key); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void resetCustomer() { |
||||
|
Device device = ctx.getDeviceService().findDeviceById(tenantId, deviceId); |
||||
|
if (device != null) { |
||||
|
this.customerId = device.getCustomerId(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private EntityKeyValue getValue(EntityId entityId, String key) { |
||||
|
try { |
||||
|
Optional<AttributeKvEntry> entry = ctx.getAttributesService().find(tenantId, entityId, DataConstants.SERVER_SCOPE, key).get(); |
||||
|
if (entry.isPresent()) { |
||||
|
return DeviceState.toEntityValue(entry.get()); |
||||
|
} |
||||
|
} catch (InterruptedException | ExecutionException e) { |
||||
|
log.warn("Failed to get attribute by key: {} for {}: [{}]", key, entityId.getEntityType(), entityId.getId()); |
||||
|
} |
||||
|
return null; |
||||
|
} |
||||
|
} |
||||
File diff suppressed because one or more lines are too long
Loading…
Reference in new issue