Yang parser refactoring.
[yangtools.git] / yang / yang-parser-impl / src / main / java / org / opendaylight / yangtools / yang / parser / builder / impl / ChoiceBuilder.java
index 05095fe93680b6031f735ec5461e4c9dbf8628ad..e8dd218ec300eefdf81e7284b8f18d3a2f5289dd 100644 (file)
@@ -7,8 +7,10 @@
  */
 package org.opendaylight.yangtools.yang.parser.builder.impl;
 
+import java.net.URI;
 import java.util.ArrayList;
 import java.util.Collections;
+import java.util.Date;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
@@ -19,6 +21,7 @@ import org.opendaylight.yangtools.yang.model.api.AugmentationSchema;
 import org.opendaylight.yangtools.yang.model.api.ChoiceCaseNode;
 import org.opendaylight.yangtools.yang.model.api.ChoiceNode;
 import org.opendaylight.yangtools.yang.model.api.ConstraintDefinition;
+import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
 import org.opendaylight.yangtools.yang.model.api.Status;
 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
@@ -26,96 +29,114 @@ import org.opendaylight.yangtools.yang.parser.builder.api.AbstractSchemaNodeBuil
 import org.opendaylight.yangtools.yang.parser.builder.api.AugmentationSchemaBuilder;
 import org.opendaylight.yangtools.yang.parser.builder.api.AugmentationTargetBuilder;
 import org.opendaylight.yangtools.yang.parser.builder.api.DataSchemaNodeBuilder;
-import org.opendaylight.yangtools.yang.parser.builder.api.GroupingMember;
 import org.opendaylight.yangtools.yang.parser.util.Comparators;
 import org.opendaylight.yangtools.yang.parser.util.ParserUtils;
 import org.opendaylight.yangtools.yang.parser.util.YangParseException;
 
 public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements DataSchemaNodeBuilder,
