Bug 5699 - Migrate existing code to use new sharding apis

[mdsal.git] / dom / mdsal-dom-broker / src / main / java / org / opendaylight / mdsal / dom / broker / TransactionChainWriteTransaction.java

diff --git a/dom/mdsal-dom-broker/src/main/java/org/opendaylight/mdsal/dom/broker/TransactionChainWriteTransaction.java b/dom/mdsal-dom-broker/src/main/java/org/opendaylight/mdsal/dom/broker/TransactionChainWriteTransaction.java
new file mode 100644 (file)
index 0000000..187bb42
--- /dev/null
+++ b/dom/mdsal-dom-broker/src/main/java/org/opendaylight/mdsal/dom/broker/TransactionChainWriteTransaction.java
@@ -0,0 +1,79 @@
+/*
+ * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.mdsal.dom.broker;
+
+import com.google.common.base.Preconditions;
+import com.google.common.util.concurrent.CheckedFuture;
+import com.google.common.util.concurrent.FutureCallback;
+import com.google.common.util.concurrent.Futures;
+import javax.annotation.Nullable;
+import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
+import org.opendaylight.mdsal.common.api.TransactionCommitFailedException;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
+
+public class TransactionChainWriteTransaction implements DOMDataTreeWriteTransaction {
+    private final DOMDataTreeWriteTransaction delegateTx;
+    private final Object identifier;
+    private final ShardedDOMTransactionChainAdapter txChain;
+
+    public TransactionChainWriteTransaction(final Object identifier, final DOMDataTreeWriteTransaction delegateTx,
+                                            final ShardedDOMTransactionChainAdapter txChain) {
+        this.delegateTx = Preconditions.checkNotNull(delegateTx);
+        this.identifier = Preconditions.checkNotNull(identifier);
+        this.txChain = Preconditions.checkNotNull(txChain);
+    }
+
+
+    @Override
+    public void put(final LogicalDatastoreType store, final YangInstanceIdentifier path, final NormalizedNode<?, ?> data) {
+        delegateTx.put(store, path, data);
+    }
+
+    @Override
+    public void merge(final LogicalDatastoreType store, final YangInstanceIdentifier path, final NormalizedNode<?, ?> data) {
+        delegateTx.merge(store, path, data);
+    }
+
+    @Override
+    public boolean cancel() {
+        txChain.closeWriteTransaction(Futures.immediateFuture(null));
+        return delegateTx.cancel();
+    }
+
+    @Override
+    public void delete(final LogicalDatastoreType store, final YangInstanceIdentifier path) {
+        delegateTx.delete(store, path);
+    }
+
+    @Override
+    public CheckedFuture<Void, TransactionCommitFailedException> submit() {
+        final CheckedFuture<Void, TransactionCommitFailedException> writeResultFuture = delegateTx.submit();
+        Futures.addCallback(writeResultFuture, new FutureCallback<Void>() {
+            @Override
+            public void onSuccess(@Nullable final Void result) {
+                // NOOP
+            }
+
+            @Override
+            public void onFailure(final Throwable t) {
+                txChain.transactionFailed(TransactionChainWriteTransaction.this, t);
+            }
+        });
+
+        txChain.closeWriteTransaction(writeResultFuture);
+        return writeResultFuture;
+    }
+
+    @Override
+    public Object getIdentifier() {
+        return identifier;
+    }
+}