yang/yang-data-impl/src/main/java/org/opendaylight/yangtools/yang/data/impl/schema/tree/MapEntryModificationStrategy.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.eclipse.jdt.annotation.NonNull;
  14 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
  15 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  16 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  17 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeConfiguration;
  18 import org.opendaylight.yangtools.yang.data.api.schema.tree.spi.TreeNode;
  19 import org.opendaylight.yangtools.yang.data.api.schema.tree.spi.Version;
  20 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableMapEntryNodeBuilder;
  21 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
  22
  23 class MapEntryModificationStrategy extends DataNodeContainerModificationStrategy<ListSchemaNode> {
  24     private static final class EnforcingMandatory extends MapEntryModificationStrategy {
  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 mandatoryVerifyValueChildren(final NormalizedNode<?, ?> writtenValue) {
  35             enforcer.enforceOnData(writtenValue);
  36         }
  37
  38         @Override
  39         protected TreeNode applyMerge(final ModifiedNode modification, final TreeNode currentMeta,
  40                 final Version version) {
  41             final TreeNode ret = super.applyMerge(modification, currentMeta, version);
  42             enforcer.enforceOnTreeNode(ret);
  43             return ret;
  44         }
  45
  46         @Override
  47         protected TreeNode applyWrite(final ModifiedNode modification, final NormalizedNode<?, ?> newValue,
  48                 final Optional<? extends TreeNode> currentMeta, final Version version) {
  49             final TreeNode ret = super.applyWrite(modification, newValue, currentMeta, version);
  50             enforcer.enforceOnTreeNode(ret);
  51             return ret;
  52         }
  53
  54         @Override
  55         protected TreeNode applyTouch(final ModifiedNode modification, final TreeNode currentMeta,
  56                 final Version version) {
  57             final TreeNode ret = super.applyTouch(modification, currentMeta, version);
  58             enforcer.enforceOnTreeNode(ret);
  59             return ret;
  60         }
  61     }
  62
  63     private static final NormalizedNodeContainerSupport<NodeIdentifierWithPredicates, MapEntryNode> SUPPORT =
  64             new NormalizedNodeContainerSupport<>(MapEntryNode.class, ImmutableMapEntryNodeBuilder::create,
  65                     ImmutableMapEntryNodeBuilder::create);
  66
  67     MapEntryModificationStrategy(final ListSchemaNode schema, final DataTreeConfiguration treeConfig) {
  68         super(SUPPORT, schema, treeConfig);
  69     }
  70
  71     static @NonNull MapEntryModificationStrategy of(final ListSchemaNode schema,
  72             final DataTreeConfiguration treeConfig) {
  73         final Optional<MandatoryLeafEnforcer> enforcer = MandatoryLeafEnforcer.forContainer(schema, treeConfig);
  74         return enforcer.isPresent() ? new EnforcingMandatory(schema, treeConfig, enforcer.get())
  75                 : new MapEntryModificationStrategy(schema, treeConfig);
  76     }
  77 }