Merge "Added more Rpc markers to yang-binding."
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / builder / impl / ModuleBuilder.java
index df3bb5910a21fbf28b14886bf34ad1087678688a..a15d9db32f5306b76b5bfb12debff1b5f4ffd38d 100644 (file)
@@ -11,13 +11,13 @@ import java.net.URI;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.Date;
+import java.util.Deque;
 import java.util.HashSet;
 import java.util.LinkedHashSet;
 import java.util.LinkedList;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
-import java.util.TreeMap;
 import java.util.TreeSet;
 
 import org.opendaylight.yangtools.yang.common.QName;
@@ -36,6 +36,7 @@ import org.opendaylight.yangtools.yang.model.api.SchemaPath;
 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.UsesNode;
+import org.opendaylight.yangtools.yang.model.api.YangNode;
 import org.opendaylight.yangtools.yang.parser.builder.api.AbstractDataNodeContainerBuilder;
 import org.opendaylight.yangtools.yang.parser.builder.api.AugmentationSchemaBuilder;
 import org.opendaylight.yangtools.yang.parser.builder.api.Builder;
@@ -47,6 +48,7 @@ import org.opendaylight.yangtools.yang.parser.builder.api.TypeAwareBuilder;
 import org.opendaylight.yangtools.yang.parser.builder.api.TypeDefinitionBuilder;
 import org.opendaylight.yangtools.yang.parser.builder.api.UsesNodeBuilder;
 import org.opendaylight.yangtools.yang.parser.util.Comparators;
+import org.opendaylight.yangtools.yang.parser.util.ModuleImportImpl;
 import org.opendaylight.yangtools.yang.parser.util.RefineHolder;
 import org.opendaylight.yangtools.yang.parser.util.YangParseException;
 
@@ -56,6 +58,7 @@ import org.opendaylight.yangtools.yang.parser.util.YangParseException;
  * otherwise result may not be valid.
  */
 public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
+
     private final ModuleImpl instance;
     private final String name;
     private final SchemaPath schemaPath;
@@ -63,133 +66,131 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
     private String prefix;
     private Date revision;
 
-    private final LinkedList<Builder> actualPath = new LinkedList<Builder>();
+    private final Deque<Builder> actualPath = new LinkedList<Builder>();
     private final Set<TypeAwareBuilder> dirtyNodes = new HashSet<TypeAwareBuilder>();
 
     private final Set<ModuleImport> imports = new HashSet<ModuleImport>();
-    private final List<AugmentationSchemaBuilder> addedAugments = new ArrayList<AugmentationSchemaBuilder>();
+    private final Set<AugmentationSchema> augments = new HashSet<>();
+    private final List<AugmentationSchemaBuilder> augmentBuilders = new ArrayList<>();
     private final List<AugmentationSchemaBuilder> allAugments = new ArrayList<AugmentationSchemaBuilder>();
-    private final Set<UsesNodeBuilder> addedUsesNodes = new HashSet<UsesNodeBuilder>();
     private final List<UsesNodeBuilder> allUsesNodes = new ArrayList<UsesNodeBuilder>();
+    private final Set<RpcDefinition> rpcs = new TreeSet<RpcDefinition>(Comparators.SCHEMA_NODE_COMP);
     private final Set<RpcDefinitionBuilder> addedRpcs = new HashSet<RpcDefinitionBuilder>();
+    private final Set<NotificationDefinition> notifications = new TreeSet<NotificationDefinition>(Comparators.SCHEMA_NODE_COMP);
     private final Set<NotificationBuilder> addedNotifications = new HashSet<NotificationBuilder>();
+    private final Set<IdentitySchemaNode> identities = new TreeSet<>(Comparators.SCHEMA_NODE_COMP);
     private final Set<IdentitySchemaNodeBuilder> addedIdentities = new HashSet<IdentitySchemaNodeBuilder>();
