com.ext_ext.mybatisext.MapperProxyExt Maven / Gradle / Ivy

Go to download
package com.ext_ext.mybatisext;

import java.io.Serializable;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.util.Map;

import org.apache.ibatis.binding.MapperMethod;
import org.apache.ibatis.session.SqlSession;

import com.ext_ext.mybatisext.interceptor.MyBatisInterceptor;
import com.ext_ext.mybatisext.interceptor.MyBatisInvocation;
import com.ext_ext.mybatisext.plugin.paging.PagingPlugin;

/**
 * @author songrubo
 * @version 2013年11月30日 下午12:09:51
 */
public class MapperProxyExt implements InvocationHandler, Serializable {

	private static final long serialVersionUID = -6424540398559729838L;

	private final SqlSession sqlSession;

	private final Class mapperInterface;

	private final Map methodCache;

	private final MyBatisInterceptor[] interceptors;

	private final static ThreadLocal myBatisInvocationHolder = new ThreadLocal();


	public MapperProxyExt(
			SqlSession sqlSession,
			Class mapperInterface,
			Map methodCache,
			MyBatisInterceptor[] interceptors ) {
		this.sqlSession = sqlSession;
		this.mapperInterface = mapperInterface;
		this.methodCache = methodCache;
		this.interceptors = interceptors;
	}


	@Override
	public Object invoke( Object proxy, Method method, Object[] args ) throws Throwable {
		if ( Object.class.equals(method.getDeclaringClass()) ) {
			return method.invoke(this, args);
		}
		try {

			MyBatisInvocation invocation = new MyBatisInvocation(mapperInterface, sqlSession, method, args,
					interceptors, methodCache);
			myBatisInvocationHolder.set(invocation);
			return invocation.execute();
		} finally {
			myBatisInvocationHolder.remove();
			PagingPlugin.removePage();
		}

	}


	public static MyBatisInvocation getMyBatisInvocation() {
		return myBatisInvocationHolder.get();
	}

}