org.flowable.spring.SpringTransactionContext Maven / Gradle / Ivy
/* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.flowable.spring;
import org.flowable.engine.common.impl.cfg.TransactionState;
import org.flowable.engine.impl.cfg.TransactionContext;
import org.flowable.engine.impl.cfg.TransactionListener;
import org.flowable.engine.impl.interceptor.CommandContext;
import org.springframework.core.Ordered;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.support.TransactionSynchronization;
import org.springframework.transaction.support.TransactionSynchronizationManager;
/**
* @author Frederik Heremans
* @author Joram Barrez
*/
public class SpringTransactionContext implements TransactionContext {
protected PlatformTransactionManager transactionManager;
protected CommandContext commandContext;
protected Integer transactionSynchronizationAdapterOrder;
public SpringTransactionContext(PlatformTransactionManager transactionManager, CommandContext commandContext) {
this(transactionManager, commandContext, null);
}
public SpringTransactionContext(PlatformTransactionManager transactionManager, CommandContext commandContext, Integer transactionSynchronizationAdapterOrder) {
this.transactionManager = transactionManager;
this.commandContext = commandContext;
if (transactionSynchronizationAdapterOrder != null) {
this.transactionSynchronizationAdapterOrder = transactionSynchronizationAdapterOrder;
} else {
// Revert to default, which is a high number as the behaviour prior
// to adding the order would
// case the TransactionSynchronizationAdapter to be called AFTER all
// Adapters that implement Ordered
this.transactionSynchronizationAdapterOrder = Integer.MAX_VALUE;
}
}
public void commit() {
// Do nothing, transaction is managed by spring
}
public void rollback() {
// Just in case the rollback isn't triggered by an
// exception, we mark the current transaction rollBackOnly.
transactionManager.getTransaction(null).setRollbackOnly();
}
public void addTransactionListener(final TransactionState transactionState, final TransactionListener transactionListener) {
if (transactionState == TransactionState.COMMITTING) {
TransactionSynchronizationManager.registerSynchronization(new TransactionSynchronizationAdapter() {
@Override
public void beforeCommit(boolean readOnly) {
transactionListener.execute(commandContext);
}
});
} else if (transactionState == TransactionState.COMMITTED) {
TransactionSynchronizationManager.registerSynchronization(new TransactionSynchronizationAdapter() {
@Override
public void afterCommit() {
transactionListener.execute(commandContext);
}
});
} else if (transactionState == TransactionState.ROLLINGBACK) {
TransactionSynchronizationManager.registerSynchronization(new TransactionSynchronizationAdapter() {
@Override
public void beforeCompletion() {
transactionListener.execute(commandContext);
}
});
} else if (transactionState == TransactionState.ROLLED_BACK) {
TransactionSynchronizationManager.registerSynchronization(new TransactionSynchronizationAdapter() {
@Override
public void afterCompletion(int status) {
if (TransactionSynchronization.STATUS_ROLLED_BACK == status) {
transactionListener.execute(commandContext);
}
}
});
}
}
protected abstract class TransactionSynchronizationAdapter implements TransactionSynchronization, Ordered {
public void suspend() {
}
public void resume() {
}
public void flush() {
}
public void beforeCommit(boolean readOnly) {
}
public void beforeCompletion() {
}
public void afterCommit() {
}
public void afterCompletion(int status) {
}
@Override
public int getOrder() {
return transactionSynchronizationAdapterOrder;
}
}
}