com.github.drinkjava2.jtransactions.tinytx.TinyTx Maven / Gradle / Ivy

Go to download
package com.github.drinkjava2.jtransactions.tinytx;

import java.sql.Connection;
import java.sql.SQLException;

import javax.sql.DataSource;

import org.aopalliance.intercept.MethodInterceptor;
import org.aopalliance.intercept.MethodInvocation;

/**
 * A transaction MethodInterceptor
 * 
 * @author Yong Zhu
 * @since 1.0.0
 */
public class TinyTx implements MethodInterceptor {
	private static final TinyTxLogger logger = TinyTxLogger.getLog(TinyTx.class);
	private static final TinyTxConnectionManager cm = TinyTxConnectionManager.instance();

	private int transactionIsolation = Connection.TRANSACTION_READ_COMMITTED;
	private DataSource ds;

	public TinyTx() {
	}

	public TinyTx(DataSource ds) {
		this.ds = ds;
	}

	public TinyTx(DataSource ds, Integer transactionIsolation) {
		this.ds = ds;
		this.transactionIsolation = transactionIsolation;
	}

	@Override
	public Object invoke(MethodInvocation caller) {//NOSONAR
		if (cm.isInTransaction(ds)) {
			try {
				return caller.proceed();
			} catch (Throwable t) {
				throw new TinyTxRuntimeException(t);
			}
		} else {
			Connection conn;
			try {
				conn = cm.getConnection(ds);
				TinyTxRuntimeException.assertNotNull(conn, "Connection can not get from DataSource in invoke method");
			} catch (Exception e) {
				throw new TinyTxRuntimeException(e);
			}
			Object invokeResult = null;
			try {
				cm.startTransaction(ds, conn);
				conn.setTransactionIsolation(transactionIsolation);
				conn.setAutoCommit(false);
				invokeResult = caller.proceed();
				conn.commit();
			} catch (Throwable t) {
				if (conn != null)
					try {
						conn.rollback();
					} catch (Exception e1) {
						logger.warn(e1.getMessage());
					}
				throw new TinyTxRuntimeException("TinyTx found a runtime Exception, transaction rollbacked.", t);
			} finally {
				cm.endTransaction(ds);
				SQLException closeExp = null;
				try {
					cm.releaseConnection(conn, ds);
				} catch (SQLException e) {
					closeExp = e;
				}
				if (closeExp != null)
					throw new TinyTxRuntimeException("Exception happen when release connection.", closeExp);//NOSONAR
			}
			return invokeResult;
		}
	}

}