-    private final Set<FeatureBuilder> addedFeatures = new HashSet<FeatureBuilder>();
-    private final Set<DeviationBuilder> addedDeviations = new HashSet<DeviationBuilder>();
-    private final Set<TypeDefinitionBuilder> addedTypedefs = new HashSet<TypeDefinitionBuilder>();
-    private final List<ExtensionBuilder> addedExtensions = new ArrayList<ExtensionBuilder>();
+    private final Set<FeatureDefinition> features = new TreeSet<FeatureDefinition>(Comparators.SCHEMA_NODE_COMP);
+    private final Set<FeatureBuilder> addedFeatures = new HashSet<>();
+    private final Set<Deviation> deviations = new HashSet<>();
+    private final Set<DeviationBuilder> deviationBuilders = new HashSet<>();
+    private final List<ExtensionDefinition> extensions = new ArrayList<>();
+    private final List<ExtensionBuilder> addedExtensions = new ArrayList<>();
     private final List<UnknownSchemaNodeBuilder> allUnknownNodes = new ArrayList<UnknownSchemaNodeBuilder>();
 
     public ModuleBuilder(final String name) {
         super(name, 0, null);
         this.name = name;
-        schemaPath = new SchemaPath(Collections.<QName>emptyList(), true);
+        schemaPath = new SchemaPath(Collections.<QName> emptyList(), true);
         instance = new ModuleImpl(name);
         actualPath.push(this);
     }
 
+    public Module build() {
+        return build(null);
+    }
+
     /**
      * Build new Module object based on this builder.
      */
     @Override
