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