yang/yang-data-impl/src/main/java/org/opendaylight/yangtools/yang/data/impl/schema/builder/impl/ImmutableMapEntryNodeBuilder.java

   1 /*
   2  * Copyright (c) 2013 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.builder.impl;
   9
  10 import java.util.LinkedHashMap;
  11 import java.util.List;
  12 import java.util.Map;
  13 import org.opendaylight.yangtools.yang.common.QName;
  14 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  15 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
  16 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
  17 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  18 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeAttrBuilder;
  19 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.valid.DataValidationException;
  20 import org.opendaylight.yangtools.yang.data.impl.schema.nodes.AbstractImmutableDataContainerAttrNode;
  21
  22 public class ImmutableMapEntryNodeBuilder extends AbstractImmutableDataContainerNodeAttrBuilder<YangInstanceIdentifier.NodeIdentifierWithPredicates, MapEntryNode> {
  23
  24     protected final Map<QName, YangInstanceIdentifier.PathArgument> childrenQNamesToPaths;
  25
  26     protected ImmutableMapEntryNodeBuilder() {
  27         super();
  28         this.childrenQNamesToPaths = new LinkedHashMap<>();
  29     }
  30
  31     protected ImmutableMapEntryNodeBuilder(final int sizeHint) {
  32         super(sizeHint);
  33         this.childrenQNamesToPaths = new LinkedHashMap<>(sizeHint);
  34     }
  35
  36     protected ImmutableMapEntryNodeBuilder(final ImmutableMapEntryNode node) {
  37         super(node);
  38         this.childrenQNamesToPaths = new LinkedHashMap<>();
  39         fillQnames(node.getValue(), childrenQNamesToPaths);
  40     }
  41
  42     public static DataContainerNodeAttrBuilder<YangInstanceIdentifier.NodeIdentifierWithPredicates, MapEntryNode> create() {
  43         return new ImmutableMapEntryNodeBuilder();
  44     }
  45
  46     public static DataContainerNodeAttrBuilder<YangInstanceIdentifier.NodeIdentifierWithPredicates, MapEntryNode> create(final int sizeHint) {
  47         return new ImmutableMapEntryNodeBuilder(sizeHint);
  48     }
  49
  50     public static DataContainerNodeAttrBuilder<YangInstanceIdentifier.NodeIdentifierWithPredicates, MapEntryNode> create(final MapEntryNode node) {
  51         if (!(node instanceof ImmutableMapEntryNode)) {
  52             throw new UnsupportedOperationException(String.format("Cannot initialize from class %s", node.getClass()));
  53         }
  54
  55         return new ImmutableMapEntryNodeBuilder((ImmutableMapEntryNode)node);
  56     }
  57
  58     private static void fillQnames(final Iterable<DataContainerChild<? extends PathArgument, ?>> iterable, final Map<QName, PathArgument> out) {
  59         for (final DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?> childId : iterable) {
  60             final YangInstanceIdentifier.PathArgument identifier = childId.getIdentifier();
  61
  62             // Augmentation nodes cannot be keys, and do not have to be present in childrenQNamesToPaths map
  63             if(isAugment(identifier)) {
  64                 continue;
  65             }
  66
  67             out.put(childId.getNodeType(), identifier);
  68         }
  69     }
  70
  71
  72     @Override
  73     public DataContainerNodeAttrBuilder<YangInstanceIdentifier.NodeIdentifierWithPredicates, MapEntryNode> withValue(final List<DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?>> value) {
  74         fillQnames(value, childrenQNamesToPaths);
  75         return super.withValue(value);
  76     }
  77
  78     private static boolean isAugment(final YangInstanceIdentifier.PathArgument identifier) {
  79         return identifier instanceof YangInstanceIdentifier.AugmentationIdentifier;
  80     }
  81
  82     @Override
  83     public DataContainerNodeAttrBuilder<YangInstanceIdentifier.NodeIdentifierWithPredicates, MapEntryNode> withChild(final DataContainerChild<?, ?> child) {
  84         // Augmentation nodes cannot be keys, and do not have to be present in childrenQNamesToPaths map
  85         if(!isAugment(child.getIdentifier())) {
  86             childrenQNamesToPaths.put(child.getNodeType(), child.getIdentifier());
  87         }
  88
  89         return super.withChild(child);
  90     }
  91
  92     @Override
  93     public MapEntryNode build() {
  94         checkKeys();
  95         return new ImmutableMapEntryNode(getNodeIdentifier(), buildValue(), getAttributes());
  96     }
  97
  98     private void checkKeys() {
  99         for (final QName keyQName : getNodeIdentifier().getKeyValues().keySet()) {
 100             DataContainerChild<?, ?> childNode = getChild(childrenQNamesToPaths.get(keyQName));
 101             DataValidationException.checkListKey(childNode, getNodeIdentifier().getKeyValues(), keyQName, getNodeIdentifier());
 102         }
 103     }
 104
 105     private static final class ImmutableMapEntryNode extends AbstractImmutableDataContainerAttrNode<YangInstanceIdentifier.NodeIdentifierWithPredicates> implements MapEntryNode {
 106
 107         ImmutableMapEntryNode(final YangInstanceIdentifier.NodeIdentifierWithPredicates nodeIdentifier,
 108                 final Map<YangInstanceIdentifier.PathArgument, DataContainerChild<? extends YangInstanceIdentifier.PathArgument, ?>> children, final Map<QName, String> attributes) {
 109             super(children, nodeIdentifier, attributes);
 110         }
 111     }
 112 }