Create AbstractPingPongTransactionChain
[mdsal.git] / dom / mdsal-dom-spi / src / main / java / org / opendaylight / mdsal / dom / spi / PingPongTransactionChain.java
index 2f24b0b947c32fd1641fba60429ad58f9b5ddf81..322616d3ce8c140ee85b4343caf875d7590f7fce 100644 (file)
@@ -7,34 +7,10 @@
  */
 package org.opendaylight.mdsal.dom.spi;
 
-import static com.google.common.base.Preconditions.checkState;
-import static com.google.common.base.Verify.verify;
-import static java.util.Objects.requireNonNull;
-
-import com.google.common.util.concurrent.FluentFuture;
-import com.google.common.util.concurrent.FutureCallback;
-import com.google.common.util.concurrent.MoreExecutors;
-import java.lang.invoke.MethodHandles;
-import java.lang.invoke.VarHandle;
-import java.util.Map;
-import java.util.Map.Entry;
-import java.util.Optional;
-import java.util.concurrent.CancellationException;
+import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
 import java.util.function.Function;
-import org.checkerframework.checker.lock.qual.GuardedBy;
-import org.checkerframework.checker.lock.qual.Holding;
-import org.eclipse.jdt.annotation.NonNull;
-import org.eclipse.jdt.annotation.Nullable;
-import org.opendaylight.mdsal.common.api.CommitInfo;
-import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeReadTransaction;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeReadWriteTransaction;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeTransaction;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
 import org.opendaylight.mdsal.dom.api.DOMTransactionChain;
 import org.opendaylight.mdsal.dom.api.DOMTransactionChainListener;
