DataSchemaContextTree should expose EffectiveModelContext

There is no point hiding the context stored underneath, expose it
to users, so they may better integrate.

Change-Id: I269943567208539c2b2fcdf90067f6b08490a689
Signed-off-by: Robert Varga <robert.varga@pantheon.tech>

diff --git a/yang/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/DataSchemaContextTree.java b/yang/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/DataSchemaContextTree.java
index 6d072096c5c239c7a7b6db1a3118388cb36a5f00..df3e019b61ec30b8fcb85a3aa0fee966696a127b 100644 (file)
--- a/yang/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/DataSchemaContextTree.java
+++ b/yang/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/DataSchemaContextTree.java
@@ -15,6 +15,7 @@ import org.eclipse.jdt.annotation.NonNull;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
+import org.opendaylight.yangtools.yang.model.spi.AbstractEffectiveModelContextProvider;
 
 /**
  * Semantic tree binding a {@link EffectiveModelContext} to a {@link NormalizedNode} tree. Since the layout of the
@@ -22,7 +23,7 @@ import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
  *
  * @author Robert Varga
  */
-public final class DataSchemaContextTree {
+public final class DataSchemaContextTree extends AbstractEffectiveModelContextProvider {
     private static final LoadingCache<EffectiveModelContext, DataSchemaContextTree> TREES = CacheBuilder.newBuilder()
             .weakKeys().weakValues().build(new CacheLoader<EffectiveModelContext, DataSchemaContextTree>() {
                 @Override
@@ -34,6 +35,7 @@ public final class DataSchemaContextTree {
     private final DataSchemaContextNode<?> root;
 
     private DataSchemaContextTree(final EffectiveModelContext ctx) {
+        super(ctx);
         root = DataSchemaContextNode.from(ctx);
     }