com.litongjava.db.activerecord.OneConnectionPerThread Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-db Show documentation

Java Database

There is a newer version: 1.4.7

package com.litongjava.db.activerecord;


import java.sql.Connection;

import com.litongjava.jfinal.aop.Interceptor;
import com.litongjava.jfinal.aop.Invocation;

import lombok.extern.slf4j.Slf4j;

/**
 * One Connection Per Thread for one request.

 * warning: can not use this interceptor with transaction feature like Tx, Db.tx(...)
 */
@Slf4j
public class OneConnectionPerThread implements Interceptor {
	
	public void intercept(Invocation inv) {
		Connection conn = DbKit.config.getThreadLocalConnection();
		if (conn != null) {
			inv.invoke();
			return ;
		}
		
		try {
			conn = DbKit.config.getConnection();
			DbKit.config.setThreadLocalConnection(conn);
			inv.invoke();
		}
		catch (RuntimeException e) {
			throw e;
		}
		catch (Exception e) {
			throw new RuntimeException(e);
		}
		finally {
			DbKit.config.removeThreadLocalConnection();
			if (conn != null) {
				try{conn.close();}catch(Exception e){log.error(e.getMessage(), e);};
			}
		}
	}
}