BUG-580: Improved parsing.
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / builder / impl / GroupingBuilderImpl.java
index c4f6a23ec65fd11eea4f382d1ee08bc016e8bb17..90a49d1b196f30986c11d84713e3e4eb53a637b6 100644 (file)
@@ -7,11 +7,11 @@
  */
 package org.opendaylight.yangtools.yang.parser.builder.impl;
 
-import java.util.Collections;
+import java.net.URI;
+import java.util.Date;
+import java.util.HashSet;
 import java.util.List;
-import java.util.Map;
 import java.util.Set;
-import java.util.TreeSet;
 
 import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
@@ -21,81 +21,147 @@ import org.opendaylight.yangtools.yang.model.api.Status;
 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.Builder;
 import org.opendaylight.yangtools.yang.parser.builder.api.DataSchemaNodeBuilder;
 import org.opendaylight.yangtools.yang.parser.builder.api.GroupingBuilder;
 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.CopyUtils;
+import org.opendaylight.yangtools.yang.parser.util.ParserUtils;
 import org.opendaylight.yangtools.yang.parser.util.YangParseException;
 
+import com.google.common.base.Preconditions;
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableSet;
+
 public final class GroupingBuilderImpl extends AbstractDataNodeContainerBuilder implements GroupingBuilder {
-    private boolean isBuilt;
-    private final GroupingDefinitionImpl instance;
+    private GroupingDefinitionImpl instance;
+    // SchemaNode args
     private SchemaPath schemaPath;
     private String description;
     private String reference;
     private Status status = Status.CURRENT;
+    // DataSchemaNode args
     private boolean addedByUses;
 
-    public GroupingBuilderImpl(final String moduleName, final int line, final QName qname) {
+    public GroupingBuilderImpl(final String moduleName, final int line, final QName qname, final SchemaPath path) {
         super(moduleName, line, qname);
-        instance = new GroupingDefinitionImpl(qname);
+        schemaPath = path;
+    }
+
+    public GroupingBuilderImpl(final String moduleName, final int line, final QName qname, final SchemaPath path,
+            final GroupingDefinition base) {
+        super(moduleName, line, base.getQName());
+        schemaPath = path;
+
+        description = base.getDescription();
+        reference = base.getReference();
+        status = base.getStatus();
+        addedByUses = base.isAddedByUses();
+
+        URI ns = qname.getNamespace();
+        Date rev = qname.getRevision();
+        String pref = qname.getPrefix();
+        addedChildNodes.addAll(ParserUtils.wrapChildNodes(moduleName, line, base.getChildNodes(), path, ns, rev, pref));
+        addedGroupings.addAll(ParserUtils.wrapGroupings(moduleName, line, base.getGroupings(), path, ns, rev, pref));
+        addedTypedefs.addAll(ParserUtils.wrapTypedefs(moduleName, line, base, path, ns, rev, pref));
+        addedUnknownNodes.addAll(ParserUtils.wrapUnknownNodes(moduleName, line, base.getUnknownSchemaNodes(), path, ns,
+                rev, pref));
+
+        usesNodes.addAll(base.getUses());
     }
 
     @Override
-    public GroupingDefinition build(YangNode parent) {
-        if (!isBuilt) {
-            instance.setParent(parent);
-            instance.setPath(schemaPath);
-            instance.setDescription(description);
-            instance.setReference(reference);
-            instance.setStatus(status);
-            instance.setAddedByUses(addedByUses);
-
-            // CHILD NODES
-            for (DataSchemaNodeBuilder node : addedChildNodes) {
-                DataSchemaNode child = node.build(instance);
-                childNodes.put(child.getQName(), child);
-            }
-            instance.setChildNodes(childNodes);
+    public GroupingDefinition build() {
+        if (instance != null) {
+            return instance;
+        }
 
-            // GROUPINGS
-            for (GroupingBuilder builder : addedGroupings) {
-                groupings.add(builder.build(instance));
-            }
-            instance.setGroupings(groupings);
+        instance = new GroupingDefinitionImpl(qname, schemaPath);
 
-            // TYPEDEFS
-            for (TypeDefinitionBuilder entry : addedTypedefs) {
-                typedefs.add(entry.build(instance));
-            }
-            instance.setTypeDefinitions(typedefs);
+        instance.description = description;
+        instance.reference = reference;
+        instance.status = status;
+        instance.addedByUses = addedByUses;
 
-            // USES
-            for (UsesNodeBuilder builder : addedUsesNodes) {
-                usesNodes.add(builder.build(instance));
-            }
-            instance.setUses(usesNodes);
+        // CHILD NODES
+        for (DataSchemaNodeBuilder node : addedChildNodes) {
+            childNodes.add(node.build());
+        }
+        instance.childNodes = ImmutableSet.copyOf(childNodes);
 
-            // UNKNOWN NODES
-            for (UnknownSchemaNodeBuilder b : addedUnknownNodes) {
-                unknownNodes.add(b.build(instance));
-            }
-            Collections.sort(unknownNodes, Comparators.SCHEMA_NODE_COMP);
-            instance.setUnknownSchemaNodes(unknownNodes);
+        // GROUPINGS
+        for (GroupingBuilder builder : addedGroupings) {
+            groupings.add(builder.build());
+        }
+        instance.groupings = ImmutableSet.copyOf(groupings);
+
+        // TYPEDEFS
+        for (TypeDefinitionBuilder entry : addedTypedefs) {
+            typedefs.add(entry.build());
+        }
+        instance.typeDefinitions = ImmutableSet.copyOf(typedefs);
+
+        // USES
+        for (UsesNodeBuilder builder : addedUsesNodes) {
+            usesNodes.add(builder.build());
+        }
+        instance.uses = ImmutableSet.copyOf(usesNodes);
 
-            isBuilt = true;
+        // UNKNOWN NODES
+        for (UnknownSchemaNodeBuilder b : addedUnknownNodes) {
+            unknownNodes.add(b.build());
         }
+        instance.unknownNodes = ImmutableList.copyOf(unknownNodes);
 
         return instance;
     }
 
+    @Override
+    public Set<DataSchemaNodeBuilder> instantiateChildNodes(Builder newParent) {
+        final Set<DataSchemaNodeBuilder> nodes = new HashSet<>();
+        for (DataSchemaNodeBuilder node : addedChildNodes) {
+            DataSchemaNodeBuilder copy = CopyUtils.copy(node, newParent, true);
+            ParserUtils.setNodeAddedByUses(copy);
+            nodes.add(copy);
+        }
+        return nodes;
+    }
+
+    @Override
+    public Set<TypeDefinitionBuilder> instantiateTypedefs(Builder newParent) {
+        final Set<TypeDefinitionBuilder> nodes = new HashSet<>();
+        for (TypeDefinitionBuilder node : addedTypedefs) {
+            TypeDefinitionBuilder copy = CopyUtils.copy(node, newParent, true);
+            nodes.add(copy);
+        }
+        return nodes;
+    }
 
     @Override
-    public void setQName(QName qname) {
-        this.qname = qname;
+    public Set<GroupingBuilder> instantiateGroupings(Builder newParent) {
+        final Set<GroupingBuilder> nodes = new HashSet<>();
+        for (GroupingBuilder node : addedGroupings) {
+            GroupingBuilder copy = CopyUtils.copy(node, newParent, true);
+            copy.setAddedByUses(true);
+            for (DataSchemaNodeBuilder childNode : copy.getChildNodeBuilders()) {
+                ParserUtils.setNodeAddedByUses(childNode);
+            }
+            nodes.add(copy);
+        }
+        return nodes;
+    }
+
+    @Override
+    public Set<UnknownSchemaNodeBuilder> instantiateUnknownNodes(Builder newParent) {
+        final Set<UnknownSchemaNodeBuilder> nodes = new HashSet<>();
+        for (UnknownSchemaNodeBuilder node : addedUnknownNodes) {
+            UnknownSchemaNodeBuilder copy = CopyUtils.copy(node, newParent, true);
+            copy.setAddedByUses(true);
+            nodes.add(copy);
+        }
+        return nodes;
     }
 
     @Override
@@ -113,18 +179,14 @@ public final class GroupingBuilderImpl extends AbstractDataNodeContainerBuilder
         addedTypedefs.add(type);
     }
 
-    public void setTypedefs(final Set<TypeDefinition<?>> typedefs) {
-        this.typedefs = typedefs;
-    }
-
     @Override
     public SchemaPath getPath() {
         return schemaPath;
     }
 
     @Override
-    public void setPath(SchemaPath schemaPath) {
-        this.schemaPath = schemaPath;
+    public void setPath(final SchemaPath path) {
+        this.schemaPath = path;
     }
 
     @Override
@@ -153,8 +215,8 @@ public final class GroupingBuilderImpl extends AbstractDataNodeContainerBuilder
     }
 
     @Override
-    public void setStatus(final Status status) {
-        this.status = status;
+    public void setStatus(Status status) {
+        this.status = Preconditions.checkNotNull(status, "status cannot be null");
     }
 
     @Override
@@ -213,23 +275,22 @@ public final class GroupingBuilderImpl extends AbstractDataNodeContainerBuilder
         return true;
     }
 
-
-    private final class GroupingDefinitionImpl implements GroupingDefinition {
+    private static final class GroupingDefinitionImpl implements GroupingDefinition {
         private final QName qname;
-        private SchemaPath path;
-        private YangNode parent;
+        private final SchemaPath path;
         private String description;
         private String reference;
         private Status status;
         private boolean addedByUses;
-        private Map<QName, DataSchemaNode> childNodes = Collections.emptyMap();
-        private Set<GroupingDefinition> groupings = Collections.emptySet();
-        private Set<TypeDefinition<?>> typeDefinitions = Collections.emptySet();
-        private Set<UsesNode> uses = Collections.emptySet();
-        private List<UnknownSchemaNode> unknownNodes = Collections.emptyList();
+        private ImmutableSet<DataSchemaNode> childNodes;
+        private ImmutableSet<GroupingDefinition> groupings;
+        private ImmutableSet<TypeDefinition<?>> typeDefinitions;
+        private ImmutableSet<UsesNode> uses;
+        private ImmutableList<UnknownSchemaNode> unknownNodes;
 
-        private GroupingDefinitionImpl(final QName qname) {
+        private GroupingDefinitionImpl(final QName qname, final SchemaPath path) {
             this.qname = qname;
+            this.path = path;
         }
 
         @Override
@@ -242,64 +303,29 @@ public final class GroupingBuilderImpl extends AbstractDataNodeContainerBuilder
             return path;
         }
 
-        private void setPath(SchemaPath path) {
-            this.path = path;
-        }
-
-        @Override
-        public YangNode getParent() {
-            return parent;
-        }
-
-        private void setParent(YangNode parent) {
-            this.parent = parent;
-        }
-
         @Override
         public String getDescription() {
             return description;
         }
 
-        private void setDescription(String description) {
-            this.description = description;
-        }
-
         @Override
         public String getReference() {
             return reference;
         }
 
-        private void setReference(String reference) {
-            this.reference = reference;
-        }
-
         @Override
         public Status getStatus() {
             return status;
         }
 
-        private void setStatus(Status status) {
-            this.status = status;
-        }
-
         @Override
         public boolean isAddedByUses() {
             return addedByUses;
         }
 
-        private void setAddedByUses(final boolean addedByUses) {
-            this.addedByUses = addedByUses;
-        }
-
         @Override
         public Set<DataSchemaNode> getChildNodes() {
-            final Set<DataSchemaNode> result = new TreeSet<DataSchemaNode>(Comparators.SCHEMA_NODE_COMP);
-            result.addAll(childNodes.values());
-            return result;
-        }
-
-        private void setChildNodes(Map<QName, DataSchemaNode> childNodes) {
-            this.childNodes = childNodes;
+            return childNodes;
         }
 
         @Override
@@ -307,54 +333,29 @@ public final class GroupingBuilderImpl extends AbstractDataNodeContainerBuilder
             return groupings;
         }
 
-        private void setGroupings(Set<GroupingDefinition> groupings) {
-            this.groupings = groupings;
-        }
-
         @Override
         public Set<UsesNode> getUses() {
             return uses;
         }
 
-        private void setUses(Set<UsesNode> uses) {
-            this.uses = uses;
-        }
-
         @Override
         public Set<TypeDefinition<?>> getTypeDefinitions() {
             return typeDefinitions;
         }
 
-        private void setTypeDefinitions(Set<TypeDefinition<?>> typeDefinitions) {
-            this.typeDefinitions = typeDefinitions;
-        }
-
         @Override
         public List<UnknownSchemaNode> getUnknownSchemaNodes() {
             return unknownNodes;
         }
 
-        private void setUnknownSchemaNodes(List<UnknownSchemaNode> unknownNodes) {
-            if (unknownNodes != null) {
-                this.unknownNodes = unknownNodes;
-            }
-        }
-
         @Override
         public DataSchemaNode getDataChildByName(QName name) {
-            return childNodes.get(name);
+            return getChildNode(childNodes, name);
         }
 
         @Override
         public DataSchemaNode getDataChildByName(String name) {
-            DataSchemaNode result = null;
-            for (Map.Entry<QName, DataSchemaNode> entry : childNodes.entrySet()) {
-                if (entry.getKey().getLocalName().equals(name)) {
-                    result = entry.getValue();
-                    break;
-                }
-            }
-            return result;
+            return getChildNode(childNodes, name);
         }
 
         @Override