c4be2a414ddabb35c935b3628a7d591c12a5cc90
[mdsal.git] / 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 @Deprecated(forRemoval = true)
23 public abstract class AbstractShardModificationFactoryBuilder<T> extends ModificationContextNodeBuilder
24         implements Builder<T> {
25     protected final Map<DOMDataTreeIdentifier, ForeignShardModificationContext> childShards = new HashMap<>();
26     protected final DOMDataTreeIdentifier root;
27
28     public AbstractShardModificationFactoryBuilder(final DOMDataTreeIdentifier root) {
29         this.root = requireNonNull(root);
30     }
31
32     public void addSubshard(final ForeignShardModificationContext value) {
33         WriteableSubshardBoundaryNode leafNode = WriteableSubshardBoundaryNode.from(value);
34         putNode(value.getIdentifier().getRootIdentifier(), leafNode);
35     }
36
37     public void addSubshard(final DOMDataTreeIdentifier prefix, final ForeignShardModificationContext value) {
38         childShards.put(prefix, value);
39     }
40
41     @Override
42     public abstract T build();
43
44     private void putNode(final YangInstanceIdentifier key, final WriteableSubshardBoundaryNode subshardNode) {
45         final Iterator<PathArgument> toBoundary = toRelative(key).getPathArguments().iterator();
46         if (toBoundary.hasNext()) {
47             ModificationContextNodeBuilder current = this;
48             while (true) {
49                 final PathArgument nextArg = toBoundary.next();
50                 if (!toBoundary.hasNext()) {
51                     current.addBoundary(nextArg, subshardNode);
52                     break;
53                 }
54
55                 current = current.getInterior(nextArg);
56             }
57         }
58     }
59
60     private YangInstanceIdentifier toRelative(final YangInstanceIdentifier key) {
61         return key.relativeTo(root.getRootIdentifier()).get();
62     }
63 }