-import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
-import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -53,455 +29,12 @@ import org.slf4j.LoggerFactory;
  * It furthermore means that the transactions returned by {@link #newReadOnlyTransaction()} counts as an outstanding
  * transaction and the user may not allocate multiple read-only transactions at the same time.
  */
-public final class PingPongTransactionChain implements DOMTransactionChain {
-    private static final Logger LOG = LoggerFactory.getLogger(PingPongTransactionChain.class);
-
-    private final DOMTransactionChainListener listener;
-    private final DOMTransactionChain delegate;
-
-    @GuardedBy("this")
-    private boolean failed;
-    @GuardedBy("this")
-    private PingPongTransaction shutdownTx;
-    @GuardedBy("this")
-    private Entry<PingPongTransaction, Throwable> deadTx;
-
-    //  This VarHandle is used to manipulate the "ready" transaction. We perform only atomic get-and-set on it.
-    private static final VarHandle READY_TX;
-    @SuppressWarnings("unused")
-    private volatile PingPongTransaction readyTx;
-
-    /*
-     * This VarHandle is used to manipulate the "locked" transaction. A locked transaction means we know that the user
-     * still holds a transaction and should at some point call us. We perform on compare-and-swap to ensure we properly
-     * detect when a user is attempting to allocated multiple transactions concurrently.
-     */
-    private static final VarHandle LOCKED_TX;
-    private volatile PingPongTransaction lockedTx;
-
-    /*
-     * This updater is used to manipulate the "inflight" transaction. There can be at most one of these at any given
-     * time. We perform only compare-and-swap on these.
-     */
-    private static final VarHandle INFLIGHT_TX;
-    private volatile PingPongTransaction inflightTx;
-
-    static {
-        final var lookup = MethodHandles.lookup();
-        try {
-            INFLIGHT_TX = lookup.findVarHandle(PingPongTransactionChain.class, "inflightTx", PingPongTransaction.class);
-            LOCKED_TX = lookup.findVarHandle(PingPongTransactionChain.class, "lockedTx", PingPongTransaction.class);
-            READY_TX = lookup.findVarHandle(PingPongTransactionChain.class, "readyTx", PingPongTransaction.class);
-        } catch (NoSuchFieldException | IllegalAccessException e) {
-            throw new ExceptionInInitializerError(e);
-        }
-    }
+public final class PingPongTransactionChain extends AbstractPingPongTransactionChain {
+    @SuppressFBWarnings(value = "SLF4J_LOGGER_SHOULD_BE_PRIVATE", justification = "API stability")
+    static final Logger LOG = LoggerFactory.getLogger(PingPongTransactionChain.class);
 
     public PingPongTransactionChain(final Function<DOMTransactionChainListener, DOMTransactionChain> delegateFactory,
             final DOMTransactionChainListener listener) {
-        this.listener = requireNonNull(listener);
-        delegate = delegateFactory.apply(new DOMTransactionChainListener() {
-            @Override
-            public void onTransactionChainFailed(final DOMTransactionChain chain,
-                    final DOMDataTreeTransaction transaction, final Throwable cause) {
-                LOG.debug("Transaction chain {} reported failure in {}", chain, transaction, cause);
-                delegateFailed(chain, cause);
-            }
-
-            @Override
-            public void onTransactionChainSuccessful(final DOMTransactionChain chain) {
-                delegateSuccessful(chain);
-            }
-        });
-    }
-
-    void delegateSuccessful(final DOMTransactionChain chain) {
-        final Entry<PingPongTransaction, Throwable> canceled;
-        synchronized (this) {
-            // This looks weird, but we need not hold the lock while invoking callbacks
-            canceled = deadTx;
-        }
-
-        if (canceled == null) {
-            listener.onTransactionChainSuccessful(this);
-            return;
-        }
-
-        // Backend shutdown successful, but we have a batch of transactions we have to report as dead due to the
-        // user calling cancel().
-        final PingPongTransaction tx = canceled.getKey();
-        final Throwable cause = canceled.getValue();
-        LOG.debug("Transaction chain {} successful, failing cancelled transaction {}", chain, tx, cause);
-
-        listener.onTransactionChainFailed(this, tx.getFrontendTransaction(), cause);
-        tx.onFailure(cause);
-    }
-
-    void delegateFailed(final DOMTransactionChain chain, final Throwable cause) {
-
-        final DOMDataTreeReadWriteTransaction frontend;
-        final PingPongTransaction tx = inflightTx;
-        if (tx == null) {
-            LOG.warn("Transaction chain {} failed with no pending transactions", chain);
-            frontend = null;
-        } else {
-            frontend = tx.getFrontendTransaction();
-        }
-
-        listener.onTransactionChainFailed(this, frontend, cause);
-
-        synchronized (this) {
-            failed = true;
-
-            /*
-             * If we do not have a locked transaction, we need to ensure that the backend transaction is cancelled.
-             * Otherwise we can defer until the user calls us.
-             */
-            if (lockedTx == null) {
-                processIfReady();
-            }
-        }
-    }
-
-    private synchronized @NonNull PingPongTransaction slowAllocateTransaction() {
-        checkState(shutdownTx == null, "Transaction chain %s has been shut down", this);
-
-        if (deadTx != null) {
-            throw new IllegalStateException(String.format(
-                "Transaction chain %s has failed due to transaction %s being canceled", this, deadTx.getKey()),
-                deadTx.getValue());
-        }
-
-        final DOMDataTreeReadWriteTransaction delegateTx = delegate.newReadWriteTransaction();
-        final PingPongTransaction newTx = new PingPongTransaction(delegateTx);
-
-        final Object witness = LOCKED_TX.compareAndExchange(this, null, newTx);
-        if (witness != null) {
-            delegateTx.cancel();
-            throw new IllegalStateException(
-                    String.format("New transaction %s raced with transaction %s", newTx, witness));
-        }
-
-        return newTx;
-    }
-
-    private @Nullable PingPongTransaction acquireReadyTx() {
-        return (PingPongTransaction) READY_TX.getAndSet(this, null);
-    }
-
-    private @NonNull PingPongTransaction allocateTransaction() {
-        // Step 1: acquire current state
-        final PingPongTransaction oldTx = acquireReadyTx();
-
-        // Slow path: allocate a delegate transaction
-        if (oldTx == null) {
-            return slowAllocateTransaction();
-        }
-
-        // Fast path: reuse current transaction. We will check failures and similar on commit().
-        final Object witness = LOCKED_TX.compareAndExchange(this, null, oldTx);
-        if (witness != null) {
-            // Ouch. Delegate chain has not detected a duplicate transaction allocation. This is the best we can do.
-            oldTx.getTransaction().cancel();
-            throw new IllegalStateException(String.format("Reusable transaction %s raced with transaction %s", oldTx,
-                witness));
-        }
-
-        return oldTx;
-    }
-
-    /**
-     * This forces allocateTransaction() on a slow path, which has to happen after this method has completed executing.
-     * Also inflightTx may be updated outside the lock, hence we need to re-check.
-     */
-    @Holding("this")
-    private void processIfReady() {
-        if (inflightTx == null) {
-            final PingPongTransaction tx = acquireReadyTx();
-            if (tx != null) {
-                processTransaction(tx);
-            }
-        }
-    }
-
-    /**
-     * Process a ready transaction. The caller needs to ensure that each transaction is seen only once by this method.
-     *
-     * @param tx Transaction which needs processing.
-     */
-    @Holding("this")
-    private void processTransaction(final @NonNull PingPongTransaction tx) {
-        if (failed) {
-            LOG.debug("Cancelling transaction {}", tx);
-            tx.getTransaction().cancel();
-            return;
-        }
-
-        LOG.debug("Submitting transaction {}", tx);
-        final Object witness = INFLIGHT_TX.compareAndExchange(this, null, tx);
-        if (witness != null) {
-            LOG.warn("Submitting transaction {} while {} is still running", tx, witness);
-        }
-
-        tx.getTransaction().commit().addCallback(new FutureCallback<CommitInfo>() {
-            @Override
-            public void onSuccess(final CommitInfo result) {
-                transactionSuccessful(tx, result);
-            }
-
-            @Override
-            public void onFailure(final Throwable throwable) {
-                transactionFailed(tx, throwable);
-            }
-        }, MoreExecutors.directExecutor());
-    }
-
-    /*
-     * We got invoked from the data store thread. We need to do two things:
-     * 1) release the in-flight transaction
-     * 2) process the potential next transaction
-     *
-     * We have to perform 2) under lock. We could perform 1) without locking, but that means the CAS result may
-     * not be accurate, as a user thread may submit the ready transaction before we acquire the lock -- and checking
-     * for next transaction is not enough, as that may have also be allocated (as a result of a quick
-     * submit/allocate/submit between 1) and 2)). Hence we'd end up doing the following:
-     * 1) CAS of inflightTx
-     * 2) take lock
-     * 3) volatile read of inflightTx
-     *
-     * Rather than doing that, we keep this method synchronized, hence performing only:
-     * 1) take lock
-     * 2) CAS of inflightTx
-     *
-     * Since the user thread is barred from submitting the transaction (in processIfReady), we can then proceed with
-     * the knowledge that inflightTx is null -- processTransaction() will still do a CAS, but that is only for
-     * correctness.
-     */
-    private synchronized void processNextTransaction(final PingPongTransaction tx) {
-        final Object witness = INFLIGHT_TX.compareAndExchange(this, tx, null);
-        checkState(witness == tx, "Completed transaction %s while %s was submitted", tx, witness);
-
-        final PingPongTransaction nextTx = acquireReadyTx();
-        if (nextTx == null) {
-            final PingPongTransaction local = shutdownTx;
-            if (local != null) {
-                processTransaction(local);
-                delegate.close();
-                shutdownTx = null;
-            }
-        } else {
-            processTransaction(nextTx);
-        }
-    }
-
-    void transactionSuccessful(final PingPongTransaction tx, final CommitInfo result) {
-        LOG.debug("Transaction {} completed successfully", tx);
-
-        tx.onSuccess(result);
-        processNextTransaction(tx);
-    }
-
-    void transactionFailed(final PingPongTransaction tx, final Throwable throwable) {
-        LOG.debug("Transaction {} failed", tx, throwable);
-
-        tx.onFailure(throwable);
-        processNextTransaction(tx);
-    }
-
-    void readyTransaction(final @NonNull PingPongTransaction tx) {
-        // First mark the transaction as not locked.
-        final Object lockedWitness = LOCKED_TX.compareAndExchange(this, tx, null);
-        checkState(lockedWitness == tx, "Attempted to submit transaction %s while we have %s", tx, lockedWitness);
-        LOG.debug("Transaction {} unlocked", tx);
-
-        /*
-         * The transaction is ready. It will then be picked up by either next allocation,
-         * or a background transaction completion callback.
-         */
-        final Object readyWitness = READY_TX.compareAndExchange(this, null, tx);
-        checkState(readyWitness == null, "Transaction %s collided on ready state with %s", tx, readyWitness);
-        LOG.debug("Transaction {} readied", tx);
-
-        /*
-         * We do not see a transaction being in-flight, so we need to take care of dispatching
-         * the transaction to the backend. We are in the ready case, we cannot short-cut
-         * the checking of readyTx, as an in-flight transaction may have completed between us
-         * setting the field above and us checking.
-         */
-        if (inflightTx == null) {
-            synchronized (this) {
-                processIfReady();
-            }
-        }
-    }
-
-    /**
-     * Transaction cancellation is a heavyweight operation. We only support cancelation of a locked transaction
-     * and return false for everything else. Cancelling such a transaction will result in all transactions in the
-     * batch to be cancelled.
-     *
-     * @param tx Backend shared transaction
-     * @param frontendTx transaction
-     * @return {@code true} if the transaction was cancelled successfully
-     */
-    synchronized boolean cancelTransaction(final PingPongTransaction tx,
-            final DOMDataTreeReadWriteTransaction frontendTx) {
-        // Attempt to unlock the operation.
-        final Object witness = LOCKED_TX.compareAndExchange(this, tx, null);
-        verify(witness == tx, "Cancelling transaction %s collided with locked transaction %s", tx, witness);
-
-        // Cancel the backend transaction, so we do not end up leaking it.
-        final boolean backendCancelled = tx.getTransaction().cancel();
-
-        if (failed) {
-            // The transaction has failed, this is probably the user just clearing up the transaction they had. We have
-            // already cancelled the transaction anyway,
-            return true;
-        }
-
-        // We have dealt with cancelling the backend transaction and have unlocked the transaction. Since we are still
-        // inside the synchronized block, any allocations are blocking on the slow path. Now we have to decide the fate
-        // of this transaction chain.
-        //
-        // If there are no other frontend transactions in this batch we are aligned with backend state and we can
-        // continue processing.
-        if (frontendTx.equals(tx.getFrontendTransaction())) {
-            if (backendCancelled) {
-                LOG.debug("Cancelled transaction {} was head of the batch, resuming processing", tx);
-                return true;
-            }
-
-            // Backend refused to cancel the transaction. Reinstate it to locked state.
-            final Object reinstateWitness = LOCKED_TX.compareAndExchange(this, null, tx);
-            verify(reinstateWitness == null, "Reinstating transaction %s collided with locked transaction %s", tx,
-                reinstateWitness);
-            return false;
-        }
-
-        if (!backendCancelled) {
-            LOG.warn("Backend transaction cannot be cancelled during cancellation of {}, attempting to continue", tx);
-        }
-
-        // There are multiple frontend transactions in this batch. We have to report them as failed, which dooms this
-        // transaction chain, too. Since we just came off of a locked transaction, we do not have a ready transaction
-        // at the moment, but there may be some transaction in-flight. So we proceed to shutdown the backend chain
-        // and mark the fact that we should be turning its completion into a failure.
-        deadTx = Map.entry(tx, new CancellationException("Transaction " + frontendTx + " canceled").fillInStackTrace());
-        delegate.close();
-        return true;
-    }
-
-    @Override
-    public synchronized void close() {
-        final PingPongTransaction notLocked = lockedTx;
-        checkState(notLocked == null, "Attempted to close chain with outstanding transaction %s", notLocked);
-
-        // This is not reliable, but if we observe it to be null and the process has already completed,
-        // the backend transaction chain will throw the appropriate error.
-        checkState(shutdownTx == null, "Attempted to close an already-closed chain");
-
-        // This may be a reaction to our failure callback, in that case the backend is already shutdown
-        if (deadTx != null) {
-            LOG.debug("Delegate {} is already closed due to failure {}", delegate, deadTx);
-            return;
-        }
-
-        // Force allocations on slow path, picking up a potentially-outstanding transaction
-        final PingPongTransaction tx = acquireReadyTx();
-
-        if (tx != null) {
-            // We have one more transaction, which needs to be processed somewhere. If we do not
-            // a transaction in-flight, we need to push it down ourselves.
-            // If there is an in-flight transaction we will schedule this last one into a dedicated
-            // slot. Allocation slow path will check its presence and fail, the in-flight path will
-            // pick it up, submit and immediately close the chain.
-            if (inflightTx == null) {
-                processTransaction(tx);
-                delegate.close();
-            } else {
-                shutdownTx = tx;
-            }
-        } else {
-            // Nothing outstanding, we can safely shutdown
-            delegate.close();
-        }
-    }
-
-    @Override
-    public DOMDataTreeReadTransaction newReadOnlyTransaction() {
-        return new PingPongReadTransaction(allocateTransaction());
-    }
-
-    @Override
-    public DOMDataTreeReadWriteTransaction newReadWriteTransaction() {
-        final PingPongTransaction tx = allocateTransaction();
-        final DOMDataTreeReadWriteTransaction ret = new PingPongReadWriteTransaction(tx);
-        tx.recordFrontendTransaction(ret);
-        return ret;
-    }
-
-    @Override
-    public DOMDataTreeWriteTransaction newWriteOnlyTransaction() {
-        return newReadWriteTransaction();
-    }
-
-    private final class PingPongReadTransaction implements DOMDataTreeReadTransaction {
-        private final @NonNull PingPongTransaction tx;
-
-        PingPongReadTransaction(final PingPongTransaction tx) {
-            this.tx = requireNonNull(tx);
-        }
-
-        @Override
-        public FluentFuture<Optional<NormalizedNode>> read(final LogicalDatastoreType store,
-                final YangInstanceIdentifier path) {
-            return tx.getTransaction().read(store, path);
-        }
-
-        @Override
-        public FluentFuture<Boolean> exists(final LogicalDatastoreType store, final YangInstanceIdentifier path) {
-            return tx.getTransaction().exists(store, path);
-        }
-
-        @Override
-        public Object getIdentifier() {
-            return tx.getTransaction().getIdentifier();
-        }
-
-        @Override
-        public void close() {
-            readyTransaction(tx);
-        }
-    }
-
-    private final class PingPongReadWriteTransaction extends ForwardingDOMDataReadWriteTransaction {
-        private final @NonNull PingPongTransaction tx;
-
-        private boolean isOpen = true;
-
-        PingPongReadWriteTransaction(final PingPongTransaction tx) {
-            this.tx = requireNonNull(tx);
-        }
-
-        @Override
-        public FluentFuture<? extends CommitInfo> commit() {
-            readyTransaction(tx);
-            isOpen = false;
-            return tx.getCommitFuture().transform(ignored -> CommitInfo.empty(), MoreExecutors.directExecutor());
-        }
-
-        @Override
-        public boolean cancel() {
-            if (isOpen && cancelTransaction(tx, this)) {
-                isOpen = false;
-                return true;
-            }
-            return false;
-        }
-
-        @Override
-        protected DOMDataTreeReadWriteTransaction delegate() {
-            return tx.getTransaction();
-        }
+        super(delegateFactory, listener);
     }
 }