Added more tests for yang parser. Updated current tests.
[controller.git] / opendaylight / sal / yang-prototype / code-generator / yang-model-parser-impl / src / main / java / org / opendaylight / controller / yang / parser / builder / impl / GroupingBuilderImpl.java
1 /*
2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.controller.yang.parser.builder.impl;
9
10 import java.util.ArrayList;
11 import java.util.Collections;
12 import java.util.HashMap;
13 import java.util.HashSet;
14 import java.util.List;
15 import java.util.Map;
16 import java.util.Set;
17 import java.util.TreeSet;
18
19 import org.opendaylight.controller.yang.common.QName;
20 import org.opendaylight.controller.yang.model.api.DataSchemaNode;
21 import org.opendaylight.controller.yang.model.api.GroupingDefinition;
22 import org.opendaylight.controller.yang.model.api.SchemaPath;
23 import org.opendaylight.controller.yang.model.api.Status;
24 import org.opendaylight.controller.yang.model.api.TypeDefinition;
25 import org.opendaylight.controller.yang.model.api.UnknownSchemaNode;
26 import org.opendaylight.controller.yang.model.api.UsesNode;
27 import org.opendaylight.controller.yang.parser.builder.api.Builder;
28 import org.opendaylight.controller.yang.parser.builder.api.DataSchemaNodeBuilder;
29 import org.opendaylight.controller.yang.parser.builder.api.GroupingBuilder;
30 import org.opendaylight.controller.yang.parser.builder.api.TypeDefinitionBuilder;
31 import org.opendaylight.controller.yang.parser.builder.api.UsesNodeBuilder;
32 import org.opendaylight.controller.yang.parser.util.Comparators;
33
34 public final class GroupingBuilderImpl implements GroupingBuilder {
35     private Builder parent;
36     private boolean isBuilt;
37     private final GroupingDefinitionImpl instance;
38     private final int line;
39     private final QName qname;
40     private SchemaPath schemaPath;
41     private String description;
42     private String reference;
43     private Status status = Status.CURRENT;
44     private boolean addedByUses;
45
46     private Set<DataSchemaNode> childNodes;
47     private final Set<DataSchemaNodeBuilder> addedChildNodes = new HashSet<DataSchemaNodeBuilder>();
48
49     private Set<GroupingDefinition> groupings;
50     private final Set<GroupingBuilder> addedGroupings = new HashSet<GroupingBuilder>();
51
52     private Set<TypeDefinition<?>> typedefs;
53     private final Set<TypeDefinitionBuilder> addedTypedefs = new HashSet<TypeDefinitionBuilder>();
54
55     private Set<UsesNode> usesNodes;
56     private final Set<UsesNodeBuilder> addedUsesNodes = new HashSet<UsesNodeBuilder>();
57
58     private List<UnknownSchemaNode> unknownNodes;
59     private final List<UnknownSchemaNodeBuilder> addedUnknownNodes = new ArrayList<UnknownSchemaNodeBuilder>();
60
61     public GroupingBuilderImpl(final QName qname, final int line) {
62         this.qname = qname;
63         instance = new GroupingDefinitionImpl(qname);
64         this.line = line;
65     }
66
67     public GroupingBuilderImpl(GroupingBuilder builder) {
68         qname = builder.getQName();
69         parent = builder.getParent();
70         instance = new GroupingDefinitionImpl(qname);
71         line = builder.getLine();
72         schemaPath = builder.getPath();
73         description = builder.getDescription();
74         reference = builder.getReference();
75         status = builder.getStatus();
76         addedByUses = builder.isAddedByUses();
77         childNodes = builder.getChildNodes();
78         addedChildNodes.addAll(builder.getChildNodeBuilders());
79         groupings = builder.getGroupings();
80         addedGroupings.addAll(builder.getGroupingBuilders());
81         addedUsesNodes.addAll(builder.getUses());
82         addedUnknownNodes.addAll(builder.getUnknownNodes());
83     }
84
85     @Override
86     public GroupingDefinition build() {
87         if (!isBuilt) {
88             instance.setPath(schemaPath);
89             instance.setDescription(description);
90             instance.setReference(reference);
91             instance.setStatus(status);
92             instance.setAddedByUses(addedByUses);
93
94             // CHILD NODES
95             final Map<QName, DataSchemaNode> childs = new HashMap<QName, DataSchemaNode>();
96             if (childNodes == null) {
97                 for (DataSchemaNodeBuilder node : addedChildNodes) {
98                     childs.put(node.getQName(), node.build());
99                 }
100             } else {
101                 for (DataSchemaNode node : childNodes) {
102                     childs.put(node.getQName(), node);
103                 }
104             }
105             instance.setChildNodes(childs);
106
107             // GROUPINGS
108             if (groupings == null) {
109                 groupings = new TreeSet<GroupingDefinition>(Comparators.SCHEMA_NODE_COMP);
110                 for (GroupingBuilder builder : addedGroupings) {
111                     groupings.add(builder.build());
112                 }
113             }
114             instance.setGroupings(groupings);
115
116             // TYPEDEFS
117             if (typedefs == null) {
118                 typedefs = new TreeSet<TypeDefinition<?>>(Comparators.SCHEMA_NODE_COMP);
119                 for (TypeDefinitionBuilder entry : addedTypedefs) {
120                     typedefs.add(entry.build());
121                 }
122             }
123             instance.setTypeDefinitions(typedefs);
124
125             // USES
126             if (usesNodes == null) {
127                 usesNodes = new HashSet<UsesNode>();
128                 for (UsesNodeBuilder builder : addedUsesNodes) {
129                     usesNodes.add(builder.build());
130                 }
131             }
132             instance.setUses(usesNodes);
133
134             // UNKNOWN NODES
135             if (unknownNodes == null) {
136                 unknownNodes = new ArrayList<UnknownSchemaNode>();
137                 for (UnknownSchemaNodeBuilder b : addedUnknownNodes) {
138                     unknownNodes.add(b.build());
139                 }
140                 Collections.sort(unknownNodes, Comparators.SCHEMA_NODE_COMP);
141             }
142             instance.setUnknownSchemaNodes(unknownNodes);
143
144             isBuilt = true;
145         }
146
147         return instance;
148     }
149
150     @Override
151     public int getLine() {
152         return line;
153     }
154
155     @Override
156     public Builder getParent() {
157         return parent;
158     }
159
160     @Override
161     public void setParent(final Builder parent) {
162         this.parent = parent;
163     }
164
165     @Override
166     public QName getQName() {
167         return qname;
168     }
169
170     @Override
171     public Set<TypeDefinitionBuilder> getTypeDefinitionBuilders() {
172         return addedTypedefs;
173     }
174
175     @Override
176     public void addTypedef(final TypeDefinitionBuilder type) {
177         addedTypedefs.add(type);
178     }
179
180     public void setTypedefs(final Set<TypeDefinition<?>> typedefs) {
181         this.typedefs = typedefs;
182     }
183
184     @Override
185     public SchemaPath getPath() {
186         return schemaPath;
187     }
188
189     @Override
190     public void setPath(SchemaPath schemaPath) {
191         this.schemaPath = schemaPath;
192     }
193
194     @Override
195     public String getDescription() {
196         return description;
197     }
198
199     @Override
200     public void setDescription(final String description) {
201         this.description = description;
202     }
203
204     @Override
205     public String getReference() {
206         return reference;
207     }
208
209     @Override
210     public void setReference(final String reference) {
211         this.reference = reference;
212     }
213
214     @Override
215     public Status getStatus() {
216         return status;
217     }
218
219     @Override
220     public void setStatus(final Status status) {
221         this.status = status;
222     }
223
224     @Override
225     public boolean isAddedByUses() {
226         return addedByUses;
227     }
228
229     @Override
230     public void setAddedByUses(final boolean addedByUses) {
231         this.addedByUses = addedByUses;
232     }
233
234     @Override
235     public DataSchemaNodeBuilder getChildNode(String name) {
236         DataSchemaNodeBuilder result = null;
237         for (DataSchemaNodeBuilder node : addedChildNodes) {
238             if (node.getQName().getLocalName().equals(name)) {
239                 result = node;
240                 break;
241             }
242         }
243         return result;
244     }
245
246     @Override
247     public Set<DataSchemaNode> getChildNodes() {
248         return childNodes;
249     }
250
251     @Override
252     public void addChildNode(final DataSchemaNodeBuilder childNode) {
253         addedChildNodes.add(childNode);
254     }
255
256     @Override
257     public Set<DataSchemaNodeBuilder> getChildNodeBuilders() {
258         return addedChildNodes;
259     }
260
261     @Override
262     public DataSchemaNodeBuilder getDataChildByName(final String name) {
263         for(DataSchemaNodeBuilder child : addedChildNodes) {
264             if(child.getQName().getLocalName().equals(name)) {
265                 return child;
266             }
267         }
268         return null;
269     }
270
271     public void setChildNodes(final Set<DataSchemaNode> childNodes) {
272         this.childNodes = childNodes;
273     }
274
275     @Override
276     public Set<GroupingDefinition> getGroupings() {
277         return Collections.emptySet();
278     }
279
280     @Override
281     public Set<GroupingBuilder> getGroupingBuilders() {
282         return addedGroupings;
283     }
284
285     @Override
286     public void addGrouping(final GroupingBuilder grouping) {
287         addedGroupings.add(grouping);
288     }
289
290     public void setGroupings(final Set<GroupingDefinition> groupings) {
291         this.groupings = groupings;
292     }
293
294     @Override
295     public Set<UsesNodeBuilder> getUses() {
296         return addedUsesNodes;
297     }
298
299     @Override
300     public void addUsesNode(final UsesNodeBuilder usesBuilder) {
301         addedUsesNodes.add(usesBuilder);
302     }
303
304     public void setUsesnodes(final Set<UsesNode> usesNodes) {
305         this.usesNodes = usesNodes;
306     }
307
308     @Override
309     public List<UnknownSchemaNodeBuilder> getUnknownNodes() {
310         return addedUnknownNodes;
311     }
312
313     @Override
314     public void addUnknownSchemaNode(final UnknownSchemaNodeBuilder unknownNode) {
315         addedUnknownNodes.add(unknownNode);
316     }
317
318     public void setUnknownNodes(List<UnknownSchemaNode> unknownNodes) {
319         this.unknownNodes = unknownNodes;
320     }
321
322     private final class GroupingDefinitionImpl implements GroupingDefinition {
323         private final QName qname;
324         private SchemaPath path;
325         private String description;
326         private String reference;
327         private Status status;
328         private boolean addedByUses;
329         private Map<QName, DataSchemaNode> childNodes = Collections.emptyMap();
330         private Set<GroupingDefinition> groupings = Collections.emptySet();
331         private Set<TypeDefinition<?>> typeDefinitions = Collections.emptySet();
332         private Set<UsesNode> uses = Collections.emptySet();
333         private List<UnknownSchemaNode> unknownNodes = Collections.emptyList();
334
335         private GroupingDefinitionImpl(final QName qname) {
336             this.qname = qname;
337         }
338
339         @Override
340         public QName getQName() {
341             return qname;
342         }
343
344         @Override
345         public SchemaPath getPath() {
346             return path;
347         }
348
349         private void setPath(SchemaPath path) {
350             this.path = path;
351         }
352
353         @Override
354         public String getDescription() {
355             return description;
356         }
357
358         private void setDescription(String description) {
359             this.description = description;
360         }
361
362         @Override
363         public String getReference() {
364             return reference;
365         }
366
367         private void setReference(String reference) {
368             this.reference = reference;
369         }
370
371         @Override
372         public Status getStatus() {
373             return status;
374         }
375
376         private void setStatus(Status status) {
377             this.status = status;
378         }
379
380         @Override
381         public boolean isAddedByUses() {
382             return addedByUses;
383         }
384
385         private void setAddedByUses(final boolean addedByUses) {
386             this.addedByUses = addedByUses;
387         }
388
389         @Override
390         public Set<DataSchemaNode> getChildNodes() {
391             final Set<DataSchemaNode> result = new TreeSet<DataSchemaNode>(Comparators.SCHEMA_NODE_COMP);
392             result.addAll(childNodes.values());
393             return result;
394         }
395
396         private void setChildNodes(Map<QName, DataSchemaNode> childNodes) {
397             this.childNodes = childNodes;
398         }
399
400         @Override
401         public Set<GroupingDefinition> getGroupings() {
402             return groupings;
403         }
404
405         private void setGroupings(Set<GroupingDefinition> groupings) {
406             this.groupings = groupings;
407         }
408
409         @Override
410         public Set<UsesNode> getUses() {
411             return uses;
412         }
413
414         private void setUses(Set<UsesNode> uses) {
415             this.uses = uses;
416         }
417
418         @Override
419         public Set<TypeDefinition<?>> getTypeDefinitions() {
420             return typeDefinitions;
421         }
422
423         private void setTypeDefinitions(Set<TypeDefinition<?>> typeDefinitions) {
424             this.typeDefinitions = typeDefinitions;
425         }
426
427         @Override
428         public List<UnknownSchemaNode> getUnknownSchemaNodes() {
429             return unknownNodes;
430         }
431
432         private void setUnknownSchemaNodes(List<UnknownSchemaNode> unknownNodes) {
433             if (unknownNodes != null) {
434                 this.unknownNodes = unknownNodes;
435             }
436         }
437
438         @Override
439         public DataSchemaNode getDataChildByName(QName name) {
440             return childNodes.get(name);
441         }
442
443         @Override
444         public DataSchemaNode getDataChildByName(String name) {
445             DataSchemaNode result = null;
446             for (Map.Entry<QName, DataSchemaNode> entry : childNodes.entrySet()) {
447                 if (entry.getKey().getLocalName().equals(name)) {
448                     result = entry.getValue();
449                     break;
450                 }
451             }
452             return result;
453         }
454
455         @Override
456         public int hashCode() {
457             final int prime = 31;
458             int result = 1;
459             result = prime * result + ((qname == null) ? 0 : qname.hashCode());
460             result = prime * result + ((path == null) ? 0 : path.hashCode());
461             return result;
462         }
463
464         @Override
465         public boolean equals(Object obj) {
466             if (this == obj) {
467                 return true;
468             }
469             if (obj == null) {
470                 return false;
471             }
472             if (getClass() != obj.getClass()) {
473                 return false;
474             }
475             final GroupingDefinitionImpl other = (GroupingDefinitionImpl) obj;
476             if (qname == null) {
477                 if (other.qname != null) {
478                     return false;
479                 }
480             } else if (!qname.equals(other.qname)) {
481                 return false;
482             }
483             if (path == null) {
484                 if (other.path != null) {
485                     return false;
486                 }
487             } else if (!path.equals(other.path)) {
488                 return false;
489             }
490             return true;
491         }
492
493         @Override
494         public String toString() {
495             StringBuilder sb = new StringBuilder(GroupingDefinitionImpl.class.getSimpleName());
496             sb.append("[");
497             sb.append("qname=" + qname);
498             sb.append("]");
499             return sb.toString();
500         }
501     }
502
503 }