Move AbstractDOMDataBroker to mdsal-dom-spi
[mdsal.git] / dom / mdsal-dom-broker / src / main / java / org / opendaylight / mdsal / dom / broker / CommitCoordinationTask.java
index 694b8411a81f082eb533d2e2598359d875a018b6..330f037c183e53c966970cefa2e8ba36a1f8a257 100644 (file)
  * 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 org.opendaylight.mdsal.dom.spi.store.DOMStoreThreePhaseCommitCohort;
+import static java.util.Objects.requireNonNull;
 
-import org.opendaylight.mdsal.common.api.TransactionCommitFailedException;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
-import com.google.common.base.Preconditions;
 import com.google.common.base.Throwables;
-import com.google.common.util.concurrent.Futures;
-import com.google.common.util.concurrent.ListenableFuture;
-import java.util.Collection;
+import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
 import java.util.concurrent.Callable;
 import java.util.concurrent.ExecutionException;
+import org.opendaylight.mdsal.common.api.CommitInfo;
+import org.opendaylight.mdsal.common.api.TransactionCommitFailedException;
+import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
+import org.opendaylight.mdsal.dom.spi.TransactionCommitFailedExceptionMapper;
+import org.opendaylight.mdsal.dom.spi.store.DOMStoreThreePhaseCommitCohort;
 import org.opendaylight.yangtools.util.DurationStatisticsTracker;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 /**
- * Implementation of blocking three-phase commit-coordination tasks without
- * support of cancellation.
+ * Implementation of blocking three-phase commit-coordination tasks without support of cancellation.
  */
