|
|
|
@ -1,12 +1,12 @@ |
|
|
|
/** |
|
|
|
* Copyright © 2016-2018 The Thingsboard Authors |
|
|
|
* |
|
|
|
* <p> |
|
|
|
* 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
|
|
|
|
* |
|
|
|
* <p> |
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
* <p> |
|
|
|
* 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. |
|
|
|
@ -17,77 +17,127 @@ package org.thingsboard.server.actors.ruleChain; |
|
|
|
|
|
|
|
import akka.actor.ActorContext; |
|
|
|
import akka.actor.ActorRef; |
|
|
|
import akka.actor.Props; |
|
|
|
import akka.event.LoggingAdapter; |
|
|
|
import com.fasterxml.jackson.core.JsonProcessingException; |
|
|
|
import org.thingsboard.server.actors.ActorSystemContext; |
|
|
|
import org.thingsboard.server.actors.plugin.*; |
|
|
|
import org.thingsboard.server.actors.service.DefaultActorService; |
|
|
|
import org.thingsboard.server.actors.shared.ComponentMsgProcessor; |
|
|
|
import org.thingsboard.server.common.data.id.PluginId; |
|
|
|
import org.thingsboard.server.common.data.EntityType; |
|
|
|
import org.thingsboard.server.common.data.id.EntityId; |
|
|
|
import org.thingsboard.server.common.data.id.RuleChainId; |
|
|
|
import org.thingsboard.server.common.data.id.RuleNodeId; |
|
|
|
import org.thingsboard.server.common.data.id.TenantId; |
|
|
|
import org.thingsboard.server.common.data.plugin.ComponentLifecycleState; |
|
|
|
import org.thingsboard.server.common.data.plugin.ComponentType; |
|
|
|
import org.thingsboard.server.common.data.plugin.PluginMetaData; |
|
|
|
import org.thingsboard.server.common.data.relation.EntityRelation; |
|
|
|
import org.thingsboard.server.common.data.rule.RuleChain; |
|
|
|
import org.thingsboard.server.common.data.rule.RuleNode; |
|
|
|
import org.thingsboard.server.common.msg.TbMsg; |
|
|
|
import org.thingsboard.server.common.msg.cluster.ClusterEventMsg; |
|
|
|
import org.thingsboard.server.common.msg.cluster.ServerAddress; |
|
|
|
import org.thingsboard.server.extensions.api.plugins.Plugin; |
|
|
|
import org.thingsboard.server.extensions.api.plugins.PluginInitializationException; |
|
|
|
import org.thingsboard.server.extensions.api.plugins.msg.FromDeviceRpcResponse; |
|
|
|
import org.thingsboard.server.extensions.api.plugins.msg.TimeoutMsg; |
|
|
|
import org.thingsboard.server.extensions.api.plugins.rest.PluginRestMsg; |
|
|
|
import org.thingsboard.server.extensions.api.plugins.rpc.PluginRpcMsg; |
|
|
|
import org.thingsboard.server.extensions.api.plugins.ws.msg.PluginWebsocketMsg; |
|
|
|
import org.thingsboard.server.extensions.api.rules.RuleException; |
|
|
|
import org.thingsboard.server.common.msg.system.ServiceToRuleEngineMsg; |
|
|
|
import org.thingsboard.server.dao.rule.RuleChainService; |
|
|
|
|
|
|
|
import java.util.ArrayList; |
|
|
|
import java.util.HashMap; |
|
|
|
import java.util.List; |
|
|
|
import java.util.Map; |
|
|
|
|
|
|
|
/** |
|
|
|
* @author Andrew Shvayka |
|
|
|
*/ |
|
|
|
public class RuleChainActorMessageProcessor extends ComponentMsgProcessor<RuleChainId> { |
|
|
|
|
|
|
|
private ComponentLifecycleState state; |
|
|
|
private final ActorRef parent; |
|
|
|
private final ActorRef self; |
|
|
|
private final Map<RuleNodeId, RuleNodeCtx> nodeActors; |
|
|
|
private final Map<RuleNodeId, List<RuleNodeRelation>> nodeRoutes; |
|
|
|
private final RuleChainService service; |
|
|
|
|
|
|
|
private RuleNodeId firstId; |
|
|
|
private RuleNodeCtx firstNode; |
|
|
|
|
|
|
|
protected RuleChainActorMessageProcessor(TenantId tenantId, RuleChainId pluginId, ActorSystemContext systemContext |
|
|
|
RuleChainActorMessageProcessor(TenantId tenantId, RuleChainId ruleChainId, ActorSystemContext systemContext |
|
|
|
, LoggingAdapter logger, ActorRef parent, ActorRef self) { |
|
|
|
super(systemContext, logger, tenantId, pluginId); |
|
|
|
super(systemContext, logger, tenantId, ruleChainId); |
|
|
|
this.parent = parent; |
|
|
|
this.self = self; |
|
|
|
this.nodeActors = new HashMap<>(); |
|
|
|
this.nodeRoutes = new HashMap<>(); |
|
|
|
this.service = systemContext.getRuleChainService(); |
|
|
|
} |
|
|
|
|
|
|
|
@Override |
|
|
|
public void start() throws Exception { |
|
|
|
|
|
|
|
public void start(ActorContext context) throws Exception { |
|
|
|
RuleChain ruleChain = service.findRuleChainById(entityId); |
|
|
|
List<RuleNode> ruleNodeList = service.getRuleChainNodes(entityId); |
|
|
|
// Creating and starting the actors;
|
|
|
|
for (RuleNode ruleNode : ruleNodeList) { |
|
|
|
String dispatcherName = tenantId.getId().equals(EntityId.NULL_UUID) ? |
|
|
|
DefaultActorService.SYSTEM_RULE_DISPATCHER_NAME : DefaultActorService.TENANT_RULE_DISPATCHER_NAME; |
|
|
|
ActorRef ruleNodeActor = context.actorOf( |
|
|
|
Props.create(new RuleNodeActor.ActorCreator(systemContext, tenantId, entityId, ruleNode.getId())) |
|
|
|
.withDispatcher(dispatcherName), ruleNode.toString()); |
|
|
|
nodeActors.put(ruleNode.getId(), new RuleNodeCtx(self, ruleNodeActor, ruleNode.getId())); |
|
|
|
} |
|
|
|
// Populating the routes map;
|
|
|
|
for (RuleNode ruleNode : ruleNodeList) { |
|
|
|
List<EntityRelation> relations = service.getRuleNodeRelations(ruleNode.getId()); |
|
|
|
for (EntityRelation relation : relations) { |
|
|
|
if (relation.getTo().getEntityType() == EntityType.RULE_NODE) { |
|
|
|
RuleNodeCtx ruleNodeCtx = nodeActors.get(new RuleNodeId(relation.getTo().getId())); |
|
|
|
if (ruleNodeCtx == null) { |
|
|
|
throw new IllegalArgumentException("Rule Node [" + relation.getFrom() + "] has invalid relation to Rule node [" + relation.getTo() + "]"); |
|
|
|
} |
|
|
|
} |
|
|
|
nodeRoutes.computeIfAbsent(ruleNode.getId(), k -> new ArrayList<>()) |
|
|
|
.add(new RuleNodeRelation(ruleNode.getId(), relation.getTo(), relation.getType())); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
firstId = ruleChain.getFirstRuleNodeId(); |
|
|
|
firstNode = nodeActors.get(ruleChain.getFirstRuleNodeId()); |
|
|
|
} |
|
|
|
|
|
|
|
@Override |
|
|
|
public void stop() throws Exception { |
|
|
|
|
|
|
|
public void stop(ActorContext context) throws Exception { |
|
|
|
nodeActors.values().stream().map(RuleNodeCtx::getSelf).forEach(context::stop); |
|
|
|
nodeActors.clear(); |
|
|
|
nodeRoutes.clear(); |
|
|
|
} |
|
|
|
|
|
|
|
@Override |
|
|
|
public void onCreated(ActorContext context) throws Exception { |
|
|
|
public void onClusterEventMsg(ClusterEventMsg msg) throws Exception { |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
@Override |
|
|
|
public void onUpdate(ActorContext context) throws Exception { |
|
|
|
|
|
|
|
void onServiceToRuleEngineMsg(ServiceToRuleEngineMsg envelope) { |
|
|
|
TbMsg tbMsg = envelope.getTbMsg(); |
|
|
|
//TODO: push to queue and act on ack in async way
|
|
|
|
pushMstToNode(firstNode, tbMsg); |
|
|
|
} |
|
|
|
|
|
|
|
@Override |
|
|
|
public void onActivate(ActorContext context) throws Exception { |
|
|
|
|
|
|
|
void onTellNext(RuleNodeToRuleChainTellNextMsg envelope) { |
|
|
|
RuleNodeId originator = envelope.getOriginator(); |
|
|
|
String targetRelationType = envelope.getRelationType(); |
|
|
|
//TODO: log debug output
|
|
|
|
List<RuleNodeRelation> relations = nodeRoutes.get(originator); |
|
|
|
for (RuleNodeRelation relation : relations) { |
|
|
|
if (targetRelationType == null || targetRelationType.equalsIgnoreCase(relation.getType())) { |
|
|
|
switch (relation.getOut().getEntityType()) { |
|
|
|
case RULE_NODE: |
|
|
|
RuleNodeId targetRuleNodeId = new RuleNodeId(relation.getOut().getId()); |
|
|
|
RuleNodeCtx targetRuleNode = nodeActors.get(targetRuleNodeId); |
|
|
|
pushMstToNode(targetRuleNode, envelope.getMsg()); |
|
|
|
break; |
|
|
|
case RULE_CHAIN: |
|
|
|
// TODO: implement
|
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
@Override |
|
|
|
public void onSuspend(ActorContext context) throws Exception { |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
@Override |
|
|
|
public void onStop(ActorContext context) throws Exception { |
|
|
|
|
|
|
|
private void pushMstToNode(RuleNodeCtx nodeCtx, TbMsg msg) { |
|
|
|
//TODO: log debug input
|
|
|
|
firstNode.getSelf().tell(new RuleChainToRuleNodeMsg(new DefaultTbContext(systemContext, nodeCtx), msg), self); |
|
|
|
} |
|
|
|
|
|
|
|
@Override |
|
|
|
public void onClusterEventMsg(ClusterEventMsg msg) throws Exception { |
|
|
|
|
|
|
|
} |
|
|
|
} |
|
|
|
|
Andrew Shvayka
8 years ago