X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-distributed-datastore%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fdatastore%2FDistributedDataStore.java;h=404a4e02033ea1c89f9fada4135cfe0e4b6a935e;hb=3145f4149ff58cd393819e24454a08b384122bb2;hp=29fc259bb75935d25f13dcb8f7a0865f74f145cd;hpb=9c0508d8b591e356f145d5d1a277c10965a647bb;p=controller.git diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/DistributedDataStore.java b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/DistributedDataStore.java index 29fc259bb7..404a4e0203 100644 --- a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/DistributedDataStore.java +++ b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/DistributedDataStore.java @@ -9,132 +9,130 @@ package org.opendaylight.controller.cluster.datastore; import akka.actor.ActorRef; -import akka.actor.ActorSelection; import akka.actor.ActorSystem; -import akka.util.Timeout; -import org.opendaylight.controller.cluster.datastore.messages.FindPrimary; -import org.opendaylight.controller.cluster.datastore.messages.PrimaryFound; + +import com.google.common.base.Preconditions; +import org.opendaylight.controller.cluster.datastore.identifiers.ShardManagerIdentifier; import org.opendaylight.controller.cluster.datastore.messages.RegisterChangeListener; import org.opendaylight.controller.cluster.datastore.messages.RegisterChangeListenerReply; import org.opendaylight.controller.cluster.datastore.messages.UpdateSchemaContext; +import org.opendaylight.controller.cluster.datastore.shardstrategy.ShardStrategyFactory; +import org.opendaylight.controller.cluster.datastore.utils.ActorContext; import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker; import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeListener; +import org.opendaylight.controller.md.sal.dom.store.impl.InMemoryDOMDataStoreConfigProperties; import org.opendaylight.controller.sal.core.spi.data.DOMStore; import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadTransaction; import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadWriteTransaction; import org.opendaylight.controller.sal.core.spi.data.DOMStoreTransactionChain; import org.opendaylight.controller.sal.core.spi.data.DOMStoreWriteTransaction; import org.opendaylight.yangtools.concepts.ListenerRegistration; -import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier; +import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier; import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode; import org.opendaylight.yangtools.yang.model.api.SchemaContext; import org.opendaylight.yangtools.yang.model.api.SchemaContextListener; import org.slf4j.Logger; import org.slf4j.LoggerFactory; -import scala.concurrent.Await; -import scala.concurrent.Future; -import scala.concurrent.duration.Duration; -import scala.concurrent.duration.FiniteDuration; - -import java.util.concurrent.TimeUnit; - -import static akka.pattern.Patterns.ask; /** * */ -public class DistributedDataStore implements DOMStore, SchemaContextListener { +public class DistributedDataStore implements DOMStore, SchemaContextListener, AutoCloseable { - private static final Logger - LOG = LoggerFactory.getLogger(DistributedDataStore.class); + private static final Logger LOG = LoggerFactory.getLogger(DistributedDataStore.class); - final FiniteDuration ASK_DURATION = Duration.create(5, TimeUnit.SECONDS); - final Duration AWAIT_DURATION = Duration.create(5, TimeUnit.SECONDS); + private final ActorContext actorContext; - private final ActorRef shardManager; - private final ActorSystem actorSystem; - private final String type; + private SchemaContext schemaContext; + public DistributedDataStore(ActorSystem actorSystem, String type, ClusterWrapper cluster, + Configuration configuration, InMemoryDOMDataStoreConfigProperties dataStoreProperties) { + Preconditions.checkNotNull(actorSystem, "actorSystem should not be null"); + Preconditions.checkNotNull(type, "type should not be null"); + Preconditions.checkNotNull(cluster, "cluster should not be null"); + Preconditions.checkNotNull(configuration, "configuration should not be null"); - public DistributedDataStore(ActorSystem actorSystem, String type) { - this.actorSystem = actorSystem; - this.type = type; - shardManager = actorSystem.actorOf(ShardManager.props(type)); - } - @Override - public >> ListenerRegistration registerChangeListener( - InstanceIdentifier path, L listener, - AsyncDataBroker.DataChangeScope scope) { + String shardManagerId = ShardManagerIdentifier.builder().type(type).build().toString(); - ActorSelection primary = findPrimary(); + LOG.info("Creating ShardManager : {}", shardManagerId); - ActorRef dataChangeListenerActor = actorSystem.actorOf(DataChangeListener.props()); - - Object result = - getResult(primary, new RegisterChangeListener(path, dataChangeListenerActor.path(), - AsyncDataBroker.DataChangeScope.BASE), ASK_DURATION); + this.actorContext = new ActorContext(actorSystem, actorSystem + .actorOf(ShardManager.props(type, cluster, configuration, dataStoreProperties), + shardManagerId ), cluster, configuration); + } - RegisterChangeListenerReply reply = (RegisterChangeListenerReply) result; - return new ListenerRegistrationProxy(reply.getListenerRegistrationPath()); + public DistributedDataStore(ActorContext actorContext) { + this.actorContext = Preconditions.checkNotNull(actorContext, "actorContext should not be null"); } - private ActorSelection findPrimary() { - Object result = getResult(shardManager, new FindPrimary(Shard.DEFAULT_NAME), ASK_DURATION); - if(result instanceof PrimaryFound){ - PrimaryFound found = (PrimaryFound) result; - LOG.error("Primary found {}", found.getPrimaryPath()); + @SuppressWarnings("unchecked") + @Override + public >> + ListenerRegistration registerChangeListener( + YangInstanceIdentifier path, L listener, + AsyncDataBroker.DataChangeScope scope) { - return actorSystem.actorSelection(found.getPrimaryPath()); - } - throw new RuntimeException("primary was not found"); - } + Preconditions.checkNotNull(path, "path should not be null"); + Preconditions.checkNotNull(listener, "listener should not be null"); - private Object getResult(ActorRef actor, Object message, FiniteDuration duration){ - Future future = - ask(actor, message, new Timeout(duration)); + LOG.debug("Registering listener: {} for path: {} scope: {}", listener, path, scope); - try { - return Await.result(future, AWAIT_DURATION); - } catch (Exception e) { - throw new RuntimeException(e); - } - } + ActorRef dataChangeListenerActor = actorContext.getActorSystem().actorOf( + DataChangeListener.props(schemaContext,listener,path )); - private Object getResult(ActorSelection actor, Object message, FiniteDuration duration){ - Future future = - ask(actor, message, new Timeout(duration)); + String shardName = ShardStrategyFactory.getStrategy(path).findShard(path); - try { - return Await.result(future, AWAIT_DURATION); - } catch (Exception e) { - throw new RuntimeException(e); + Object result = actorContext.executeLocalShardOperation(shardName, + new RegisterChangeListener(path, dataChangeListenerActor.path(), scope), + ActorContext.ASK_DURATION); + + if (result != null) { + RegisterChangeListenerReply reply = (RegisterChangeListenerReply) result; + return new DataChangeListenerRegistrationProxy(actorContext + .actorSelection(reply.getListenerRegistrationPath()), listener, + dataChangeListenerActor); } - } + LOG.debug( + "No local shard for shardName {} was found so returning a noop registration", + shardName); + + return new NoOpDataChangeListenerRegistration(listener); + } @Override public DOMStoreTransactionChain createTransactionChain() { - return new TransactionChainProxy(); + return new TransactionChainProxy(actorContext, schemaContext); } @Override public DOMStoreReadTransaction newReadOnlyTransaction() { - return new TransactionProxy(); + return new TransactionProxy(actorContext, TransactionProxy.TransactionType.READ_ONLY, + schemaContext); } @Override public DOMStoreWriteTransaction newWriteOnlyTransaction() { - return new TransactionProxy(); + return new TransactionProxy(actorContext, TransactionProxy.TransactionType.WRITE_ONLY, + schemaContext); } @Override public DOMStoreReadWriteTransaction newReadWriteTransaction() { - return new TransactionProxy(); + return new TransactionProxy(actorContext, TransactionProxy.TransactionType.READ_WRITE, + schemaContext); } @Override public void onGlobalContextUpdated(SchemaContext schemaContext) { - shardManager.tell(new UpdateSchemaContext(schemaContext), null); + this.schemaContext = schemaContext; + actorContext.getShardManager().tell( + new UpdateSchemaContext(schemaContext), null); + } + + @Override public void close() throws Exception { + actorContext.shutdown(); + } }