BUG 6057: Rewrite ShardedDOMProducer to use new cursor api
[mdsal.git] / binding / mdsal-binding-dom-adapter / src / main / java / org / opendaylight / mdsal / binding / dom / adapter / BindingDOMDataTreeWriteCursorAdapter.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.binding.dom.adapter;
10
11 import java.util.AbstractMap.SimpleEntry;
12 import java.util.ArrayDeque;
13 import java.util.Deque;
14 import java.util.Map.Entry;
15 import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
16 import org.opendaylight.mdsal.binding.api.DataTreeWriteCursor;
17 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteCursor;
18 import org.opendaylight.yangtools.yang.binding.DataObject;
19 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
20 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.PathArgument;
21 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
22 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
23
24 public class BindingDOMDataTreeWriteCursorAdapter<T extends DOMDataTreeWriteCursor> implements DataTreeWriteCursor {
25
26     private T delegate;
27     private BindingToNormalizedNodeCodec codec;
28     private final Deque<PathArgument> stack = new ArrayDeque<>();
29
30     public BindingDOMDataTreeWriteCursorAdapter(final DataTreeIdentifier<?> path, final T delegate, final BindingToNormalizedNodeCodec codec) {
31
32         this.delegate = delegate;
33         this.codec = codec;
34         path.getRootIdentifier().getPathArguments().forEach(stack::push);
35     }
36
37     private YangInstanceIdentifier.PathArgument convertToNormalized(final PathArgument child) {
38         stack.push(child);
39         final InstanceIdentifier<?> iid = InstanceIdentifier.create(stack);
40         final YangInstanceIdentifier ret = codec.toNormalized(iid);
41         stack.pop();
42         return ret.getLastPathArgument();
43     }
44
45     private <T extends DataObject> Entry<YangInstanceIdentifier, NormalizedNode<?, ?>> convertToNormalized(final PathArgument child, final T data) {
46         stack.push(child);
47         final InstanceIdentifier<?> iid = InstanceIdentifier.create(stack);
48         final Entry<YangInstanceIdentifier, NormalizedNode<?, ?>> entry = codec.toNormalizedNode(new SimpleEntry<>(iid, data));
49         stack.pop();
50         return entry;
51     }
52
53     @Override
54     public void delete(final PathArgument child) {
55         delegate.delete(convertToNormalized(child));
56     }
57
58     @Override
59     public <T extends DataObject> void merge(final PathArgument child, final T data) {
60         final Entry<YangInstanceIdentifier, NormalizedNode<?, ?>> entry = convertToNormalized(child, data);
61         delegate.merge(entry.getKey().getLastPathArgument(), entry.getValue());
62     }
63
64     @Override
65     public <T extends DataObject> void write(PathArgument child, T data) {
66         final Entry<YangInstanceIdentifier, NormalizedNode<?, ?>> entry = convertToNormalized(child, data);
67         delegate.write(entry.getKey().getLastPathArgument(), entry.getValue());
68     }
69
70     @Override
71     public void enter(final PathArgument child) {
72         stack.push(child);
73     }
74
75     @Override
76     public void enter(final PathArgument... path) {
77         for (final PathArgument pathArgument : path) {
78             enter(pathArgument);
79         }
80     }
81
82     @Override
83     public void enter(final Iterable<PathArgument> path) {
84         path.forEach(this::enter);
85     }
86
87     @Override
88     public void exit() {
89         stack.pop();
90     }
91
92     @Override
93     public void exit(int depth) {
94         for (int i = 0; i < depth; i++) {
95             exit();
96         }
97     }
98
99     @Override
100     public void close() {
101         delegate.close();
102     }
103 }