X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-distributed-datastore%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fdatastore%2FDistributedDataStore.java;h=2c1e0a77a88dff2e38b02a64ea195c5ee48d9078;hp=4b2013cd287bd4a91d4e6aa06bf61a1402328ca2;hb=28e9832cc97a345d5ceb69262784e5c8fef77e37;hpb=600e07c2adb213b614cad127070f1e7ff074b42b 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 4b2013cd28..2c1e0a77a8 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 @@ -8,153 +8,37 @@ package org.opendaylight.controller.cluster.datastore; -import akka.actor.ActorRef; import akka.actor.ActorSystem; import com.google.common.annotations.VisibleForTesting; -import com.google.common.base.Preconditions; -import com.google.common.util.concurrent.Uninterruptibles; -import java.util.concurrent.CountDownLatch; -import java.util.concurrent.TimeUnit; +import org.opendaylight.controller.cluster.access.concepts.ClientIdentifier; import org.opendaylight.controller.cluster.datastore.config.Configuration; -import org.opendaylight.controller.cluster.datastore.identifiers.ShardManagerIdentifier; -import org.opendaylight.controller.cluster.datastore.jmx.mbeans.DatastoreConfigurationMXBeanImpl; -import org.opendaylight.controller.cluster.datastore.jmx.mbeans.DatastoreInfoMXBeanImpl; -import org.opendaylight.controller.cluster.datastore.messages.DatastoreSnapshot; +import org.opendaylight.controller.cluster.datastore.persisted.DatastoreSnapshot; import org.opendaylight.controller.cluster.datastore.utils.ActorContext; -import org.opendaylight.controller.cluster.datastore.utils.Dispatchers; -import org.opendaylight.controller.cluster.datastore.utils.PrimaryShardInfoFutureCache; -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.api.DOMDataTreeChangeListener; -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.DOMStoreTreeChangePublisher; import org.opendaylight.controller.sal.core.spi.data.DOMStoreWriteTransaction; -import org.opendaylight.yangtools.concepts.ListenerRegistration; -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; /** - * + * Implements a distributed DOMStore using Akka Patterns.ask(). */ -public class DistributedDataStore implements DOMStore, SchemaContextListener, - DatastoreContextConfigAdminOverlay.Listener, DOMStoreTreeChangePublisher, AutoCloseable { - - private static final Logger LOG = LoggerFactory.getLogger(DistributedDataStore.class); - private static final String UNKNOWN_TYPE = "unknown"; - - private static final long READY_WAIT_FACTOR = 3; - - private final ActorContext actorContext; - private final long waitTillReadyTimeInMillis; - - private AutoCloseable closeable; - - private DatastoreConfigurationMXBeanImpl datastoreConfigMXBean; - - private DatastoreInfoMXBeanImpl datastoreInfoMXBean; - - private final CountDownLatch waitTillReadyCountDownLatch = new CountDownLatch(1); - - private final String type; +public class DistributedDataStore extends AbstractDataStore { private final TransactionContextFactory txContextFactory; - public DistributedDataStore(ActorSystem actorSystem, ClusterWrapper cluster, - Configuration configuration, DatastoreContextFactory datastoreContextFactory, - DatastoreSnapshot restoreFromSnapshot) { - Preconditions.checkNotNull(actorSystem, "actorSystem should not be null"); - Preconditions.checkNotNull(cluster, "cluster should not be null"); - Preconditions.checkNotNull(configuration, "configuration should not be null"); - Preconditions.checkNotNull(datastoreContextFactory, "datastoreContextFactory should not be null"); - - this.type = datastoreContextFactory.getBaseDatastoreContext().getDataStoreName(); - - String shardManagerId = ShardManagerIdentifier.builder().type(type).build().toString(); - - LOG.info("Creating ShardManager : {}", shardManagerId); - - String shardDispatcher = - new Dispatchers(actorSystem.dispatchers()).getDispatcherPath(Dispatchers.DispatcherType.Shard); - - PrimaryShardInfoFutureCache primaryShardInfoCache = new PrimaryShardInfoFutureCache(); - - ShardManager.Builder builder = ShardManager.builder().cluster(cluster).configuration(configuration). - datastoreContextFactory(datastoreContextFactory).waitTillReadyCountdownLatch(waitTillReadyCountDownLatch). - primaryShardInfoCache(primaryShardInfoCache).restoreFromSnapshot(restoreFromSnapshot); - - actorContext = new ActorContext(actorSystem, createShardManager(actorSystem, builder, shardDispatcher, - shardManagerId), cluster, configuration, datastoreContextFactory.getBaseDatastoreContext(), primaryShardInfoCache); - - this.waitTillReadyTimeInMillis = - actorContext.getDatastoreContext().getShardLeaderElectionTimeout().duration().toMillis() * READY_WAIT_FACTOR; - - this.txContextFactory = TransactionContextFactory.create(actorContext); - - datastoreConfigMXBean = new DatastoreConfigurationMXBeanImpl( - datastoreContextFactory.getBaseDatastoreContext().getDataStoreMXBeanType()); - datastoreConfigMXBean.setContext(datastoreContextFactory.getBaseDatastoreContext()); - datastoreConfigMXBean.registerMBean(); - - datastoreInfoMXBean = new DatastoreInfoMXBeanImpl(datastoreContextFactory.getBaseDatastoreContext(). - getDataStoreMXBeanType(), actorContext); - datastoreInfoMXBean.registerMBean(); + public DistributedDataStore(final ActorSystem actorSystem, final ClusterWrapper cluster, + final Configuration configuration, final DatastoreContextFactory datastoreContextFactory, + final DatastoreSnapshot restoreFromSnapshot) { + super(actorSystem, cluster, configuration, datastoreContextFactory, restoreFromSnapshot); + this.txContextFactory = new TransactionContextFactory(getActorContext(), getIdentifier()); } @VisibleForTesting - DistributedDataStore(ActorContext actorContext) { - this.actorContext = Preconditions.checkNotNull(actorContext, "actorContext should not be null"); - this.txContextFactory = TransactionContextFactory.create(actorContext); - this.type = UNKNOWN_TYPE; - this.waitTillReadyTimeInMillis = - actorContext.getDatastoreContext().getShardLeaderElectionTimeout().duration().toMillis() * READY_WAIT_FACTOR; - } - - public void setCloseable(AutoCloseable closeable) { - this.closeable = closeable; + DistributedDataStore(final ActorContext actorContext, final ClientIdentifier identifier) { + super(actorContext, identifier); + this.txContextFactory = new TransactionContextFactory(getActorContext(), getIdentifier()); } - @SuppressWarnings("unchecked") - @Override - public >> - ListenerRegistration registerChangeListener( - final YangInstanceIdentifier path, L listener, - AsyncDataBroker.DataChangeScope scope) { - - Preconditions.checkNotNull(path, "path should not be null"); - Preconditions.checkNotNull(listener, "listener should not be null"); - - LOG.debug("Registering listener: {} for path: {} scope: {}", listener, path, scope); - - String shardName = actorContext.getShardStrategyFactory().getStrategy(path).findShard(path); - - final DataChangeListenerRegistrationProxy listenerRegistrationProxy = - new DataChangeListenerRegistrationProxy(shardName, actorContext, listener); - listenerRegistrationProxy.init(path, scope); - - return listenerRegistrationProxy; - } - - @Override - public ListenerRegistration registerTreeChangeListener(YangInstanceIdentifier treeId, L listener) { - Preconditions.checkNotNull(treeId, "treeId should not be null"); - Preconditions.checkNotNull(listener, "listener should not be null"); - - final String shardName = actorContext.getShardStrategyFactory().getStrategy(treeId).findShard(treeId); - LOG.debug("Registering tree listener: {} for tree: {} shard: {}", listener, treeId, shardName); - - final DataTreeChangeListenerProxy listenerRegistrationProxy = - new DataTreeChangeListenerProxy(actorContext, listener); - listenerRegistrationProxy.init(shardName, treeId); - - return listenerRegistrationProxy; - } @Override public DOMStoreTransactionChain createTransactionChain() { @@ -163,95 +47,24 @@ public class DistributedDataStore implements DOMStore, SchemaContextListener, @Override public DOMStoreReadTransaction newReadOnlyTransaction() { - return new TransactionProxy(txContextFactory, TransactionType.READ_ONLY); + return new TransactionProxy(txContextFactory, TransactionType.READ_ONLY); } @Override public DOMStoreWriteTransaction newWriteOnlyTransaction() { - actorContext.acquireTxCreationPermit(); + getActorContext().acquireTxCreationPermit(); return new TransactionProxy(txContextFactory, TransactionType.WRITE_ONLY); } @Override public DOMStoreReadWriteTransaction newReadWriteTransaction() { - actorContext.acquireTxCreationPermit(); + getActorContext().acquireTxCreationPermit(); return new TransactionProxy(txContextFactory, TransactionType.READ_WRITE); } - @Override - public void onGlobalContextUpdated(SchemaContext schemaContext) { - actorContext.setSchemaContext(schemaContext); - } - - @Override - public void onDatastoreContextUpdated(DatastoreContextFactory contextFactory) { - LOG.info("DatastoreContext updated for data store {}", actorContext.getDataStoreName()); - - actorContext.setDatastoreContext(contextFactory); - datastoreConfigMXBean.setContext(contextFactory.getBaseDatastoreContext()); - } - @Override public void close() { - LOG.info("Closing data store {}", type); - - if (datastoreConfigMXBean != null) { - datastoreConfigMXBean.unregisterMBean(); - } - if (datastoreInfoMXBean != null) { - datastoreInfoMXBean.unregisterMBean(); - } - - if (closeable != null) { - try { - closeable.close(); - } catch (Exception e) { - LOG.debug("Error closing instance", e); - } - } - txContextFactory.close(); - actorContext.shutdown(); - } - - public ActorContext getActorContext() { - return actorContext; - } - - public void waitTillReady(){ - LOG.info("Beginning to wait for data store to become ready : {}", type); - - try { - if (waitTillReadyCountDownLatch.await(waitTillReadyTimeInMillis, TimeUnit.MILLISECONDS)) { - LOG.debug("Data store {} is now ready", type); - } else { - LOG.error("Shared leaders failed to settle in {} seconds, giving up", TimeUnit.MILLISECONDS.toSeconds(waitTillReadyTimeInMillis)); - } - } catch (InterruptedException e) { - LOG.error("Interrupted while waiting for shards to settle", e); - } - } - - private static ActorRef createShardManager(ActorSystem actorSystem, ShardManager.Builder builder, - String shardDispatcher, String shardManagerId) { - Exception lastException = null; - - for(int i=0;i<100;i++) { - try { - return actorSystem.actorOf(builder.props().withDispatcher(shardDispatcher).withMailbox( - ActorContext.MAILBOX), shardManagerId); - } catch (Exception e){ - lastException = e; - Uninterruptibles.sleepUninterruptibly(100, TimeUnit.MILLISECONDS); - LOG.debug(String.format("Could not create actor %s because of %s - waiting for sometime before retrying (retry count = %d)", shardManagerId, e.getMessage(), i)); - } - } - - throw new IllegalStateException("Failed to create Shard Manager", lastException); - } - - @VisibleForTesting - public CountDownLatch getWaitTillReadyCountDownLatch() { - return waitTillReadyCountDownLatch; + super.close(); } }