X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-distributed-datastore%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fdatabroker%2Factors%2Fdds%2FClientTransaction.java;h=efa503a90c0f3a8c909222a0c13d624a50111ba3;hb=e1c283de301355cb8fa3f7d4fa28a6dd0af501eb;hp=334ab71d585d2f4295b8d78b2e5094fff839973e;hpb=5cb0787412ab63a3aa5dcc044511e1ce569662cf;p=controller.git

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/databroker/actors/dds/ClientTransaction.java b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/databroker/actors/dds/ClientTransaction.java
index 334ab71d58..efa503a90c 100644
--- a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/databroker/actors/dds/ClientTransaction.java
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/databroker/actors/dds/ClientTransaction.java
@@ -13,8 +13,11 @@ import com.google.common.base.Preconditions;
 import com.google.common.collect.Iterables;
 import com.google.common.util.concurrent.CheckedFuture;
 import java.util.Collection;
+import javax.annotation.Nonnull;
 import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
 import org.opendaylight.mdsal.common.api.ReadFailedException;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeCursor;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteCursor;
 import org.opendaylight.mdsal.dom.spi.store.DOMStoreThreePhaseCommitCohort;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
@@ -50,13 +53,14 @@ import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  * @author Robert Varga
  */
 @Beta
-public final class ClientTransaction extends AbstractClientHandle<AbstractProxyTransaction> {
+public class ClientTransaction extends AbstractClientHandle<AbstractProxyTransaction> {
+
+    private ClientTransactionCursor cursor;
 
     ClientTransaction(final AbstractClientHistory parent, final TransactionIdentifier transactionId) {
         super(parent, transactionId);
     }
 
-
     private AbstractProxyTransaction createProxy(final Long shard) {
         return parent().createTransactionProxy(getIdentifier(), shard);
     }
@@ -65,6 +69,12 @@ public final class ClientTransaction extends AbstractClientHandle<AbstractProxyT
         return ensureProxy(path, this::createProxy);
     }
 
+    public DOMDataTreeWriteCursor openCursor() {
+        Preconditions.checkState(cursor == null, "Transaction %s has open cursor", getIdentifier());
+        cursor = new ClientTransactionCursor(this);
+        return cursor;
+    }
+
     public CheckedFuture<Boolean, ReadFailedException> exists(final YangInstanceIdentifier path) {
         return ensureTransactionProxy(path).exists(path);
     }
@@ -107,4 +117,10 @@ public final class ClientTransaction extends AbstractClientHandle<AbstractProxyT
 
         return parent().onTransactionReady(this, cohort);
     }
+
+    void closeCursor(@Nonnull final DOMDataTreeCursor cursor) {
+        if (cursor.equals(this.cursor)) {
+            this.cursor = null;
+        }
+    }
 }