X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-remoterpc-connector%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fremote%2Frpc%2FRpcManager.java;h=8e53bcba83464fd9b336f935ba03fc095be113a6;hb=9ddc65e1ddae50f691566cd9382707679436c055;hp=5c56455bd0c208a40709b4a221fb6a22ed0b65a1;hpb=eb887b1c2c8cd2768f8b4c2ed2b5054f97798466;p=controller.git diff --git a/opendaylight/md-sal/sal-remoterpc-connector/src/main/java/org/opendaylight/controller/remote/rpc/RpcManager.java b/opendaylight/md-sal/sal-remoterpc-connector/src/main/java/org/opendaylight/controller/remote/rpc/RpcManager.java index 5c56455bd0..8e53bcba83 100644 --- a/opendaylight/md-sal/sal-remoterpc-connector/src/main/java/org/opendaylight/controller/remote/rpc/RpcManager.java +++ b/opendaylight/md-sal/sal-remoterpc-connector/src/main/java/org/opendaylight/controller/remote/rpc/RpcManager.java @@ -13,116 +13,138 @@ import akka.actor.ActorRef; import akka.actor.OneForOneStrategy; import akka.actor.Props; import akka.actor.SupervisorStrategy; -import akka.japi.Creator; +import akka.actor.SupervisorStrategy.Directive; import akka.japi.Function; +import com.google.common.base.Preconditions; +import java.util.ArrayList; +import java.util.HashSet; +import java.util.List; +import java.util.Set; +import org.opendaylight.controller.cluster.common.actor.AbstractUntypedActor; +import org.opendaylight.controller.md.sal.dom.api.DOMRpcIdentifier; +import org.opendaylight.controller.md.sal.dom.api.DOMRpcProviderService; +import org.opendaylight.controller.md.sal.dom.api.DOMRpcService; +import org.opendaylight.controller.md.sal.dom.broker.spi.rpc.RpcRoutingStrategy; import org.opendaylight.controller.remote.rpc.messages.UpdateSchemaContext; -import org.opendaylight.controller.remote.rpc.registry.ClusterWrapper; import org.opendaylight.controller.remote.rpc.registry.RpcRegistry; -import org.opendaylight.controller.sal.core.api.Broker; -import org.opendaylight.controller.sal.core.api.RpcProvisionRegistry; -import org.opendaylight.yangtools.yang.common.QName; +import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier; +import org.opendaylight.yangtools.yang.model.api.Module; +import org.opendaylight.yangtools.yang.model.api.RpcDefinition; import org.opendaylight.yangtools.yang.model.api.SchemaContext; -import org.slf4j.Logger; -import org.slf4j.LoggerFactory; import scala.concurrent.duration.Duration; -import java.util.Set; /** - * This class acts as a supervisor, creates all the actors, resumes them, if an exception is thrown. - * - * It also starts the rpc listeners + * This class acts as a supervisor, creates all the actors, resumes them, if an exception is thrown. It also starts + * the rpc listeners */ public class RpcManager extends AbstractUntypedActor { + private SchemaContext schemaContext; + private ActorRef rpcBroker; + private ActorRef rpcRegistry; + private final RemoteRpcProviderConfig config; + private RpcListener rpcListener; + private RemoteRpcImplementation rpcImplementation; + private final DOMRpcProviderService rpcProvisionRegistry; + private final DOMRpcService rpcServices; + + private RpcManager(final SchemaContext schemaContext, + final DOMRpcProviderService rpcProvisionRegistry, + final DOMRpcService rpcSevices, + final RemoteRpcProviderConfig config) { + this.schemaContext = schemaContext; + this.rpcProvisionRegistry = rpcProvisionRegistry; + rpcServices = rpcSevices; + this.config = config; + + createRpcActors(); + startListeners(); + } - private static final Logger LOG = LoggerFactory.getLogger(RpcManager.class); - - private SchemaContext schemaContext; - private final ClusterWrapper clusterWrapper; - private ActorRef rpcBroker; - private ActorRef rpcRegistry; - private final Broker.ProviderSession brokerSession; - private RpcListener rpcListener; - private RoutedRpcListener routeChangeListener; - private RemoteRpcImplementation rpcImplementation; - private final RpcProvisionRegistry rpcProvisionRegistry; - - private RpcManager(ClusterWrapper clusterWrapper, SchemaContext schemaContext, - Broker.ProviderSession brokerSession, RpcProvisionRegistry rpcProvisionRegistry) { - this.clusterWrapper = clusterWrapper; - this.schemaContext = schemaContext; - this.brokerSession = brokerSession; - this.rpcProvisionRegistry = rpcProvisionRegistry; - - createRpcActors(); - startListeners(); - } - - - public static Props props(final ClusterWrapper clusterWrapper, final SchemaContext schemaContext, - final Broker.ProviderSession brokerSession, final RpcProvisionRegistry rpcProvisionRegistry) { - return Props.create(new Creator() { - @Override - public RpcManager create() throws Exception { - return new RpcManager(clusterWrapper, schemaContext, brokerSession, rpcProvisionRegistry); - } - }); - } - - private void createRpcActors() { - LOG.debug("Create rpc registry and broker actors"); - - rpcRegistry = getContext().actorOf(RpcRegistry.props(clusterWrapper), ActorConstants.RPC_REGISTRY); - rpcBroker = getContext().actorOf(RpcBroker.props(brokerSession, rpcRegistry, schemaContext), ActorConstants.RPC_BROKER); - } - - private void startListeners() { - LOG.debug("Registers rpc listeners"); - - String rpcBrokerPath = clusterWrapper.getAddress().toString() + ActorConstants.RPC_BROKER_PATH; - rpcListener = new RpcListener(rpcRegistry, rpcBrokerPath); - routeChangeListener = new RoutedRpcListener(rpcRegistry, rpcBrokerPath); - rpcImplementation = new RemoteRpcImplementation(rpcBroker, schemaContext); - - brokerSession.addRpcRegistrationListener(rpcListener); - rpcProvisionRegistry.registerRouteChangeListener(routeChangeListener); - rpcProvisionRegistry.setRoutedRpcDefaultDelegate(rpcImplementation); - announceSupportedRpcs(); - } - - /** - * Add all the locally registered RPCs in the clustered routing table - */ - private void announceSupportedRpcs(){ - LOG.debug("Adding all supported rpcs to routing table"); - Set currentlySupported = brokerSession.getSupportedRpcs(); - for (QName rpc : currentlySupported) { - rpcListener.onRpcImplementationAdded(rpc); + + public static Props props(final SchemaContext schemaContext, final DOMRpcProviderService rpcProvisionRegistry, + final DOMRpcService rpcServices, final RemoteRpcProviderConfig config) { + Preconditions.checkNotNull(schemaContext, "SchemaContext can not be null!"); + Preconditions.checkNotNull(rpcProvisionRegistry, "RpcProviderService can not be null!"); + Preconditions.checkNotNull(rpcServices, "RpcService can not be null!"); + return Props.create(RpcManager.class, schemaContext, rpcProvisionRegistry, rpcServices, config); } - } + private void createRpcActors() { + LOG.debug("Create rpc registry and broker actors"); + + rpcRegistry = getContext().actorOf(RpcRegistry.props(config) + .withMailbox(config.getMailBoxName()), config.getRpcRegistryName()); + + rpcBroker = getContext().actorOf(RpcBroker.props(rpcServices) + .withMailbox(config.getMailBoxName()), config.getRpcBrokerName()); - @Override - protected void handleReceive(Object message) throws Exception { - if(message instanceof UpdateSchemaContext) { - updateSchemaContext((UpdateSchemaContext) message); + final RpcRegistry.Messages.SetLocalRouter localRouter = new RpcRegistry.Messages.SetLocalRouter(rpcBroker); + rpcRegistry.tell(localRouter, self()); } - } + private void startListeners() { + LOG.debug("Registers rpc listeners"); - private void updateSchemaContext(UpdateSchemaContext message) { - this.schemaContext = message.getSchemaContext(); - } + rpcListener = new RpcListener(rpcRegistry); + rpcImplementation = new RemoteRpcImplementation(rpcRegistry, config); - @Override - public SupervisorStrategy supervisorStrategy() { - return new OneForOneStrategy(10, Duration.create("1 minute"), - new Function() { - @Override - public SupervisorStrategy.Directive apply(Throwable t) { - return SupervisorStrategy.resume(); - } + rpcServices.registerRpcListener(rpcListener); + + registerRoutedRpcDelegate(); + announceSupportedRpcs(); + } + + private void registerRoutedRpcDelegate() { + final Set rpcIdentifiers = new HashSet<>(); + final Set modules = schemaContext.getModules(); + for (final Module module : modules) { + for (final RpcDefinition rpcDefinition : module.getRpcs()) { + if (RpcRoutingStrategy.from(rpcDefinition).isContextBasedRouted()) { + LOG.debug("Adding routed rpcDefinition for path {}", rpcDefinition.getPath()); + rpcIdentifiers.add(DOMRpcIdentifier.create(rpcDefinition.getPath(), YangInstanceIdentifier.EMPTY)); + } + } + } + rpcProvisionRegistry.registerRpcImplementation(rpcImplementation, rpcIdentifiers); + } + + /** + * Add all the locally registered RPCs in the clustered routing table. + */ + private void announceSupportedRpcs() { + LOG.debug("Adding all supported rpcs to routing table"); + final Set currentlySupportedRpc = schemaContext.getOperations(); + final List rpcs = new ArrayList<>(); + for (final RpcDefinition rpcDef : currentlySupportedRpc) { + rpcs.add(DOMRpcIdentifier.create(rpcDef.getPath())); } - ); - } + + if (!rpcs.isEmpty()) { + rpcListener.onRpcAvailable(rpcs); + } + } + + + @Override + protected void handleReceive(final Object message) throws Exception { + if (message instanceof UpdateSchemaContext) { + updateSchemaContext((UpdateSchemaContext) message); + } + } + + private void updateSchemaContext(final UpdateSchemaContext message) { + schemaContext = message.getSchemaContext(); + registerRoutedRpcDelegate(); + rpcBroker.tell(message, ActorRef.noSender()); + } + + @Override + public SupervisorStrategy supervisorStrategy() { + return new OneForOneStrategy(10, Duration.create("1 minute"), (Function) t -> { + LOG.error("An exception happened actor will be resumed", t); + + return SupervisorStrategy.resume(); + }); + } }