package org.opendaylight.controller.cluster.datastore;
import akka.actor.ActorSystem;
-import akka.actor.Props;
-import akka.osgi.BundleDelegatingClassLoader;
-import com.typesafe.config.ConfigFactory;
-import org.opendaylight.controller.cluster.datastore.config.ConfigurationReader;
-import org.opendaylight.controller.cluster.datastore.shardstrategy.ShardStrategyFactory;
-import org.opendaylight.controller.sal.core.api.model.SchemaService;
-import org.osgi.framework.BundleContext;
-
-public class DistributedDataStoreFactory {
- private static final String ACTOR_SYSTEM_NAME = "opendaylight-cluster-data";
- private static final String CONFIGURATION_NAME = "odl-cluster-data";
-
- private static volatile ActorSystem persistentActorSystem = null;
-
- public static DistributedDataStore createInstance(SchemaService schemaService,
- DatastoreContext datastoreContext, BundleContext bundleContext) {
-
- ActorSystem actorSystem = getOrCreateInstance(bundleContext, datastoreContext.getConfigurationReader());
- Configuration config = new ConfigurationImpl("module-shards.conf", "modules.conf");
- final DistributedDataStore dataStore =
- new DistributedDataStore(actorSystem, new ClusterWrapperImpl(actorSystem),
- config, datastoreContext);
-
- ShardStrategyFactory.setConfiguration(config);
- schemaService.registerSchemaContextListener(dataStore);
- return dataStore;
+import org.opendaylight.controller.cluster.ActorSystemProvider;
+import org.opendaylight.controller.cluster.databroker.ClientBackedDataStore;
+import org.opendaylight.controller.cluster.datastore.config.Configuration;
+import org.opendaylight.controller.cluster.datastore.config.ConfigurationImpl;
+import org.opendaylight.controller.cluster.datastore.persisted.DatastoreSnapshot;
+import org.opendaylight.mdsal.dom.api.DOMSchemaService;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public final class DistributedDataStoreFactory {
+
+ private static final Logger LOG = LoggerFactory.getLogger(DistributedDataStoreFactory.class);
+ private static final String DEFAULT_MODULE_SHARDS_PATH = "./configuration/initial/module-shards.conf";
+ private static final String DEFAULT_MODULES_PATH = "./configuration/initial/modules.conf";
+
+ private DistributedDataStoreFactory() {
}
- private static final ActorSystem getOrCreateInstance(final BundleContext bundleContext, ConfigurationReader configurationReader) {
- ActorSystem ret = persistentActorSystem;
- if (ret == null) {
- synchronized (DistributedDataStoreFactory.class) {
- ret = persistentActorSystem;
- if (ret == null) {
- // Create an OSGi bundle classloader for actor system
- BundleDelegatingClassLoader classLoader = new BundleDelegatingClassLoader(bundleContext.getBundle(),
- Thread.currentThread().getContextClassLoader());
-
- ret = ActorSystem.create(ACTOR_SYSTEM_NAME,
- ConfigFactory.load(configurationReader.read()).getConfig(CONFIGURATION_NAME), classLoader);
- ret.actorOf(Props.create(TerminationMonitor.class), "termination-monitor");
-
- persistentActorSystem = ret;
- }
- }
+ public static AbstractDataStore createInstance(final DOMSchemaService schemaService,
+ final DatastoreContext initialDatastoreContext, final DatastoreSnapshotRestore datastoreSnapshotRestore,
+ final ActorSystemProvider actorSystemProvider, final DatastoreContextIntrospector introspector,
+ final DatastoreContextPropertiesUpdater updater) {
+ return createInstance(schemaService, initialDatastoreContext, datastoreSnapshotRestore, actorSystemProvider,
+ introspector, updater, null);
+ }
+
+ public static AbstractDataStore createInstance(final DOMSchemaService schemaService,
+ final DatastoreContext initialDatastoreContext, final DatastoreSnapshotRestore datastoreSnapshotRestore,
+ final ActorSystemProvider actorSystemProvider, final DatastoreContextIntrospector introspector,
+ final DatastoreContextPropertiesUpdater updater, final Configuration orgConfig) {
+
+ final String datastoreName = initialDatastoreContext.getDataStoreName();
+ LOG.info("Create data store instance of type : {}", datastoreName);
+
+ final ActorSystem actorSystem = actorSystemProvider.getActorSystem();
+ final DatastoreSnapshot restoreFromSnapshot = datastoreSnapshotRestore.getAndRemove(datastoreName);
+
+ Configuration config;
+ if (orgConfig == null) {
+ config = new ConfigurationImpl(DEFAULT_MODULE_SHARDS_PATH, DEFAULT_MODULES_PATH);
+ } else {
+ config = orgConfig;
+ }
+ final ClusterWrapper clusterWrapper = new ClusterWrapperImpl(actorSystem);
+ final DatastoreContextFactory contextFactory = introspector.newContextFactory();
+
+ // This is the potentially-updated datastore context, distinct from the initial one
+ final DatastoreContext datastoreContext = contextFactory.getBaseDatastoreContext();
+
+ final AbstractDataStore dataStore;
+ if (datastoreContext.isUseTellBasedProtocol()) {
+ dataStore = new ClientBackedDataStore(actorSystem, clusterWrapper, config, contextFactory,
+ restoreFromSnapshot);
+ LOG.info("Data store {} is using tell-based protocol", datastoreName);
+ } else {
+ dataStore = new DistributedDataStore(actorSystem, clusterWrapper, config, contextFactory,
+ restoreFromSnapshot);
+ LOG.info("Data store {} is using ask-based protocol", datastoreName);
}
+ updater.setListener(dataStore);
- return ret;
+ schemaService.registerSchemaContextListener(dataStore);
+
+ dataStore.setCloseable(updater);
+ dataStore.waitTillReady();
+
+ return dataStore;
}
}