Add utility wrappers for instantiating builders/nodes
[yangtools.git] / yang / yang-data-impl / src / main / java / org / opendaylight / yangtools / yang / data / impl / schema / tree / ListEntryModificationStrategy.java
1 /*
2  * Copyright (c) 2014 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.tree;
9
10 import static java.util.Objects.requireNonNull;
11
12 import java.util.Optional;
13 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
14 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
15 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
16 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeConfiguration;
17 import org.opendaylight.yangtools.yang.data.api.schema.tree.spi.TreeNode;
18 import org.opendaylight.yangtools.yang.data.api.schema.tree.spi.Version;
19 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableMapEntryNodeBuilder;
20 import org.opendaylight.yangtools.yang.data.impl.schema.tree.NormalizedNodeContainerSupport.Single;
21 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
22
23 class ListEntryModificationStrategy extends DataNodeContainerModificationStrategy<ListSchemaNode> {
24     private static final class EnforcingMandatory extends ListEntryModificationStrategy {
25         private final MandatoryLeafEnforcer enforcer;
26
27         EnforcingMandatory(final ListSchemaNode schemaNode, final DataTreeConfiguration treeConfig,
28                 final MandatoryLeafEnforcer enforcer) {
29             super(schemaNode, treeConfig);
30             this.enforcer = requireNonNull(enforcer);
31         }
32
33         @Override
34         void verifyStructure(final NormalizedNode<?, ?> writtenValue, final boolean verifyChildren) {
35             super.verifyStructure(writtenValue, verifyChildren);
36             if (verifyChildren) {
37                 enforcer.enforceOnData(writtenValue);
38             }
39         }
40
41         @Override
42         protected TreeNode applyMerge(final ModifiedNode modification, final TreeNode currentMeta,
43                 final Version version) {
44             final TreeNode ret = super.applyMerge(modification, currentMeta, version);
45             enforcer.enforceOnTreeNode(ret);
46             return ret;
47         }
48
49         @Override
50         protected TreeNode applyWrite(final ModifiedNode modification, final NormalizedNode<?, ?> newValue,
51                 final Optional<TreeNode> currentMeta, final Version version) {
52             final TreeNode ret = super.applyWrite(modification, newValue, currentMeta, version);
53             enforcer.enforceOnTreeNode(ret);
54             return ret;
55         }
56
57         @Override
58         protected TreeNode applyTouch(final ModifiedNode modification, final TreeNode currentMeta,
59                 final Version version) {
60             final TreeNode ret = super.applyTouch(modification, currentMeta, version);
61             enforcer.enforceOnTreeNode(ret);
62             return ret;
63         }
64     }
65
66     private static final Single<NodeIdentifierWithPredicates, MapEntryNode> SUPPORT = new Single<>(MapEntryNode.class,
67             ImmutableMapEntryNodeBuilder::create, ImmutableMapEntryNodeBuilder::create);
68
69     ListEntryModificationStrategy(final ListSchemaNode schema, final DataTreeConfiguration treeConfig) {
70         super(SUPPORT, schema, treeConfig);
71     }
72
73     static ListEntryModificationStrategy of(final ListSchemaNode schema, final DataTreeConfiguration treeConfig) {
74         final Optional<MandatoryLeafEnforcer> enforcer = MandatoryLeafEnforcer.forContainer(schema, treeConfig);
75         return enforcer.isPresent() ? new EnforcingMandatory(schema, treeConfig, enforcer.get())
76                 : new ListEntryModificationStrategy(schema, treeConfig);
77     }
78 }