-    public Module build() {
+    public Module build(YangNode parent) {
         instance.setPrefix(prefix);
         instance.setRevision(revision);
         instance.setImports(imports);
         instance.setNamespace(namespace);
 
         // TYPEDEFS
-        final Set<TypeDefinition<?>> typedefs = new TreeSet<TypeDefinition<?>>(Comparators.SCHEMA_NODE_COMP);
         for (TypeDefinitionBuilder tdb : addedTypedefs) {
-            typedefs.add(tdb.build());
+            typedefs.add(tdb.build(instance));
         }
         instance.setTypeDefinitions(typedefs);
 
         // CHILD NODES
-        final Map<QName, DataSchemaNode> children = new TreeMap<QName, DataSchemaNode>(Comparators.QNAME_COMP);
         for (DataSchemaNodeBuilder child : addedChildNodes) {
-            children.put(child.getQName(), child.build());
+            DataSchemaNode childNode = child.build(instance);
+            childNodes.put(childNode.getQName(), childNode);
         }
-        instance.setChildNodes(children);
+        instance.setChildNodes(childNodes);
 
         // GROUPINGS
-        final Set<GroupingDefinition> groupings = new TreeSet<GroupingDefinition>(Comparators.SCHEMA_NODE_COMP);
         for (GroupingBuilder gb : addedGroupings) {
-            groupings.add(gb.build());
+            groupings.add(gb.build(instance));
         }
         instance.setGroupings(groupings);
 
         // USES
-        final Set<UsesNode> usesDefinitions = new HashSet<UsesNode>();
         for (UsesNodeBuilder unb : addedUsesNodes) {
-            usesDefinitions.add(unb.build());
+            usesNodes.add(unb.build(instance));
         }
-        instance.setUses(usesDefinitions);
+        instance.setUses(usesNodes);
 
         // FEATURES
-        final Set<FeatureDefinition> features = new TreeSet<FeatureDefinition>(Comparators.SCHEMA_NODE_COMP);
         for (FeatureBuilder fb : addedFeatures) {
-            features.add(fb.build());
+            features.add(fb.build(instance));
         }
         instance.setFeatures(features);
 
         // NOTIFICATIONS
-        final Set<NotificationDefinition> notifications = new TreeSet<NotificationDefinition>(
-                Comparators.SCHEMA_NODE_COMP);
         for (NotificationBuilder entry : addedNotifications) {
-            notifications.add(entry.build());
+            notifications.add(entry.build(instance));
         }
         instance.setNotifications(notifications);
 
         // AUGMENTATIONS
-        final Set<AugmentationSchema> augmentations = new HashSet<AugmentationSchema>();
-        for (AugmentationSchemaBuilder builder : addedAugments) {
-            augmentations.add(builder.build());
+        for (AugmentationSchemaBuilder builder : augmentBuilders) {
+            augments.add(builder.build(instance));
         }
-        instance.setAugmentations(augmentations);
+        instance.setAugmentations(augments);
 
         // RPCs
-        final Set<RpcDefinition> rpcs = new TreeSet<RpcDefinition>(Comparators.SCHEMA_NODE_COMP);
         for (RpcDefinitionBuilder rpc : addedRpcs) {
-            rpcs.add(rpc.build());
+            rpcs.add(rpc.build(instance));
         }
         instance.setRpcs(rpcs);
 
         // DEVIATIONS
-        final Set<Deviation> deviations = new HashSet<Deviation>();
-        for (DeviationBuilder entry : addedDeviations) {
-            deviations.add(entry.build());
+        for (DeviationBuilder entry : deviationBuilders) {
+            deviations.add(entry.build(instance));
         }
         instance.setDeviations(deviations);
 
         // EXTENSIONS
-        final List<ExtensionDefinition> extensions = new ArrayList<ExtensionDefinition>();
         for (ExtensionBuilder eb : addedExtensions) {
-            extensions.add(eb.build());
+            extensions.add(eb.build(instance));
         }
         Collections.sort(extensions, Comparators.SCHEMA_NODE_COMP);
         instance.setExtensionSchemaNodes(extensions);
 
         // IDENTITIES
-        final Set<IdentitySchemaNode> identities = new TreeSet<IdentitySchemaNode>(Comparators.SCHEMA_NODE_COMP);
         for (IdentitySchemaNodeBuilder id : addedIdentities) {
-            identities.add(id.build());
+            identities.add(id.build(instance));
         }
         instance.setIdentities(identities);
 
         // UNKNOWN NODES
-        final List<UnknownSchemaNode> unknownNodes = new ArrayList<UnknownSchemaNode>();
         for (UnknownSchemaNodeBuilder unb : addedUnknownNodes) {
-            unknownNodes.add(unb.build());
+            unknownNodes.add(unb.build(instance));
         }
+        Collections.sort(unknownNodes, Comparators.SCHEMA_NODE_COMP);
         instance.setUnknownSchemaNodes(unknownNodes);
 
         return instance;
     }
 
-    public boolean allUsesLoadDone() {
-        for(UsesNodeBuilder usesNode : allUsesNodes) {
-            if(!usesNode.isLoadDone()) {
+    public boolean isAllUsesDataCollected() {
+        for (UsesNodeBuilder usesNode : allUsesNodes) {
+            if (!usesNode.isDataCollected()) {
                 return false;
             }
         }
@@ -223,7 +224,7 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         if (actualPath.isEmpty()) {
             return null;
         } else {
-            return actualPath.get(0);
+            return actualPath.peekFirst();
         }
     }
 
@@ -231,7 +232,10 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         if (actualPath.size() < 2) {
             return null;
         } else {
-            return actualPath.get(1);
+            Builder builderChild = actualPath.removeFirst();
+            Builder builderParent = actualPath.peekFirst();
+            actualPath.addFirst(builderChild);
+            return builderParent;
         }
     }
 
@@ -252,7 +256,11 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
     }
 
     public Set<DeviationBuilder> getDeviations() {
-        return addedDeviations;
+        return deviationBuilders;
+    }
+
+    public List<ExtensionBuilder> getExtensions() {
+        return addedExtensions;
     }
 
     public List<UnknownSchemaNodeBuilder> getAllUnknownNodes() {
@@ -322,14 +330,19 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
     }
 
     public ExtensionBuilder addExtension(final QName qname, final int line) {
+        Builder parent = getActualNode();
+        if (!(parent.equals(this))) {
+            throw new YangParseException(name, line, "extension can be defined only in module or submodule");
+        }
+
         final String extName = qname.getLocalName();
         for (ExtensionBuilder addedExtension : addedExtensions) {
             if (addedExtension.getQName().getLocalName().equals(extName)) {
-                throw new YangParseException(moduleName, line, "Can not add extension '" + extName
-                        + "': extension with same name already declared at line " + addedExtension.getLine());
+                raiseYangParserException("extension", "node", extName, line, addedExtension.getLine());
             }
         }
         final ExtensionBuilder builder = new ExtensionBuilder(name, line, qname);
+        builder.setParent(parent);
         addedExtensions.add(builder);
         return builder;
     }
@@ -384,8 +397,7 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         if (parent.equals(this)) {
             for (GroupingBuilder addedGrouping : addedGroupings) {
                 if (addedGrouping.getQName().getLocalName().equals(groupingName)) {
-                    throw new YangParseException(name, line, "grouping with same name '" + groupingName
-                            + "' already declared at line " + addedGrouping.getLine());
+                    raiseYangParserException("", "Grouping", groupingName, line, addedGrouping.getLine());
                 }
             }
             addedGroupings.add(builder);
@@ -394,8 +406,7 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
                 DataNodeContainerBuilder parentNode = (DataNodeContainerBuilder) parent;
                 for (GroupingBuilder addedGrouping : parentNode.getGroupingBuilders()) {
                     if (addedGrouping.getQName().getLocalName().equals(groupingName)) {
-                        throw new YangParseException(name, line, "grouping with same name '" + groupingName
-                                + "' already declared at line " + addedGrouping.getLine());
+                        raiseYangParserException("", "Grouping", groupingName, line, addedGrouping.getLine());
                     }
                 }
                 parentNode.addGrouping(builder);
@@ -403,8 +414,7 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
                 RpcDefinitionBuilder parentNode = (RpcDefinitionBuilder) parent;
                 for (GroupingBuilder child : parentNode.getGroupings()) {
                     if (child.getQName().getLocalName().equals(groupingName)) {
-                        throw new YangParseException(name, line, "grouping with same name '" + groupingName
-                                + "' already declared at line " + child.getLine());
+                        raiseYangParserException("", "Grouping", groupingName, line, child.getLine());
                     }
                 }
                 parentNode.addGrouping(builder);
@@ -424,10 +434,15 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
 
         if (parent.equals(this)) {
             // augment can be declared only under 'module' ...
-            addedAugments.add(builder);
+            augmentBuilders.add(builder);
         } else {
             // ... or 'uses' statement
             if (parent instanceof UsesNodeBuilder) {
+                if (augmentTargetStr.startsWith("/")) {
+                    throw new YangParseException(name, line,
+                            "If 'augment' statement is a substatement to the 'uses' statement, it cannot contain absolute path ("
+                                    + augmentTargetStr + ")");
+                }
                 ((UsesNodeBuilder) parent).addAugment(builder);
             } else {
                 throw new YangParseException(name, line, "Augment can be declared only under module or uses statement.");
@@ -438,11 +453,6 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         return builder;
     }
 
-    @Override
-    public Set<UsesNodeBuilder> getUsesNodes() {
-        return addedUsesNodes;
-    }
-
     @Override
     public void addUsesNode(UsesNodeBuilder usesBuilder) {
         addedUsesNodes.add(usesBuilder);
@@ -461,11 +471,13 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
             if (!(parent instanceof DataNodeContainerBuilder)) {
                 throw new YangParseException(name, line, "Unresolved parent of uses '" + groupingPathStr + "'.");
             }
-            if (parent instanceof AugmentationSchemaBuilder) {
-                usesBuilder.setAugmenting(true);
-            }
             ((DataNodeContainerBuilder) parent).addUsesNode(usesBuilder);
         }
+        if(parent instanceof AugmentationSchemaBuilder) {
+            usesBuilder.setAugmenting(true);
+            usesBuilder.setParentAugment((AugmentationSchemaBuilder)parent);
+        }
+
         allUsesNodes.add(usesBuilder);
         return usesBuilder;
     }
@@ -491,20 +503,17 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         String rpcName = qname.getLocalName();
         for (RpcDefinitionBuilder rpc : addedRpcs) {
             if (rpc.getQName().getLocalName().equals(rpcName)) {
-                throw new YangParseException(name, line, "rpc with same name '" + rpcName
-                        + "' already declared at line " + rpc.getLine());
+                raiseYangParserException("", "rpc", rpcName, line, rpc.getLine());
             }
         }
         for (DataSchemaNodeBuilder addedChild : addedChildNodes) {
             if (addedChild.getQName().getLocalName().equals(rpcName)) {
-                throw new YangParseException(name, line, "Can not add rpc: node with same name '" + rpcName
-                        + "' already declared at line " + addedChild.getLine());
+                raiseYangParserException("rpc", "node", rpcName, line, addedChild.getLine());
             }
         }
         for (NotificationBuilder addedNotification : addedNotifications) {
             if (addedNotification.getQName().getLocalName().equals(rpcName)) {
-                throw new YangParseException(name, line, "Can not add rpc: notification with same name '" + rpcName
-                        + "' already declared at line " + addedNotification.getLine());
+                raiseYangParserException("rpc", "notification", rpcName, line, addedNotification.getLine());
             }
         }
         addedRpcs.add(rpcBuilder);
@@ -539,6 +548,10 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         return outputBuilder;
     }
 
+    public void addNotification(NotificationDefinition notification) {
+        notifications.add(notification);
+    }
+
     public NotificationBuilder addNotification(final int line, final QName qname) {
         final Builder parent = getActualNode();
         if (!(parent.equals(this))) {
@@ -548,20 +561,17 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         String notificationName = qname.getLocalName();
         for (NotificationBuilder nb : addedNotifications) {
             if (nb.getQName().equals(qname)) {
-                throw new YangParseException(name, line, "notification with same name '" + notificationName
-                        + "' already declared at line " + nb.getLine());
+                raiseYangParserException("", "notification", notificationName, line, nb.getLine());
             }
         }
         for (RpcDefinitionBuilder rpc : addedRpcs) {
             if (rpc.getQName().getLocalName().equals(notificationName)) {
-                throw new YangParseException(name, line, "Can not add notification: rpc with same name '"
-                        + notificationName + "' already declared at line " + rpc.getLine());
+                raiseYangParserException("notification", "rpc", notificationName, line, rpc.getLine());
             }
         }
         for (DataSchemaNodeBuilder addedChild : addedChildNodes) {
             if (addedChild.getQName().getLocalName().equals(notificationName)) {
-                throw new YangParseException(name, line, "Can not add notification: node with same name '"
-                        + notificationName + "' already declared at line " + addedChild.getLine());
+                raiseYangParserException("notification", "node", notificationName, line, addedChild.getLine());
             }
         }
 
@@ -584,8 +594,7 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         String featureName = qname.getLocalName();
         for (FeatureBuilder addedFeature : addedFeatures) {
             if (addedFeature.getQName().getLocalName().equals(featureName)) {
-                throw new YangParseException(name, line, "feature with same name '" + featureName
-                        + "' already declared at line " + addedFeature.getLine());
+                raiseYangParserException("", "feature", featureName, line, addedFeature.getLine());
             }
         }
         addedFeatures.add(builder);
@@ -637,8 +646,7 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         String nodeName = typedefBuilder.getQName().getLocalName();
         for (TypeDefinitionBuilder tdb : addedTypedefs) {
             if (tdb.getQName().getLocalName().equals(nodeName)) {
-                throw new YangParseException(name, typedefBuilder.getLine(), "typedef with same name '" + nodeName
-                        + "' already declared at line " + tdb.getLine());
+                raiseYangParserException("", "typedef", nodeName, typedefBuilder.getLine(), tdb.getLine());
             }
         }
         addedTypedefs.add(typedefBuilder);
@@ -654,8 +662,7 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         if (parent.equals(this)) {
             for (TypeDefinitionBuilder tdb : addedTypedefs) {
                 if (tdb.getQName().getLocalName().equals(typedefName)) {
-                    throw new YangParseException(name, line, "typedef with same name '" + typedefName
-                            + "' already declared at line " + tdb.getLine());
+                    raiseYangParserException("", "typedef", typedefName, line, tdb.getLine());
                 }
             }
             addedTypedefs.add(builder);
@@ -664,8 +671,7 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
                 DataNodeContainerBuilder parentNode = (DataNodeContainerBuilder) parent;
                 for (TypeDefinitionBuilder child : parentNode.getTypeDefinitionBuilders()) {
                     if (child.getQName().getLocalName().equals(typedefName)) {
-                        throw new YangParseException(name, line, "typedef with same name '" + typedefName
-                                + "' already declared at line " + child.getLine());
+                        raiseYangParserException("", "typedef", typedefName, line, child.getLine());
                     }
                 }
                 parentNode.addTypedef(builder);
@@ -673,8 +679,7 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
                 RpcDefinitionBuilder rpcParent = (RpcDefinitionBuilder) parent;
                 for (TypeDefinitionBuilder tdb : rpcParent.getTypeDefinitions()) {
                     if (tdb.getQName().getLocalName().equals(builder.getQName().getLocalName())) {
-                        throw new YangParseException(name, line, "typedef with same name '" + typedefName
-                                + "' already declared at line " + tdb.getLine());
+                        raiseYangParserException("", "typedef", typedefName, line, tdb.getLine());
                     }
                 }
                 rpcParent.addTypedef(builder);
@@ -688,7 +693,7 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
 
     public void setType(final TypeDefinition<?> type) {
         Builder parent = getActualNode();
-        if (parent == null || !(parent instanceof TypeAwareBuilder)) {
+        if (!(parent instanceof TypeAwareBuilder)) {
             throw new YangParseException("Failed to set type '" + type.getQName().getLocalName()
                     + "'. Invalid parent node: " + parent);
         }
@@ -735,11 +740,11 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
 
         final DeviationBuilder builder = new DeviationBuilder(name, line, targetPath);
         builder.setParent(parent);
-        addedDeviations.add(builder);
+        deviationBuilders.add(builder);
         return builder;
     }
 
-    public IdentitySchemaNodeBuilder addIdentity(final QName qname, final List<String> parentPath, final int line) {
+    public IdentitySchemaNodeBuilder addIdentity(final QName qname, final int line) {
         Builder parent = getActualNode();
         if (!(parent.equals(this))) {
             throw new YangParseException(name, line, "identity can be defined only in module or submodule");
@@ -747,8 +752,7 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         String identityName = qname.getLocalName();
         for (IdentitySchemaNodeBuilder idBuilder : addedIdentities) {
             if (idBuilder.getQName().equals(qname)) {
-                throw new YangParseException(name, line, "identity with same name '" + identityName
-                        + "' already declared at line " + idBuilder.getLine());
+                raiseYangParserException("", "identity", identityName, line, idBuilder.getLine());
             }
         }
 
@@ -788,6 +792,14 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
         return builder;
     }
 
+    public Set<RpcDefinitionBuilder> getRpcs() {
+        return addedRpcs;
+    }
+
+    public Set<NotificationBuilder> getNotifications() {
+        return addedNotifications;
+    }
+
     @Override
     public String toString() {
         return "module " + name;
@@ -821,6 +833,11 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
             this.name = name;
         }
 
+        @Override
+        public YangNode getParent() {
+            return null;
+        }
+
         @Override
         public URI getNamespace() {
             return namespace;
@@ -1144,131 +1161,169 @@ public class ModuleBuilder extends AbstractDataNodeContainerBuilder {
      * @param childName
      */
     private void addChildToParent(final Builder parent, final DataSchemaNodeBuilder child, final String childName) {
-        final int line = child.getLine();
+        final int lineNum = child.getLine();
         if (parent.equals(this)) {
-            // if parent == null => node is defined under module
-            // All leafs, leaf-lists, lists, containers, choices, rpcs,
-            // notifications, and anyxmls defined within a parent node or at the
-            // top level of the module or its submodules share the same
-            // identifier namespace.
-            for (DataSchemaNodeBuilder childNode : addedChildNodes) {
-                if (childNode.getQName().getLocalName().equals(childName)) {
-                    throw new YangParseException(name, line, "Can not add '" + child
-                            + "': node with same name already declared at line " + childNode.getLine());
-                }
-            }
-            for (RpcDefinitionBuilder rpc : addedRpcs) {
-                if (rpc.getQName().getLocalName().equals(childName)) {
-                    throw new YangParseException(name, line, "Can not add '" + child
-                            + "': rpc with same name already declared at line " + rpc.getLine());
-                }
+            addChildToModule(child, childName, lineNum);
+        } else {
+            addChildToSubnodeOfModule(parent, child, childName, lineNum);
+        }
+    }
+
+    /**
+     * Adds child node <code>child</code> to the set of nodes child nodes.
+     *
+     * The method reduces the complexity of the method
+     * {@link #addChildToParent(Builder, DataSchemaNodeBuilder, String)
+     * addChildToParent}.
+     *
+     * @param child
+     *            data schema node builder for child node
+     * @param childName
+     *            string with name of child node
+     * @param lineNum
+     *            line number in YANG file where is the node with the name equal
+     *            to <code>childName</code> is defined
+     */
+    private void addChildToModule(final DataSchemaNodeBuilder child, final String childName, final int lineNum) {
+        // if parent == null => node is defined under module
+        // All leafs, leaf-lists, lists, containers, choices, rpcs,
+        // notifications, and anyxmls defined within a parent node or at the
+        // top level of the module or its submodules share the same
+        // identifier namespace.
+        for (DataSchemaNodeBuilder childNode : addedChildNodes) {
+            if (childNode.getQName().getLocalName().equals(childName)) {
+                raiseYangParserException("'" + child + "'", "node", childName, lineNum, childNode.getLine());
             }
-            for (NotificationBuilder notification : addedNotifications) {
-                if (notification.getQName().getLocalName().equals(childName)) {
-                    throw new YangParseException(name, line, "Can not add '" + child
-                            + "': notification with same name already declared at line " + notification.getLine());
-                }
+        }
+        for (RpcDefinitionBuilder rpc : addedRpcs) {
+            if (rpc.getQName().getLocalName().equals(childName)) {
+                raiseYangParserException("'" + child + "'", "rpc", childName, lineNum, rpc.getLine());
             }
-            addedChildNodes.add(child);
-        } else {
-            if(parent instanceof AugmentationSchemaBuilder) {
-                child.setAugmenting(true);
+        }
+        for (NotificationBuilder notification : addedNotifications) {
+            if (notification.getQName().getLocalName().equals(childName)) {
+                raiseYangParserException("'" + child + "'", "notification", childName, lineNum, notification.getLine());
             }
-            // no need for checking rpc and notification because they can be
-            // defined only under module or submodule
-            if (parent instanceof DataNodeContainerBuilder) {
-                DataNodeContainerBuilder parentNode = (DataNodeContainerBuilder) parent;
-                for (DataSchemaNodeBuilder childNode : parentNode.getChildNodeBuilders()) {
-                    if (childNode.getQName().getLocalName().equals(childName)) {
-                        throw new YangParseException(name, line, "Can not add '" + child + "': node with same name '"
-                                + childName + "' already declared at line " + childNode.getLine());
-                    }
+        }
+        addedChildNodes.add(child);
+    }
+
+    /**
+     * Adds child node <code>child</code> to the group of child nodes of the
+     * <code>parent</code>
+     *
+     * The method reduces the complexity of the method
+     * {@link #addChildToParent(Builder, DataSchemaNodeBuilder, String)
+     * addChildToParent}. *
+     *
+     * @param parent
+     *            builder of node which is parent for <code>child</code>
+     * @param child
+     *            data schema node builder for child node
+     * @param childName
+     *            string with name of child node
+     * @param lineNum
+     *            line number in YANG file where is the node with the name equal
+     *            to <code>childName</code> is defined
+     */
+    private void addChildToSubnodeOfModule(final Builder parent, final DataSchemaNodeBuilder child,
+            final String childName, final int lineNum) {
+        // no need for checking rpc and notification because they can be
+        // defined only under module or submodule
+        if (parent instanceof DataNodeContainerBuilder) {
+            DataNodeContainerBuilder parentNode = (DataNodeContainerBuilder) parent;
+            for (DataSchemaNodeBuilder childNode : parentNode.getChildNodeBuilders()) {
+                if (childNode.getQName().getLocalName().equals(childName)) {
+                    raiseYangParserException("'" + child + "'", "node", childName, lineNum, childNode.getLine());
                 }
-                parentNode.addChildNode(child);
-            } else if (parent instanceof ChoiceBuilder) {
-                ChoiceBuilder parentNode = (ChoiceBuilder) parent;
-                for (ChoiceCaseBuilder caseBuilder : parentNode.getCases()) {
-                    if (caseBuilder.getQName().getLocalName().equals(childName)) {
-                        throw new YangParseException(name, line, "Can not add '" + child + "': case with same name '"
-                                + childName + "' already declared at line " + caseBuilder.getLine());
-                    }
+            }
+            parentNode.addChildNode(child);
+        } else if (parent instanceof ChoiceBuilder) {
+            ChoiceBuilder parentNode = (ChoiceBuilder) parent;
+            for (ChoiceCaseBuilder caseBuilder : parentNode.getCases()) {
+                if (caseBuilder.getQName().getLocalName().equals(childName)) {
+                    raiseYangParserException("'" + child + "'", "node", childName, lineNum, caseBuilder.getLine());
                 }
-                parentNode.addCase(child);
-            } else {
-                throw new YangParseException(name, line, "Unresolved parent of node '" + childName + "'.");
             }
+            parentNode.addCase(child);
+        } else {
+            throw new YangParseException(name, lineNum, "Unresolved parent of node '" + childName + "'.");
         }
     }
 
     private ModuleImport createModuleImport(final String moduleName, final Date revision, final String prefix) {
-        final ModuleImport moduleImport = new ModuleImport() {
-            @Override
-            public String getModuleName() {
-                return moduleName;
-            }
+        final ModuleImport moduleImport = new ModuleImportImpl(moduleName, revision, prefix);
+        return moduleImport;
+    }
 
-            @Override
-            public Date getRevision() {
-                return revision;
-            }
+    private void raiseYangParserException(final String cantAddType, final String type, final String name,
+            final int currentLine, final int duplicateLine) {
 
-            @Override
-            public String getPrefix() {
-                return prefix;
-            }
+        StringBuilder msgPrefix = new StringBuilder("");
+        if (cantAddType != null && !cantAddType.isEmpty()) {
+            msgPrefix.append("Can not add ");
+            msgPrefix.append(cantAddType);
+            msgPrefix.append(": ");
+        }
 
-            @Override
-            public int hashCode() {
-                final int prime = 31;
-                int result = 1;
-                result = prime * result + ((moduleName == null) ? 0 : moduleName.hashCode());
-                result = prime * result + ((revision == null) ? 0 : revision.hashCode());
-                result = prime * result + ((prefix == null) ? 0 : prefix.hashCode());
-                return result;
-            }
+        String msg = String.format("%s%s with same name '%s' already declared at line %d.", msgPrefix, type, name,
+                duplicateLine);
+        throw new YangParseException(moduleName, currentLine, msg);
+    }
 
-            @Override
-            public boolean equals(Object obj) {
-                if (this == obj) {
-                    return true;
-                }
-                if (obj == null) {
-                    return false;
-                }
-                if (getClass() != obj.getClass()) {
-                    return false;
-                }
-                ModuleImport other = (ModuleImport) obj;
-                if (getModuleName() == null) {
-                    if (other.getModuleName() != null) {
-                        return false;
-                    }
-                } else if (!getModuleName().equals(other.getModuleName())) {
-                    return false;
-                }
-                if (getRevision() == null) {
-                    if (other.getRevision() != null) {
-                        return false;
-                    }
-                } else if (!getRevision().equals(other.getRevision())) {
-                    return false;
-                }
-                if (getPrefix() == null) {
-                    if (other.getPrefix() != null) {
-                        return false;
-                    }
-                } else if (!getPrefix().equals(other.getPrefix())) {
-                    return false;
-                }
-                return true;
-            }
+    @Override
+    public int hashCode() {
+        final int prime = 31;
+        int result = 1;
+        result = prime * result + ((name == null) ? 0 : name.hashCode());
+        result = prime * result + ((namespace == null) ? 0 : namespace.hashCode());
+        result = prime * result + ((revision == null) ? 0 : revision.hashCode());
+        result = prime * result + ((prefix == null) ? 0 : prefix.hashCode());
+
+        return result;
+    }
 
-            @Override
-            public String toString() {
-                return "ModuleImport[moduleName=" + moduleName + ", revision=" + revision + ", prefix=" + prefix + "]";
+    @Override
+    public boolean equals(Object obj) {
+        if (this == obj) {
+            return true;
+        }
+        if (obj == null) {
+            return false;
+        }
+        if (getClass() != obj.getClass()) {
+            return false;
+        }
+        ModuleBuilder other = (ModuleBuilder) obj;
+        if (name == null) {
+            if (other.name != null) {
+                return false;
             }
-        };
-        return moduleImport;
+        } else if (!name.equals(other.name)) {
+            return false;
+        }
+        if (namespace == null) {
+            if (other.namespace != null) {
+                return false;
+            }
+        } else if (!namespace.equals(other.namespace)) {
+            return false;
+        }
+        if (prefix == null) {
+            if (other.prefix != null) {
+                return false;
+            }
+        } else if (!prefix.equals(other.prefix)) {
+            return false;
+        }
+        if (revision == null) {
+            if (other.revision != null) {
+                return false;
+            }
+        } else if (!revision.equals(other.revision)) {
+            return false;
+        }
+        return true;
     }
 
 }