-        AugmentationTargetBuilder, GroupingMember {
+        AugmentationTargetBuilder {
     private boolean isBuilt;
     private final ChoiceNodeImpl instance;
     // DataSchemaNode args
-    private boolean augmenting;
-    private boolean addedByUses;
-    private Boolean configuration;
     private final ConstraintsBuilder constraints;
     // AugmentationTarget args
-    private final Set<AugmentationSchemaBuilder> addedAugmentations = new HashSet<AugmentationSchemaBuilder>();
+    private final List<AugmentationSchema> augmentations = new ArrayList<>();
+    private final List<AugmentationSchemaBuilder> augmentationBuilders = new ArrayList<>();
     // ChoiceNode args
-    private Set<ChoiceCaseNode> cases;
-    private final Set<ChoiceCaseBuilder> addedCases = new HashSet<ChoiceCaseBuilder>();
+    private Set<ChoiceCaseNode> cases = new TreeSet<>(Comparators.SCHEMA_NODE_COMP);
+    private final Set<ChoiceCaseBuilder> caseBuilders = new HashSet<>();
     private String defaultCase;
 
-    public ChoiceBuilder(final String moduleName, final int line, final QName qname) {
+    public ChoiceBuilder(final String moduleName, final int line, final QName qname, final SchemaPath path) {
         super(moduleName, line, qname);
-        instance = new ChoiceNodeImpl(qname);
+        this.schemaPath = path;
+        instance = new ChoiceNodeImpl(qname, path);
         constraints = new ConstraintsBuilder(moduleName, line);
     }
 
+    public ChoiceBuilder(final String moduleName, final int line, final QName qname, final SchemaPath path,
+            final ChoiceNode base) {
+        super(moduleName, line, qname);
+        this.schemaPath = path;
+        instance = new ChoiceNodeImpl(qname, path);
+        constraints = new ConstraintsBuilder(moduleName, line, base.getConstraints());
+
+        instance.description = base.getDescription();
+        instance.reference = base.getReference();
+        instance.status = base.getStatus();
+        instance.augmenting = base.isAugmenting();
+        instance.addedByUses = base.isAddedByUses();
+        instance.configuration = base.isConfiguration();
+        instance.constraints = base.getConstraints();
+        instance.augmentations.addAll(base.getAvailableAugmentations());
+
+        URI ns = qname.getNamespace();
+        Date rev = qname.getRevision();
+        String pref = qname.getPrefix();
+        Set<DataSchemaNodeBuilder> wrapped = ParserUtils.wrapChildNodes(moduleName, line, new HashSet<DataSchemaNode>(
+                base.getCases()), path, ns, rev, pref);
+        for (DataSchemaNodeBuilder wrap : wrapped) {
+            if (wrap instanceof ChoiceCaseBuilder) {
+                caseBuilders.add((ChoiceCaseBuilder) wrap);
+            }
+        }
+
+        instance.unknownNodes.addAll(base.getUnknownSchemaNodes());
+    }
+
     @Override
     public ChoiceNode build() {
         if (!isBuilt) {
-            instance.setPath(schemaPath);
-            instance.setDescription(description);
-            instance.setReference(reference);
-            instance.setStatus(status);
-            instance.setAugmenting(augmenting);
-            instance.setAddedByUses(addedByUses);
-            instance.setConfiguration(configuration);
             instance.setConstraints(constraints.build());
             instance.setDefaultCase(defaultCase);
 
             // CASES
-            if (cases == null) {
-                cases = new TreeSet<ChoiceCaseNode>(Comparators.SCHEMA_NODE_COMP);
-                for (ChoiceCaseBuilder caseBuilder : addedCases) {
-                    cases.add(caseBuilder.build());
-                }
+            for (ChoiceCaseBuilder caseBuilder : caseBuilders) {
+                cases.add(caseBuilder.build());
             }
-            instance.setCases(cases);
+            instance.addCases(cases);
 
             // AUGMENTATIONS
-            final Set<AugmentationSchema> augmentations = new HashSet<AugmentationSchema>();
-            for (AugmentationSchemaBuilder builder : addedAugmentations) {
+            for (AugmentationSchemaBuilder builder : augmentationBuilders) {
                 augmentations.add(builder.build());
             }
-            instance.setAvailableAugmentations(augmentations);
+            instance.addAvailableAugmentations(new HashSet<>(augmentations));
 
             // UNKNOWN NODES
-            if (unknownNodes == null) {
-                unknownNodes = new ArrayList<UnknownSchemaNode>();
-                for (UnknownSchemaNodeBuilder b : addedUnknownNodes) {
-                    unknownNodes.add(b.build());
-                }
-                Collections.sort(unknownNodes, Comparators.SCHEMA_NODE_COMP);
+            for (UnknownSchemaNodeBuilder b : addedUnknownNodes) {
+                unknownNodes.add(b.build());
             }
-            instance.setUnknownSchemaNodes(unknownNodes);
+            Collections.sort(unknownNodes, Comparators.SCHEMA_NODE_COMP);
+            instance.addUnknownSchemaNodes(unknownNodes);
 
             isBuilt = true;
         }
         return instance;
     }
 
+    public Set<ChoiceCaseBuilder> getCases() {
+        return caseBuilders;
+    }
+
     @Override
-    public void rebuild() {
-        isBuilt = false;
-        build();
+    public SchemaPath getPath() {
+        return instance.path;
     }
 
-    public Set<ChoiceCaseBuilder> getCases() {
-        return addedCases;
+    @Override
+    public void setPath(SchemaPath path) {
+        instance.path = path;
     }
 
     /**
      * Get case by name.
-     * 
+     *
      * @param caseName
      *            name of case to search
      * @return case with given name if present, null otherwise
      */
     public ChoiceCaseBuilder getCaseNodeByName(String caseName) {
-        for (ChoiceCaseBuilder addedCase : addedCases) {
+        for (ChoiceCaseBuilder addedCase : caseBuilders) {
             if (addedCase.getQName().getLocalName().equals(caseName)) {
                 return addedCase;
             }
@@ -125,17 +146,18 @@ public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements Da
 
     /**
      * Add case node to this choice.
-     * 
+     *
      * If node is not declared with 'case' keyword, create new case builder and
      * make this node child of newly created case.
-     * 
+     *
      * @param caseNode
      *            case node
      */
     public void addCase(DataSchemaNodeBuilder caseNode) {
         QName caseQName = caseNode.getQName();
         String caseName = caseQName.getLocalName();
-        for (ChoiceCaseBuilder addedCase : addedCases) {
+
+        for (ChoiceCaseBuilder addedCase : caseBuilders) {
             if (addedCase.getQName().getLocalName().equals(caseName)) {
                 throw new YangParseException(caseNode.getModuleName(), caseNode.getLine(), "Can not add '" + caseNode
                         + "' to node '" + qname.getLocalName() + "' in module '" + moduleName
@@ -144,55 +166,83 @@ public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements Da
         }
 
         if (caseNode instanceof ChoiceCaseBuilder) {
-            addedCases.add((ChoiceCaseBuilder) caseNode);
+            caseBuilders.add((ChoiceCaseBuilder) caseNode);
         } else {
             ChoiceCaseBuilder caseBuilder = new ChoiceCaseBuilder(caseNode.getModuleName(), caseNode.getLine(),
-                    caseQName);
+                    caseQName, caseNode.getPath());
             if (caseNode.isAugmenting()) {
                 // if node is added by augmentation, set case builder augmenting
                 // as true and node augmenting as false
                 caseBuilder.setAugmenting(true);
                 caseNode.setAugmenting(false);
             }
-            caseBuilder.setPath(caseNode.getPath());
             SchemaPath newPath = ParserUtils.createSchemaPath(caseNode.getPath(), caseQName);
             caseNode.setPath(newPath);
             caseBuilder.addChildNode(caseNode);
-            addedCases.add(caseBuilder);
+            caseBuilders.add(caseBuilder);
         }
     }
 
-    public void setCases(Set<ChoiceCaseNode> cases) {
-        this.cases = cases;
+    @Override
+    public String getDescription() {
+        return instance.description;
+    }
+
+    @Override
+    public void setDescription(final String description) {
+        instance.description = description;
+    }
+
+    @Override
+    public String getReference() {
+        return instance.reference;
+    }
+
+    @Override
+    public void setReference(final String reference) {
+        instance.reference = reference;
+    }
+
+    @Override
+    public Status getStatus() {
+        return instance.status;
+    }
+
+    @Override
+    public void setStatus(Status status) {
+        if (status != null) {
+            instance.status = status;
+        }
     }
 
     @Override
     public boolean isAugmenting() {
-        return augmenting;
+        return instance.augmenting;
     }
 
     @Override
     public void setAugmenting(boolean augmenting) {
-        this.augmenting = augmenting;
+        instance.augmenting = augmenting;
     }
 
     @Override
     public boolean isAddedByUses() {
-        return addedByUses;
+        return instance.addedByUses;
     }
 
     @Override
     public void setAddedByUses(final boolean addedByUses) {
-        this.addedByUses = addedByUses;
+        instance.addedByUses = addedByUses;
     }
 
+    @Override
     public Boolean isConfiguration() {
-        return configuration;
+        return instance.configuration;
     }
 
     @Override
     public void setConfiguration(Boolean configuration) {
-        this.configuration = configuration;
+        instance.configuration = configuration;
     }
 
     @Override
@@ -200,17 +250,13 @@ public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements Da
         return constraints;
     }
 
-    public Set<AugmentationSchemaBuilder> getAugmentations() {
-        return addedAugmentations;
-    }
-
     @Override
     public void addAugmentation(AugmentationSchemaBuilder augment) {
-        addedAugmentations.add(augment);
+        augmentationBuilders.add(augment);
     }
 
-    public List<UnknownSchemaNodeBuilder> getUnknownNodes() {
-        return addedUnknownNodes;
+    public List<AugmentationSchemaBuilder> getAugmentationBuilders() {
+        return augmentationBuilders;
     }
 
     public String getDefaultCase() {
@@ -248,11 +294,11 @@ public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements Da
         } else if (!schemaPath.equals(other.schemaPath)) {
             return false;
         }
-        if (parent == null) {
-            if (other.parent != null) {
+        if (parentBuilder == null) {
+            if (other.parentBuilder != null) {
                 return false;
             }
-        } else if (!parent.equals(other.parent)) {
+        } else if (!parentBuilder.equals(other.parentBuilder)) {
             return false;
         }
         return true;
@@ -273,13 +319,14 @@ public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements Da
         private boolean addedByUses;
         private boolean configuration;
         private ConstraintDefinition constraints;
-        private Set<ChoiceCaseNode> cases = Collections.emptySet();
-        private Set<AugmentationSchema> augmentations = Collections.emptySet();
-        private List<UnknownSchemaNode> unknownNodes = Collections.emptyList();
+        private final Set<ChoiceCaseNode> cases = new HashSet<>();
+        private final Set<AugmentationSchema> augmentations = new HashSet<>();
+        private final List<UnknownSchemaNode> unknownNodes = new ArrayList<>();
         private String defaultCase;
 
-        private ChoiceNodeImpl(QName qname) {
+        private ChoiceNodeImpl(QName qname, SchemaPath path) {
             this.qname = qname;
+            this.path = path;
         }
 
         @Override
@@ -292,66 +339,36 @@ public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements Da
             return path;
         }
 
-        private void setPath(SchemaPath path) {
-            this.path = path;
-        }
-
         @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) {
-            if (status != null) {
-                this.status = status;
-            }
-        }
-
         @Override
         public boolean isAugmenting() {
             return augmenting;
         }
 
-        private void setAugmenting(boolean augmenting) {
-            this.augmenting = augmenting;
-        }
-
         @Override
         public boolean isAddedByUses() {
             return addedByUses;
         }
 
-        private void setAddedByUses(boolean addedByUses) {
-            this.addedByUses = addedByUses;
-        }
-
         @Override
         public boolean isConfiguration() {
             return configuration;
         }
 
-        private void setConfiguration(boolean configuration) {
-            this.configuration = configuration;
-        }
-
         @Override
         public ConstraintDefinition getConstraints() {
             return constraints;
@@ -363,29 +380,29 @@ public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements Da
 
         @Override
         public Set<AugmentationSchema> getAvailableAugmentations() {
-            return augmentations;
+            return Collections.unmodifiableSet(augmentations);
         }
 
-        private void setAvailableAugmentations(Set<AugmentationSchema> availableAugmentations) {
+        private void addAvailableAugmentations(Set<AugmentationSchema> availableAugmentations) {
             if (availableAugmentations != null) {
-                this.augmentations = availableAugmentations;
+                this.augmentations.addAll(availableAugmentations);
             }
         }
 
         @Override
         public List<UnknownSchemaNode> getUnknownSchemaNodes() {
-            return unknownNodes;
+            return Collections.unmodifiableList(unknownNodes);
         }
 
-        private void setUnknownSchemaNodes(List<UnknownSchemaNode> unknownSchemaNodes) {
+        private void addUnknownSchemaNodes(List<UnknownSchemaNode> unknownSchemaNodes) {
             if (unknownSchemaNodes != null) {
-                this.unknownNodes = unknownSchemaNodes;
+                this.unknownNodes.addAll(unknownSchemaNodes);
             }
         }
 
         @Override
         public Set<ChoiceCaseNode> getCases() {
-            return cases;
+            return Collections.unmodifiableSet(cases);
         }
 
         @Override
@@ -394,10 +411,8 @@ public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements Da
                 throw new IllegalArgumentException("Choice Case QName cannot be NULL!");
             }
             for (final ChoiceCaseNode caseNode : cases) {
-                if (caseNode != null) {
-                    if (name.equals(caseNode.getQName())) {
-                        return caseNode;
-                    }
+                if (caseNode != null && name.equals(caseNode.getQName())) {
+                    return caseNode;
                 }
             }
             return null;
@@ -409,18 +424,17 @@ public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements Da
                 throw new IllegalArgumentException("Choice Case string Name cannot be NULL!");
             }
             for (final ChoiceCaseNode caseNode : cases) {
-                if (caseNode != null && (caseNode.getQName() != null)) {
-                    if (name.equals(caseNode.getQName().getLocalName())) {
-                        return caseNode;
-                    }
+                if (caseNode != null && (caseNode.getQName() != null)
+                        && name.equals(caseNode.getQName().getLocalName())) {
+                    return caseNode;
                 }
             }
             return null;
         }
 
-        private void setCases(Set<ChoiceCaseNode> cases) {
+        private void addCases(Set<ChoiceCaseNode> cases) {
             if (cases != null) {
-                this.cases = cases;
+                this.cases.addAll(cases);
             }
         }
 
@@ -433,10 +447,6 @@ public final class ChoiceBuilder extends AbstractSchemaNodeBuilder implements Da
             this.defaultCase = defaultCase;
         }
 
-        public ChoiceBuilder toBuilder() {
-            return ChoiceBuilder.this;
-        }
-
         @Override
         public int hashCode() {
             final int prime = 31;