Examples with TransactionException cn.taketoday.transaction.TransactionException used on opensource projects

Example 6 with TransactionException

use of cn.taketoday.transaction.TransactionException in project today-infrastructure by TAKETODAY.

the class AbstractReactiveTransactionManager method processCommit.

/**
 * Process an actual commit.
 * Rollback-only flags have already been checked and applied.
 *
 * @param synchronizationManager the synchronization manager bound to the current transaction
 * @param status object representing the transaction
 * @throws TransactionException in case of commit failure
 */
private Mono<Void> processCommit(TransactionSynchronizationManager synchronizationManager, GenericReactiveTransaction status) throws TransactionException {
    AtomicBoolean beforeCompletionInvoked = new AtomicBoolean();
    Mono<Object> commit = prepareForCommit(synchronizationManager, status).then(triggerBeforeCommit(synchronizationManager, status)).then(triggerBeforeCompletion(synchronizationManager, status)).then(Mono.defer(() -> {
        beforeCompletionInvoked.set(true);
        if (status.isNewTransaction()) {
            if (status.isDebug()) {
                logger.debug("Initiating transaction commit");
            }
            return doCommit(synchronizationManager, status);
        }
        return Mono.empty();
    })).then(Mono.empty().onErrorResume(ex -> {
        Mono<Object> propagateException = Mono.error(ex);
        // Store result in a local variable in order to appease the
        // Eclipse compiler with regard to inferred generics.
        Mono<Object> result = propagateException;
        if (ErrorPredicates.UNEXPECTED_ROLLBACK.test(ex)) {
            result = triggerAfterCompletion(synchronizationManager, status, TransactionSynchronization.STATUS_ROLLED_BACK).then(propagateException);
        } else if (ErrorPredicates.TRANSACTION_EXCEPTION.test(ex)) {
            result = triggerAfterCompletion(synchronizationManager, status, TransactionSynchronization.STATUS_UNKNOWN).then(propagateException);
        } else if (ErrorPredicates.RUNTIME_OR_ERROR.test(ex)) {
            Mono<Void> mono;
            if (!beforeCompletionInvoked.get()) {
                mono = triggerBeforeCompletion(synchronizationManager, status);
            } else {
                mono = Mono.empty();
            }
            result = mono.then(doRollbackOnCommitException(synchronizationManager, status, ex)).then(propagateException);
        }
        return result;
    })).then(Mono.defer(() -> triggerAfterCommit(synchronizationManager, status).onErrorResume(ex -> triggerAfterCompletion(synchronizationManager, status, TransactionSynchronization.STATUS_COMMITTED).then(Mono.error(ex))).then(triggerAfterCompletion(synchronizationManager, status, TransactionSynchronization.STATUS_COMMITTED))));
    return commit.onErrorResume(ex -> cleanupAfterCompletion(synchronizationManager, status).then(Mono.error(ex))).then(cleanupAfterCompletion(synchronizationManager, status));
}