BUG-5280: split DistributedDataStore

[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / main / java / org / opendaylight / controller / cluster / databroker / ClientBackedTransactionChain.java

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/databroker/ClientBackedTransactionChain.java b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/databroker/ClientBackedTransactionChain.java
new file mode 100644 (file)
index 0000000..db30372
--- /dev/null
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/databroker/ClientBackedTransactionChain.java
@@ -0,0 +1,93 @@
+/*
+ * 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.controller.cluster.databroker;
+
+import com.google.common.base.Preconditions;
+import java.util.Map;
+import java.util.WeakHashMap;
+import javax.annotation.concurrent.GuardedBy;
+import org.opendaylight.controller.cluster.databroker.actors.dds.AbstractClientHandle;
+import org.opendaylight.controller.cluster.databroker.actors.dds.ClientLocalHistory;
+import org.opendaylight.controller.cluster.databroker.actors.dds.ClientSnapshot;
+import org.opendaylight.controller.cluster.databroker.actors.dds.ClientTransaction;
+import org.opendaylight.controller.md.sal.common.api.data.TransactionChainClosedException;
+import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadTransaction;
+import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadWriteTransaction;
+import org.opendaylight.controller.sal.core.spi.data.DOMStoreTransactionChain;
+import org.opendaylight.controller.sal.core.spi.data.DOMStoreWriteTransaction;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * An implementation of {@link DOMStoreTransactionChain} backed by a {@link ClientLocalHistory}.
+ *
+ * @author Robert Varga
+ */
+final class ClientBackedTransactionChain implements DOMStoreTransactionChain {
+    private static final Logger LOG = LoggerFactory.getLogger(ClientBackedTransactionChain.class);
+
+    @GuardedBy("this")
+    private final Map<AbstractClientHandle<?>, Boolean> openSnapshots = new WeakHashMap<>();
+
+    private final ClientLocalHistory history;
+
+    ClientBackedTransactionChain(final ClientLocalHistory history) {
+        this.history = Preconditions.checkNotNull(history);
+    }
+
+    @Override
+    public DOMStoreReadTransaction newReadOnlyTransaction() {
+        return new ClientBackedReadTransaction(createSnapshot(), this);
+    }
+
+    @Override
+    public DOMStoreReadWriteTransaction newReadWriteTransaction() {
+        return new ClientBackedReadWriteTransaction(createTransaction());
+    }
+
+    @Override
+    public DOMStoreWriteTransaction newWriteOnlyTransaction() {
+        return new ClientBackedWriteTransaction(createTransaction());
+    }
+
+    @Override
+    public synchronized void close() {
+        for (AbstractClientHandle<?> snap : openSnapshots.keySet()) {
+            LOG.warn("Aborting unclosed transaction {}", snap.getIdentifier());
+            snap.abort();
+        }
+        openSnapshots.clear();
+
+        history.close();
+    }
+
+    synchronized void snapshotClosed(final ClientSnapshot clientTransaction) {
+        openSnapshots.remove(clientTransaction);
+    }
+
+    private ClientSnapshot createSnapshot() {
+        try {
+            return recordSnapshot(history.takeSnapshot());
+        } catch (org.opendaylight.mdsal.common.api.TransactionChainClosedException e) {
+            throw new TransactionChainClosedException("Transaction chain has been closed", e);
+        }
+    }
+
+    private ClientTransaction createTransaction() {
+        try {
+            return recordSnapshot(history.createTransaction());
+        } catch (org.opendaylight.mdsal.common.api.TransactionChainClosedException e) {
+            throw new TransactionChainClosedException("Transaction chain has been closed", e);
+        }
+    }
+
+    private synchronized <T extends AbstractClientHandle<?>> T recordSnapshot(final T snapshot) {
+        openSnapshots.put(snapshot, Boolean.TRUE);
+        return snapshot;
+    }
+}