Bug 5485: Improve DataTreeModification pruning on recovery

[controller.git] / opendaylight / md-sal / sal-clustering-commons / src / main / java / org / opendaylight / controller / cluster / datastore / node / utils / transformer / NormalizedNodeBuilderWrapper.java

diff --git a/opendaylight/md-sal/sal-clustering-commons/src/main/java/org/opendaylight/controller/cluster/datastore/node/utils/transformer/NormalizedNodeBuilderWrapper.java b/opendaylight/md-sal/sal-clustering-commons/src/main/java/org/opendaylight/controller/cluster/datastore/node/utils/transformer/NormalizedNodeBuilderWrapper.java
index 5295c88153dc15750ac421229bba681e4ddccf80..259529eaa9b37d8cb50af973e365c093eaac2838 100644 (file)
--- a/opendaylight/md-sal/sal-clustering-commons/src/main/java/org/opendaylight/controller/cluster/datastore/node/utils/transformer/NormalizedNodeBuilderWrapper.java
+++ b/opendaylight/md-sal/sal-clustering-commons/src/main/java/org/opendaylight/controller/cluster/datastore/node/utils/transformer/NormalizedNodeBuilderWrapper.java
@@ -8,18 +8,22 @@
 
 package org.opendaylight.controller.cluster.datastore.node.utils.transformer;
 
+import com.google.common.base.Optional;
 import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.NormalizedNodeContainerBuilder;
+import org.opendaylight.yangtools.yang.data.util.DataSchemaContextNode;
 
 public class NormalizedNodeBuilderWrapper {
     private final NormalizedNodeContainerBuilder<?,?,?,?> builder;
     private final YangInstanceIdentifier.PathArgument identifier;
+    private final Optional<DataSchemaContextNode<?>> schemaNode;
 
-
-    NormalizedNodeBuilderWrapper(NormalizedNodeContainerBuilder<?,?,?,?> builder, YangInstanceIdentifier.PathArgument identifier) {
+    NormalizedNodeBuilderWrapper(NormalizedNodeContainerBuilder<?,?,?,?> builder,
+            YangInstanceIdentifier.PathArgument identifier, Optional<DataSchemaContextNode<?>> schemaNode) {
         this.builder = builder;
         this.identifier = identifier;
+        this.schemaNode = schemaNode;
     }
 
     public NormalizedNodeContainerBuilder builder(){
@@ -34,4 +38,7 @@ public class NormalizedNodeBuilderWrapper {
         return identifier;
     }
 
+    public Optional<DataSchemaContextNode<?>> getSchema() {
+        return schemaNode;
+    }
 }