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 / BindingDOMDataTreeProducerAdapter.java
1 /*
2  * Copyright (c) 2015 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.binding.dom.adapter;
9
10 import com.google.common.base.Preconditions;
11 import java.util.Collection;
12 import org.opendaylight.mdsal.binding.api.CursorAwareWriteTransaction;
13 import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
14 import org.opendaylight.mdsal.binding.api.DataTreeProducer;
15 import org.opendaylight.mdsal.binding.api.DataTreeProducerException;
16 import org.opendaylight.mdsal.binding.api.WriteTransaction;
17 import org.opendaylight.mdsal.dom.api.DOMDataTreeCursorAwareTransaction;
18 import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
19 import org.opendaylight.mdsal.dom.api.DOMDataTreeProducer;
20 import org.opendaylight.mdsal.dom.api.DOMDataTreeProducerBusyException;
21 import org.opendaylight.mdsal.dom.api.DOMDataTreeProducerException;
22 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
23
24 class BindingDOMDataTreeProducerAdapter implements DataTreeProducer {
25
26     private final DOMDataTreeProducer delegate;
27     private final BindingToNormalizedNodeCodec codec;
28
29     protected BindingDOMDataTreeProducerAdapter(final DOMDataTreeProducer delegate,
30             final BindingToNormalizedNodeCodec codec) {
31         super();
32         this.delegate = Preconditions.checkNotNull(delegate);
33         this.codec = codec;
34     }
35
36     @Override
37     public CursorAwareWriteTransaction createTransaction(final boolean isolated) {
38         final DOMDataTreeCursorAwareTransaction domTx = delegate.createTransaction(isolated);
39         return new BindingDOMCursorAwareWriteTransactionAdapter<>(domTx, codec);
40     }
41
42     static DataTreeProducer create(final DOMDataTreeProducer domProducer,
43             final BindingToNormalizedNodeCodec codec) {
44         return new BindingDOMDataTreeProducerAdapter(domProducer, codec);
45     }
46
47     @Override
48     public DataTreeProducer createProducer(final Collection<DataTreeIdentifier<?>> subtrees) {
49         final Collection<DOMDataTreeIdentifier> domSubtrees = codec.toDOMDataTreeIdentifiers(subtrees);
50         final DOMDataTreeProducer domChildProducer = delegate.createProducer(domSubtrees);
51         return BindingDOMDataTreeProducerAdapter.create(domChildProducer, codec);
52     }
53
54     @Override
55     public void close() throws DataTreeProducerException {
56         try {
57             delegate.close();
58         } catch (final DOMDataTreeProducerBusyException e) {
59             throw new DataTreeProducerException(e.getMessage(), e);
60         } catch (final DOMDataTreeProducerException e) {
61             throw new DataTreeProducerException(e.getMessage(), e);
62         }
63     }
64
65 }