Bug 4105: Add CreateShard message in ShardManager

[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / main / java / org / opendaylight / controller / cluster / datastore / ShardDataTree.java

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/ShardDataTree.java b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/ShardDataTree.java
index e852cfe420416bbfa38049f0938bc954472b8584..fd42740784a3334a0670bec9fcedf90951d4f4ad 100644 (file)
--- a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/ShardDataTree.java
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/datastore/ShardDataTree.java
@@ -7,18 +7,14 @@
  */
 package org.opendaylight.controller.cluster.datastore;
 
-import com.google.common.annotations.VisibleForTesting;
 import com.google.common.base.Optional;
 import com.google.common.base.Preconditions;
 import com.google.common.base.Strings;
-import java.net.URI;
 import java.util.AbstractMap.SimpleEntry;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Map.Entry;
-import java.util.Set;
 import javax.annotation.concurrent.NotThreadSafe;
-import org.opendaylight.controller.cluster.datastore.node.utils.transformer.NormalizedNodePruner;
 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeListener;
 import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeChangeListener;
@@ -53,8 +49,7 @@ public class ShardDataTree extends ShardDataTreeTransactionParent {
     private final ShardDataTreeChangePublisher treeChangePublisher = new ShardDataTreeChangePublisher();
     private final ListenerTree listenerTree = ListenerTree.create();
     private final TipProducingDataTree dataTree;
-    private Set<URI> validNamespaces;
-    private ShardDataTreeTransactionFactory transactionFactory = new RecoveryShardDataTreeTransactionFactory();
+    private SchemaContext schemaContext;
 
     ShardDataTree(final SchemaContext schemaContext) {
         dataTree = InMemoryDataTreeFactory.getInstance().create();
@@ -66,10 +61,14 @@ public class ShardDataTree extends ShardDataTreeTransactionParent {
         return dataTree;
     }
 
+    SchemaContext getSchemaContext() {
+        return schemaContext;
+    }
+
     void updateSchemaContext(final SchemaContext schemaContext) {
         Preconditions.checkNotNull(schemaContext);
+        this.schemaContext = schemaContext;
         dataTree.setSchemaContext(schemaContext);
-        validNamespaces = NormalizedNodePruner.namespaces(schemaContext);
     }
 
     private ShardDataTreeTransactionChain ensureTransactionChain(final String chainId) {
@@ -84,7 +83,7 @@ public class ShardDataTree extends ShardDataTreeTransactionParent {
 
     ReadOnlyShardDataTreeTransaction newReadOnlyTransaction(final String txId, final String chainId) {
         if (Strings.isNullOrEmpty(chainId)) {
-            return transactionFactory.newReadOnlyTransaction(txId, chainId);
+            return new ReadOnlyShardDataTreeTransaction(txId, dataTree.takeSnapshot());
         }
 
         return ensureTransactionChain(chainId).newReadOnlyTransaction(txId);
@@ -92,7 +91,8 @@ public class ShardDataTree extends ShardDataTreeTransactionParent {
 
     ReadWriteShardDataTreeTransaction newReadWriteTransaction(final String txId, final String chainId) {
         if (Strings.isNullOrEmpty(chainId)) {
-            return transactionFactory.newReadWriteTransaction(txId, chainId);
+            return new ReadWriteShardDataTreeTransaction(ShardDataTree.this, txId, dataTree.takeSnapshot()
+                    .newModification());
         }
 
         return ensureTransactionChain(chainId).newReadWriteTransaction(txId);
@@ -182,52 +182,4 @@ public class ShardDataTree extends ShardDataTreeTransactionParent {
         snapshot.ready();
         return new SimpleShardDataTreeCohort(this, snapshot, transaction.getId());
     }
-
-    void recoveryDone(){
-        transactionFactory = new RegularShardDataTreeTransactionFactory();
-    }
-
-    @VisibleForTesting
-    ShardDataTreeTransactionFactory getTransactionFactory(){
-        return transactionFactory;
-    }
-
-    @VisibleForTesting
-    static interface ShardDataTreeTransactionFactory {
-        ReadOnlyShardDataTreeTransaction newReadOnlyTransaction(final String txId, final String chainId);
-
-        ReadWriteShardDataTreeTransaction newReadWriteTransaction(final String txId, final String chainId);
-    }
-
-    @VisibleForTesting
-    class RecoveryShardDataTreeTransactionFactory implements ShardDataTreeTransactionFactory {
-
-        @Override
-        public ReadOnlyShardDataTreeTransaction newReadOnlyTransaction(String txId, String chainId) {
-            return new ReadOnlyShardDataTreeTransaction(txId,
-                    new PruningShardDataTreeSnapshot(dataTree.takeSnapshot(), validNamespaces));
-        }
-
-        @Override
-        public ReadWriteShardDataTreeTransaction newReadWriteTransaction(String txId, String chainId) {
-            return new ReadWriteShardDataTreeTransaction(ShardDataTree.this, txId,
-                    new PruningShardDataTreeSnapshot(dataTree.takeSnapshot(), validNamespaces).newModification());
-        }
-    }
-
-    @VisibleForTesting
-    class RegularShardDataTreeTransactionFactory implements ShardDataTreeTransactionFactory {
-
-        @Override
-        public ReadOnlyShardDataTreeTransaction newReadOnlyTransaction(String txId, String chainId) {
-            return new ReadOnlyShardDataTreeTransaction(txId, dataTree.takeSnapshot());
-
-        }
-
-        @Override
-        public ReadWriteShardDataTreeTransaction newReadWriteTransaction(String txId, String chainId) {
-            return new ReadWriteShardDataTreeTransaction(ShardDataTree.this, txId, dataTree.takeSnapshot()
-                    .newModification());
-        }
-    }
 }