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 com.google.common.base.Preconditions.checkArgument;
  11 import com.google.common.base.Optional;
  12 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  13 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  14 import org.opendaylight.yangtools.yang.data.api.schema.tree.TreeType;
  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.builder.api.DataContainerNodeBuilder;
  18 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableMapEntryNodeBuilder;
  19 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
  20
  21 final class ListEntryModificationStrategy extends AbstractDataNodeContainerModificationStrategy<ListSchemaNode> {
  22     private final MandatoryLeafEnforcer enforcer;
  23
  24     ListEntryModificationStrategy(final ListSchemaNode schema, final TreeType treeType) {
  25         super(schema, MapEntryNode.class, treeType);
  26         enforcer = MandatoryLeafEnforcer.forContainer(schema, treeType);
  27     }
  28
  29     @Override
  30     protected TreeNode applyMerge(final ModifiedNode modification, final TreeNode currentMeta, final Version version) {
  31         final TreeNode ret = super.applyMerge(modification, currentMeta, version);
  32         enforcer.enforceOnTreeNode(ret);
  33         return ret;
  34     }
  35
  36     @Override
  37     protected TreeNode applyWrite(final ModifiedNode modification, final Optional<TreeNode> currentMeta,
  38             final Version version) {
  39         final TreeNode ret = super.applyWrite(modification, currentMeta, version);
  40         enforcer.enforceOnTreeNode(ret);
  41         return ret;
  42     }
  43
  44     @Override
  45     protected TreeNode applyTouch(final ModifiedNode modification, final TreeNode currentMeta, final Version version) {
  46         final TreeNode ret = super.applyTouch(modification, currentMeta, version);
  47         enforcer.enforceOnTreeNode(ret);
  48         return ret;
  49     }
  50
  51     @Override
  52     @SuppressWarnings("rawtypes")
  53     protected DataContainerNodeBuilder createBuilder(final NormalizedNode<?, ?> original) {
  54         checkArgument(original instanceof MapEntryNode);
  55         return ImmutableMapEntryNodeBuilder.create((MapEntryNode) original);
  56     }
  57 }