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