- public static DistributedDataStore createInstance(String name, SchemaService schemaService,
- DatastoreContext datastoreContext, BundleContext bundleContext) {
-
- ActorSystem actorSystem = ActorSystemFactory.createInstance(bundleContext);
- Configuration config = new ConfigurationImpl("module-shards.conf", "modules.conf");
- final DistributedDataStore dataStore =
- new DistributedDataStore(actorSystem, name, new ClusterWrapperImpl(actorSystem),
- config, datastoreContext );
- ShardStrategyFactory.setConfiguration(config);
+ private static final Logger LOG = LoggerFactory.getLogger(DistributedDataStoreFactory.class);
+
+ public static AbstractDataStore createInstance(final SchemaService schemaService,
+ final DatastoreContext initialDatastoreContext, final DatastoreSnapshotRestore datastoreSnapshotRestore,
+ final ActorSystemProvider actorSystemProvider, final BundleContext bundleContext) {
+
+ 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);
+ final DatastoreContextIntrospector introspector = new DatastoreContextIntrospector(initialDatastoreContext);
+ final DatastoreContextConfigAdminOverlay overlay = new DatastoreContextConfigAdminOverlay(
+ introspector, bundleContext);
+
+ final Configuration config = new ConfigurationImpl("module-shards.conf", "modules.conf");
+ 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);
+ }
+
+ overlay.setListener(dataStore);
+