Fix mandatory enforcer failure on augmented nodes
[yangtools.git] / yang / yang-data-impl / src / main / java / org / opendaylight / yangtools / yang / data / impl / schema / InterningLeafSetNodeBuilder.java
1 /*
2  * Copyright (c) 2015 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;
9
10 import static java.util.Objects.requireNonNull;
11
12 import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode;
13 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.ListNodeBuilder;
14 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafSetNodeBuilder;
15 import org.opendaylight.yangtools.yang.data.util.LeafsetEntryInterner;
16 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
17 import org.opendaylight.yangtools.yang.model.api.LeafListSchemaNode;
18
19 final class InterningLeafSetNodeBuilder<T> extends ImmutableLeafSetNodeBuilder<T> {
20     private final LeafsetEntryInterner interner;
21
22     private InterningLeafSetNodeBuilder(final LeafsetEntryInterner interner) {
23         this.interner = requireNonNull(interner);
24     }
25
26     private InterningLeafSetNodeBuilder(final LeafsetEntryInterner interner, final int sizeHint) {
27         super(sizeHint);
28         this.interner = requireNonNull(interner);
29     }
30
31     private static LeafsetEntryInterner getInterner(final DataSchemaNode schema) {
32         return schema instanceof LeafListSchemaNode ? LeafsetEntryInterner.forSchema((LeafListSchemaNode) schema)
33                 : null;
34     }
35
36     static <T> ListNodeBuilder<T, LeafSetEntryNode<T>> create(final DataSchemaNode schema) {
37         final LeafsetEntryInterner interner = getInterner(schema);
38         if (interner != null) {
39             return new InterningLeafSetNodeBuilder<>(interner);
40         }
41
42         return ImmutableLeafSetNodeBuilder.create();
43     }
44
45     static <T> ListNodeBuilder<T, LeafSetEntryNode<T>> create(final DataSchemaNode schema, final int sizeHint) {
46         final LeafsetEntryInterner interner = getInterner(schema);
47         if (interner != null) {
48             return new InterningLeafSetNodeBuilder<>(interner, sizeHint);
49         }
50
51         return ImmutableLeafSetNodeBuilder.create(sizeHint);
52     }
53
54     @Override
55     public ListNodeBuilder<T, LeafSetEntryNode<T>> withChild(final LeafSetEntryNode<T> child) {
56         return super.withChild(interner.intern(child));
57     }
58 }