dom/mdsal-dom-inmemory-datastore/src/main/java/org/opendaylight/mdsal/dom/store/inmemory/ShardDataModificationFactoryBuilder.java

   1 /*
   2  * Copyright (c) 2016 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
   9 package org.opendaylight.mdsal.dom.store.inmemory;
  10
  11 import static java.util.Objects.requireNonNull;
  12
  13 import java.util.HashMap;
  14 import java.util.Iterator;
  15 import java.util.Map;
  16 import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
  17 import org.opendaylight.mdsal.dom.spi.shard.ForeignShardModificationContext;
  18 import org.opendaylight.mdsal.dom.spi.shard.ModificationContextNodeBuilder;
  19 import org.opendaylight.mdsal.dom.spi.shard.WriteableSubshardBoundaryNode;
  20 import org.opendaylight.yangtools.concepts.Builder;
  21 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
  23
  24 public abstract class ShardDataModificationFactoryBuilder<T> extends ModificationContextNodeBuilder
  25         implements Builder<T> {
  26     protected final DOMDataTreeIdentifier root;
  27     protected final Map<DOMDataTreeIdentifier, ForeignShardModificationContext> childShards = new HashMap<>();
  28
  29     public ShardDataModificationFactoryBuilder(final DOMDataTreeIdentifier root) {
  30         this.root = requireNonNull(root);
  31     }
  32
  33     public void addSubshard(final ForeignShardModificationContext value) {
  34         WriteableSubshardBoundaryNode leafNode = WriteableSubshardBoundaryNode.from(value);
  35         putNode(value.getIdentifier().getRootIdentifier(), leafNode);
  36     }
  37
  38     public void addSubshard(final DOMDataTreeIdentifier prefix, final ForeignShardModificationContext value) {
  39         childShards.put(prefix, value);
  40     }
  41
  42     private void putNode(final YangInstanceIdentifier key, final WriteableSubshardBoundaryNode subshardNode) {
  43         final Iterator<PathArgument> toBoundary = toRelative(key).getPathArguments().iterator();
  44         if (toBoundary.hasNext()) {
  45             ModificationContextNodeBuilder current = this;
  46             while (true) {
  47                 final PathArgument nextArg = toBoundary.next();
  48                 if (!toBoundary.hasNext()) {
  49                     current.addBoundary(nextArg, subshardNode);
  50                     break;
  51                 }
  52
  53                 current = current.getInterior(nextArg);
  54             }
  55         }
  56     }
  57
  58     private YangInstanceIdentifier toRelative(final YangInstanceIdentifier key) {
  59         return key.relativeTo(root.getRootIdentifier()).get();
  60     }
  61
  62     public abstract T build();
  63 }