-final class CommitCoordinationTask implements Callable<Void> {
-    private static enum Phase {
-        canCommit,
-        preCommit,
-        doCommit,
+sealed class CommitCoordinationTask implements Callable<CommitInfo> {
+    static final class WithTracker extends CommitCoordinationTask {
+        private final DurationStatisticsTracker commitStatTracker;
+
+        WithTracker(final DOMDataTreeWriteTransaction transaction, final DOMStoreThreePhaseCommitCohort cohort,
+                final DurationStatisticsTracker commitStatTracker) {
+            super(transaction, cohort);
+            this.commitStatTracker = requireNonNull(commitStatTracker);
+        }
+
+        @Override
+        public CommitInfo call() throws TransactionCommitFailedException {
+            final long startTime = System.nanoTime();
+
+            try {
+                return super.call();
+            } finally {
+                commitStatTracker.addDuration(System.nanoTime() - startTime);
+            }
+        }
+    }
+
+    private enum Phase {
+        CAN_COMMIT,
+        PRE_COMMIT,
+        DO_COMMIT
     }
 
     private static final Logger LOG = LoggerFactory.getLogger(CommitCoordinationTask.class);
-    private final Collection<DOMStoreThreePhaseCommitCohort> cohorts;
-    private final DurationStatisticsTracker commitStatTracker;
+
+    private final DOMStoreThreePhaseCommitCohort cohort;
     private final DOMDataTreeWriteTransaction tx;
 
-    public CommitCoordinationTask(final DOMDataTreeWriteTransaction transaction,
-            final Collection<DOMStoreThreePhaseCommitCohort> cohorts,
-            final DurationStatisticsTracker commitStatTracker) {
-        this.tx = Preconditions.checkNotNull(transaction, "transaction must not be null");
-        this.cohorts = Preconditions.checkNotNull(cohorts, "cohorts must not be null");
-        this.commitStatTracker = commitStatTracker;
+    CommitCoordinationTask(final DOMDataTreeWriteTransaction transaction, final DOMStoreThreePhaseCommitCohort cohort) {
+        tx = requireNonNull(transaction, "transaction must not be null");
+        this.cohort = requireNonNull(cohort, "cohort must not be null");
     }
 
     @Override
-    public Void call() throws TransactionCommitFailedException {
-        final long startTime = commitStatTracker != null ? System.nanoTime() : 0;
-
-        Phase phase = Phase.canCommit;
-
+    public CommitInfo call() throws TransactionCommitFailedException {
+        var phase = Phase.CAN_COMMIT;
         try {
             LOG.debug("Transaction {}: canCommit Started", tx.getIdentifier());
             canCommitBlocking();
 
-            phase = Phase.preCommit;
+            phase = Phase.PRE_COMMIT;
             LOG.debug("Transaction {}: preCommit Started", tx.getIdentifier());
             preCommitBlocking();
 
-            phase = Phase.doCommit;
+            phase = Phase.DO_COMMIT;
             LOG.debug("Transaction {}: doCommit Started", tx.getIdentifier());
             commitBlocking();
 
             LOG.debug("Transaction {}: doCommit completed", tx.getIdentifier());
-            return null;
+            return CommitInfo.empty();
         } catch (final TransactionCommitFailedException e) {
             LOG.warn("Tx: {} Error during phase {}, starting Abort", tx.getIdentifier(), phase, e);
             abortBlocking(e);
             throw e;
-        } finally {
-            if (commitStatTracker != null) {
-                commitStatTracker.addDuration(System.nanoTime() - startTime);
-            }
         }
     }
 
     /**
+     * Invokes canCommit on underlying cohort and blocks till the result is returned.
      *
-     * Invokes canCommit on underlying cohorts and blocks till
-     * all results are returned.
-     *
-     * Valid state transition is from SUBMITTED to CAN_COMMIT,
-     * if currentPhase is not SUBMITTED throws IllegalStateException.
-     *
-     * @throws TransactionCommitFailedException
-     *             If one of cohorts failed can Commit
+     * <p>
+     * Valid state transition is from SUBMITTED to CAN_COMMIT, if currentPhase is not SUBMITTED throws
+     * IllegalStateException.
      *
+     * @throws TransactionCommitFailedException If cohort fails Commit
      */
+    @SuppressFBWarnings("BC_UNCONFIRMED_CAST_OF_RETURN_VALUE")
     private void canCommitBlocking() throws TransactionCommitFailedException {
-        for (final ListenableFuture<?> canCommit : canCommitAll()) {
-            try {
-                final Boolean result = (Boolean)canCommit.get();
-                if (result == null || !result) {
-                    throw new TransactionCommitFailedException("Can Commit failed, no detailed cause available.");
-                }
-            } catch (InterruptedException | ExecutionException e) {
-                throw TransactionCommitFailedExceptionMapper.CAN_COMMIT_ERROR_MAPPER.apply(e);
-            }
+        final var future = cohort.canCommit();
+        final Boolean result;
+        try {
+            result = future.get();
+        } catch (InterruptedException | ExecutionException e) {
+            throw TransactionCommitFailedExceptionMapper.CAN_COMMIT_ERROR_MAPPER.apply(e);
         }
-    }
 
-    /**
-     *
-     * Invokes canCommit on underlying cohorts and returns composite future
-     * which will contains {@link Boolean#TRUE} only and only if
-     * all cohorts returned true.
-     *
-     * Valid state transition is from SUBMITTED to CAN_COMMIT,
-     * if currentPhase is not SUBMITTED throws IllegalStateException.
-     *
-     * @return List of all cohorts futures from can commit phase.
-     *
-     */
-    private ListenableFuture<?>[] canCommitAll() {
-        final ListenableFuture<?>[] ops = new ListenableFuture<?>[cohorts.size()];
-        int i = 0;
-        for (final DOMStoreThreePhaseCommitCohort cohort : cohorts) {
-            ops[i++] = cohort.canCommit();
+        if (!Boolean.TRUE.equals(result)) {
+            throw new TransactionCommitFailedException("Can Commit failed, no detailed cause available.");
         }
-        return ops;
     }
 
     /**
+     * Invokes preCommit on underlying cohort and blocks until the result is returned.
      *
-     * Invokes preCommit on underlying cohorts and blocks till
-     * all results are returned.
-     *
-     * Valid state transition is from CAN_COMMIT to PRE_COMMIT, if current
-     * state is not CAN_COMMIT
-     * throws IllegalStateException.
-     *
-     * @throws TransactionCommitFailedException
-     *             If one of cohorts failed preCommit
+     * <p>
+     * Valid state transition is from CAN_COMMIT to PRE_COMMIT, if current state is not CAN_COMMIT throws
+     * IllegalStateException.
      *
+     * @throws TransactionCommitFailedException If cohort fails preCommit
      */
+    @SuppressFBWarnings("BC_UNCONFIRMED_CAST_OF_RETURN_VALUE")
     private void preCommitBlocking() throws TransactionCommitFailedException {
-        final ListenableFuture<?>[] preCommitFutures = preCommitAll();
         try {
-            for(final ListenableFuture<?> future : preCommitFutures) {
-                future.get();
-            }
+            cohort.preCommit().get();
         } catch (InterruptedException | ExecutionException e) {
             throw TransactionCommitFailedExceptionMapper.PRE_COMMIT_MAPPER.apply(e);
         }
     }
 
     /**
+     * Invokes commit on underlying cohort and blocks until result is returned.
      *
-     * Invokes preCommit on underlying cohorts and returns future
-     * which will complete once all preCommit on cohorts completed or
-     * failed.
-     *
-     *
-     * Valid state transition is from CAN_COMMIT to PRE_COMMIT, if current
-     * state is not CAN_COMMIT
-     * throws IllegalStateException.
-     *
-     * @return List of all cohorts futures from can commit phase.
-     *
-     */
-    private ListenableFuture<?>[] preCommitAll() {
-        final ListenableFuture<?>[] ops = new ListenableFuture<?>[cohorts.size()];
-        int i = 0;
-        for (final DOMStoreThreePhaseCommitCohort cohort : cohorts) {
-            ops[i++] = cohort.preCommit();
-        }
-        return ops;
-    }
-
-    /**
-     *
-     * Invokes commit on underlying cohorts and blocks till
-     * all results are returned.
-     *
-     * Valid state transition is from PRE_COMMIT to COMMIT, if not throws
-     * IllegalStateException.
-     *
-     * @throws TransactionCommitFailedException
-     *             If one of cohorts failed preCommit
+     * <p>
+     * Valid state transition is from PRE_COMMIT to COMMIT, if not throws IllegalStateException.
      *
+     * @throws TransactionCommitFailedException If cohort fails preCommit
      */
+    @SuppressFBWarnings("BC_UNCONFIRMED_CAST_OF_RETURN_VALUE")
     private void commitBlocking() throws TransactionCommitFailedException {
-        final ListenableFuture<?>[] commitFutures = commitAll();
         try {
-            for(final ListenableFuture<?> future : commitFutures) {
-                future.get();
-            }
+            cohort.commit().get();
         } catch (InterruptedException | ExecutionException e) {
             throw TransactionCommitFailedExceptionMapper.COMMIT_ERROR_MAPPER.apply(e);
         }
     }
 
-    /**
-     *
-     * Invokes commit on underlying cohorts and returns future which
-     * completes
-     * once all commits on cohorts are completed.
-     *
-     * Valid state transition is from PRE_COMMIT to COMMIT, if not throws
-     * IllegalStateException
-     *
-     * @return List of all cohorts futures from can commit phase.
-     */
-    private ListenableFuture<?>[] commitAll() {
-        final ListenableFuture<?>[] ops = new ListenableFuture<?>[cohorts.size()];
-        int i = 0;
-        for (final DOMStoreThreePhaseCommitCohort cohort : cohorts) {
-            ops[i++] = cohort.commit();
-        }
-        return ops;
-    }
-
     /**
      * Aborts transaction.
      *
-     * Invokes {@link DOMStoreThreePhaseCommitCohort#abort()} on all
-     * cohorts, blocks
-     * for all results. If any of the abort failed throws
-     * IllegalStateException,
-     * which will contains originalCause as suppressed Exception.
+     * <p>
+     * Invokes {@link DOMStoreThreePhaseCommitCohort#abort()} on underlying cohort, blocks the results. If
+     * abort failed throws IllegalStateException, which will contains originalCause as suppressed Exception.
      *
-     * If aborts we're successful throws supplied exception
+     * <p>
+     * If abort was successful throws supplied exception
      *
-     * @param originalCause
-     *            Exception which should be used to fail transaction for
-     *            consumers of transaction
-     *            future and listeners of transaction failure.
-     * @param phase phase in which the problem ensued
-     * @throws TransactionCommitFailedException
-     *             on invocation of this method.
-     *             originalCa
-     * @throws IllegalStateException
-     *             if abort failed.
+     * @param originalCause Exception which should be used to fail transaction for consumers of transaction future
+     *                      and listeners of transaction failure.
+     * @throws IllegalStateException if abort failed.
+     * @throws TransactionCommitFailedException on invocation of this method.
      */
-    private void abortBlocking(final TransactionCommitFailedException originalCause) throws TransactionCommitFailedException {
+    private void abortBlocking(final TransactionCommitFailedException originalCause)
+            throws TransactionCommitFailedException {
         Exception cause = originalCause;
         try {
-            abortAsyncAll().get();
+            cohort.abort().get();
         } catch (InterruptedException | ExecutionException e) {
             LOG.error("Tx: {} Error during Abort.", tx.getIdentifier(), e);
             cause = new IllegalStateException("Abort failed.", e);
@@ -247,28 +173,4 @@ final class CommitCoordinationTask implements Callable<Void> {
         }
         Throwables.propagateIfPossible(cause, TransactionCommitFailedException.class);
     }
-
-    /**
-     * Invokes abort on underlying cohorts and returns future which
-     * completes once all abort on cohorts are completed.
-     *
-     * @return Future which will complete once all cohorts completed
-     *         abort.
-     */
-    @SuppressWarnings({"unchecked", "rawtypes"})
-    private ListenableFuture<Void> abortAsyncAll() {
-
-        final ListenableFuture<?>[] ops = new ListenableFuture<?>[cohorts.size()];
-        int i = 0;
-        for (final DOMStoreThreePhaseCommitCohort cohort : cohorts) {
-            ops[i++] = cohort.abort();
-        }
-
-        /*
-         * We are returning all futures as list, not only succeeded ones in
-         * order to fail composite future if any of them failed.
-         * See Futures.allAsList for this description.
-         */
-        return (ListenableFuture) Futures.allAsList(ops);
-    }
 }