Merge "BUG-1275: teach NormalizedNode builders about size hints"
[yangtools.git] / yang / yang-data-impl / src / main / java / org / opendaylight / yangtools / yang / data / impl / schema / builder / impl / ImmutableAugmentationNodeBuilder.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.yangtools.yang.data.impl.schema.builder.impl;
9
10 import java.util.Map;
11
12 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
13 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
14 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
15 import org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode;
16 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
17 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
18 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.valid.DataValidationException;
19 import org.opendaylight.yangtools.yang.data.impl.schema.nodes.AbstractImmutableDataContainerNode;
20
21 public class ImmutableAugmentationNodeBuilder extends AbstractImmutableDataContainerNodeBuilder<YangInstanceIdentifier.AugmentationIdentifier, AugmentationNode> {
22
23     protected ImmutableAugmentationNodeBuilder() {
24         super();
25     }
26
27     protected ImmutableAugmentationNodeBuilder(final int sizeHint) {
28         super(sizeHint);
29     }
30
31     public ImmutableAugmentationNodeBuilder(final ImmutableAugmentationNode node) {
32         super(node);
33     }
34
35     public static DataContainerNodeBuilder<YangInstanceIdentifier.AugmentationIdentifier, AugmentationNode> create() {
36         return new ImmutableAugmentationNodeBuilder();
37     }
38
39     public static DataContainerNodeBuilder<YangInstanceIdentifier.AugmentationIdentifier, AugmentationNode> create(final int sizeHint) {
40         return new ImmutableAugmentationNodeBuilder(sizeHint);
41     }
42
43     public static DataContainerNodeBuilder<YangInstanceIdentifier.AugmentationIdentifier, AugmentationNode> create(final AugmentationNode node) {
44         if (!(node instanceof ImmutableAugmentationNode)) {
45             throw new UnsupportedOperationException(String.format("Cannot initialize from class %s", node.getClass()));
46         }
47
48         return new ImmutableAugmentationNodeBuilder((ImmutableAugmentationNode)node);
49     }
50
51     @Override
52     public DataContainerNodeBuilder<YangInstanceIdentifier.AugmentationIdentifier, AugmentationNode> withChild(
53             final DataContainerChild<?, ?> child) {
54         // Check nested augments
55         DataValidationException.checkLegalData(child instanceof AugmentationNode == false,
56                 "Unable to add: %s, as a child for: %s, Nested augmentations are not permitted", child.getNodeType(),
57                 getNodeIdentifier() == null ? this : getNodeIdentifier());
58
59         return super.withChild(child);
60     }
61
62     @Override
63     public DataContainerNodeBuilder<AugmentationIdentifier, AugmentationNode> withoutChild(final PathArgument key) {
64         return super.withoutChild(key);
65     }
66
67     @Override
68     public AugmentationNode build() {
69         return new ImmutableAugmentationNode(getNodeIdentifier(), buildValue());
70     }
71
72     private static final class ImmutableAugmentationNode extends AbstractImmutableDataContainerNode<YangInstanceIdentifier.AugmentationIdentifier> implements AugmentationNode {
73
74         ImmutableAugmentationNode(final YangInstanceIdentifier.AugmentationIdentifier nodeIdentifier, final Map<YangInstanceIdentifier.PathArgument, DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?>> children) {
75             super(children, nodeIdentifier);
76         }
77     }
78
79 }