Refactored parsing of yang uses statement.

[yangtools.git] / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / builder / impl / AugmentationSchemaBuilderImpl.java

diff --git a/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/builder/impl/AugmentationSchemaBuilderImpl.java b/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/builder/impl/AugmentationSchemaBuilderImpl.java
index a7a9d2b4667ad3bbbbd3a26dfb428d0b70865583..ef7b28b6fd1e66e73d984cd606ba67c1736e8cc5 100644 (file)
--- a/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/builder/impl/AugmentationSchemaBuilderImpl.java
+++ b/yang-parser-impl/src/main/java/org/opendaylight/yangtools/yang/parser/builder/impl/AugmentationSchemaBuilderImpl.java
@@ -93,6 +93,12 @@ public final class AugmentationSchemaBuilderImpl extends AbstractDataNodeContain
     @Override
     public AugmentationSchema build() {
         if (!built) {
+            // process uses
+            for(UsesNodeBuilder use : usesNodes) {
+                addedChildNodes.addAll(use.getTargetChildren());
+                addedUnknownNodes.addAll(use.getTargetUnknownNodes());
+            }
+
             instance.setDescription(description);
             instance.setReference(reference);
             instance.setStatus(status);