Special-case schema-mount nodes

[yangtools.git] / yang / yang-data-util / src / main / java / org / opendaylight / yangtools / yang / data / util / MountPointData.java

diff --git a/yang/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/MountPointData.java b/yang/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/MountPointData.java
index 393da67922e52e7a5f77abb44d93a8be19210423..f60a23ba46cf403e7de2793ab86a3df1c3cf3900 100644 (file)
--- a/yang/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/MountPointData.java
+++ b/yang/yang-data-util/src/main/java/org/opendaylight/yangtools/yang/data/util/MountPointData.java
@@ -46,6 +46,8 @@ public final class MountPointData extends AbstractIdentifiable<MountPointIdentif
     private final Map<ContainerName, MountPointChild> yangLib = new EnumMap<>(ContainerName.class);
     private final List<MountPointChild> children = new ArrayList<>();
 
+    private MountPointChild schemaMounts;
+
     MountPointData(final QName label) {
         super(MountPointIdentifier.of(label));
     }
@@ -56,6 +58,12 @@ public final class MountPointData extends AbstractIdentifiable<MountPointIdentif
         addChild(data);
     }
 
+    public void setSchemaMounts(final @NonNull MountPointChild data) {
+        checkState(schemaMounts == null, "Attempted to reset schema-mounts from %s to %s", schemaMounts, data);
+        schemaMounts = requireNonNull(data);
+        addChild(data);
+    }
+
     public void addChild(final @NonNull MountPointChild data) {
         children.add(requireNonNull(data));
     }