Remove MapModificationStrategy hack
[yangtools.git] / yang / yang-data-impl / src / main / java / org / opendaylight / yangtools / yang / data / impl / schema / tree / AugmentationModificationStrategy.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 java.util.Optional;
11 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
12 import org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode;
13 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeConfiguration;
14 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataValidationFailedException;
15 import org.opendaylight.yangtools.yang.data.api.schema.tree.spi.TreeNode;
16 import org.opendaylight.yangtools.yang.data.api.schema.tree.spi.Version;
17 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
18 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableAugmentationNodeBuilder;
19 import org.opendaylight.yangtools.yang.data.util.DataSchemaContextNode;
20 import org.opendaylight.yangtools.yang.model.api.AugmentationSchemaNode;
21 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
22 import org.opendaylight.yangtools.yang.model.util.EffectiveAugmentationSchema;
23
24 final class AugmentationModificationStrategy
25         extends DataNodeContainerModificationStrategy<AugmentationSchemaNode> {
26     private static final NormalizedNodeContainerSupport<AugmentationIdentifier, AugmentationNode> SUPPORT =
27             new NormalizedNodeContainerSupport<>(AugmentationNode.class, ImmutableAugmentationNodeBuilder::create,
28                     ImmutableAugmentationNodeBuilder::create);
29
30     private final AugmentationNode emptyNode;
31
32     AugmentationModificationStrategy(final AugmentationSchemaNode schema, final DataNodeContainer resolved,
33             final DataTreeConfiguration treeConfig) {
34         super(SUPPORT, EffectiveAugmentationSchema.create(schema, resolved), treeConfig);
35         emptyNode = Builders.augmentationBuilder()
36                 .withNodeIdentifier(DataSchemaContextNode.augmentationIdentifierFrom(schema))
37                 .build();
38     }
39
40     @Override
41     Optional<TreeNode> apply(final ModifiedNode modification, final Optional<TreeNode> storeMeta,
42             final Version version) {
43         return AutomaticLifecycleMixin.apply(super::apply, this::applyWrite, emptyNode, modification, storeMeta,
44             version);
45     }
46
47     @Override
48     void checkApplicable(final ModificationPath path, final NodeModification modification,
49             final Optional<TreeNode> current, final Version version) throws DataValidationFailedException {
50         AutomaticLifecycleMixin.checkApplicable(super::checkApplicable, emptyNode, path, modification, current,
51             version);
52     }
53 }