Bug 5699 - Migrate existing code to use new sharding apis
[mdsal.git] / dom / mdsal-dom-broker / src / main / java / org / opendaylight / mdsal / dom / broker / TransactionChainReadTransaction.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.broker;
10
11 import com.google.common.base.Function;
12 import com.google.common.base.Optional;
13 import com.google.common.util.concurrent.CheckedFuture;
14 import com.google.common.util.concurrent.FutureCallback;
15 import com.google.common.util.concurrent.Futures;
16 import com.google.common.util.concurrent.ListenableFuture;
17 import com.google.common.util.concurrent.SettableFuture;
18 import javax.annotation.Nullable;
19 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
20 import org.opendaylight.mdsal.common.api.ReadFailedException;
21 import org.opendaylight.mdsal.dom.api.DOMDataTreeReadTransaction;
22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
23 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
24
25 public class TransactionChainReadTransaction implements DOMDataTreeReadTransaction {
26
27     private final DOMDataTreeReadTransaction delegateReadTx;
28     private final ListenableFuture<Void> previousWriteTxFuture;
29     private final Object identifier;
30     private final ShardedDOMTransactionChainAdapter txChain;
31
32     TransactionChainReadTransaction(final Object txIdentifier, final DOMDataTreeReadTransaction delegateReadTx,
33                                     final ListenableFuture<Void> previousWriteTxFuture,
34                                     final ShardedDOMTransactionChainAdapter txChain) {
35         this.delegateReadTx = delegateReadTx;
36         this.previousWriteTxFuture = previousWriteTxFuture;
37         this.identifier = txIdentifier;
38         this.txChain = txChain;
39     }
40
41     @Override
42     public CheckedFuture<Optional<NormalizedNode<?, ?>>, ReadFailedException> read(final LogicalDatastoreType store,
43                                                                                    final YangInstanceIdentifier path) {
44         final SettableFuture<Optional<NormalizedNode<?, ?>>> readResult = SettableFuture.create();
45
46         Futures.addCallback(previousWriteTxFuture, new FutureCallback<Void>() {
47             @Override
48             public void onSuccess(@Nullable final Void result) {
49                 Futures.addCallback(delegateReadTx.read(store, path),
50                         new FutureCallback<Optional<NormalizedNode<?, ?>>>() {
51
52                             @Override
53                             public void onSuccess(@Nullable final Optional<NormalizedNode<?, ?>> result) {
54                                 readResult.set(result);
55                             }
56
57                             @Override
58                             public void onFailure(final Throwable t) {
59                                 txChain.transactionFailed(TransactionChainReadTransaction.this, t);
60                                 readResult.setException(t);
61                             }
62                         });
63             }
64
65             @Override
66             public void onFailure(final Throwable t) {
67                 // we don't have to notify txchain about this failure
68                 // failed write transaction should do this
69                 readResult.setException(t);
70             }
71         });
72
73         return Futures.makeChecked(readResult, ReadFailedException.MAPPER);
74     }
75
76     @Override
77     public CheckedFuture<Boolean, ReadFailedException> exists(final LogicalDatastoreType store,
78                                                               final YangInstanceIdentifier path) {
79         final Function<Optional<NormalizedNode<?, ?>>, Boolean> transform =
80                 optionalNode -> optionalNode.isPresent() ? Boolean.TRUE : Boolean.FALSE;
81         final ListenableFuture<Boolean> existsResult = Futures.transform(read(store, path), transform);
82         return Futures.makeChecked(existsResult, ReadFailedException.MAPPER);
83     }
84
85     @Override
86     public void close() {
87         delegateReadTx.close();
88         txChain.closeReadTransaction();
89     }
90
91     @Override
92     public Object getIdentifier() {
93         return identifier;
94     }
95 }