com.buschmais.xo.impl.interceptor.TransactionInterceptor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xo.impl Show documentation
The newest version!
package com.buschmais.xo.impl.interceptor;

import java.lang.reflect.Method;

import com.buschmais.xo.api.Transaction;
import com.buschmais.xo.api.XOException;
import com.buschmais.xo.api.XOTransaction;
import com.buschmais.xo.spi.interceptor.InvocationContext;
import com.buschmais.xo.spi.interceptor.XOInterceptor;

public class TransactionInterceptor implements XOInterceptor {

    private final XOTransaction xoTransaction;
    private final Transaction.TransactionAttribute defaultTransactionAttribute;

    public TransactionInterceptor(XOTransaction xoTransaction, Transaction.TransactionAttribute defaultTransactionAttribute) {
        this.xoTransaction = xoTransaction;
        this.defaultTransactionAttribute = defaultTransactionAttribute;
    }

    @Override
    public boolean isActive() {
        return !Transaction.TransactionAttribute.NONE.equals(defaultTransactionAttribute);
    }

    @Override
    public Object invoke(InvocationContext context) throws Throwable {
        Method method = context.getMethod();
        Transaction.TransactionAttribute transactionAttribute;
        Transaction transaction = method.getAnnotation(Transaction.class);
        if (transaction != null) {
            transactionAttribute = transaction.value();
        } else {
            transactionAttribute = this.defaultTransactionAttribute;
        }
        switch (transactionAttribute) {
        case MANDATORY:
            if (!this.xoTransaction.isActive()) {
                throw new XOException("An active transaction is MANDATORY when calling method '" + method.getDeclaringClass()
                    .getName() + "#" + method.getName() + "'");
            }
            return context.proceed();
        case REQUIRES: {
            if (!this.xoTransaction.isActive()) {
                try {
                    this.xoTransaction.begin();
                    Object result = context.proceed();
                    this.xoTransaction.commit();
                    return result;
                } catch (RuntimeException e) {
                    if (this.xoTransaction.isActive()) {
                        this.xoTransaction.rollback();
                    }
                    throw e;
                } catch (Exception e) {
                    if (this.xoTransaction.isActive()) {
                        this.xoTransaction.commit();
                    }
                    throw e;
                }
            } else {
                return context.proceed();
            }
        }
        case NOT_SUPPORTED:
            return context.proceed();
        default: {
            throw new XOException("Unsupported transaction attribute '" + transactionAttribute + "'");
        }
        }
    }
}