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

Go to download
package com.ext_ext.mybatisext;

import java.lang.reflect.Field;
import java.lang.reflect.Proxy;
import java.util.Map;

import org.apache.ibatis.binding.BindingException;
import org.apache.ibatis.binding.MapperProxyFactory;
import org.apache.ibatis.binding.MapperRegistry;
import org.apache.ibatis.logging.Log;
import org.apache.ibatis.logging.LogFactory;
import org.apache.ibatis.session.Configuration;
import org.apache.ibatis.session.SqlSession;

import com.ext_ext.mybatisext.interceptor.MyBatisInterceptor;

/**
 * @author songrubo
 * @version 2013年11月30日 下午12:04:57
 */
public class MapperRegistryExt extends MapperRegistry {

	private static final Log logger = LogFactory.getLog(MapperRegistryExt.class);

	private Map, MapperProxyFactory> knownMappers;

	MyBatisInterceptor[] interceptors;


	@SuppressWarnings("unchecked")
	public MapperRegistryExt( Configuration config, MyBatisInterceptor[] interceptors ) {
		super(config);
		this.interceptors = interceptors;
		try {
			Field field = MapperRegistry.class.getDeclaredField("knownMappers");
			field.setAccessible(true);
			knownMappers = (Map, MapperProxyFactory>) field.get(this);
		} catch ( Exception e ) {
			logger.error(e.getMessage(), e);
		}
	}


	@SuppressWarnings("unchecked")
	@Override
	public  T getMapper( Class type, SqlSession sqlSession ) {
		final MapperProxyFactory mapperProxyFactory = (MapperProxyFactory) knownMappers.get(type);
		if ( mapperProxyFactory == null ) {
			throw new BindingException("Type " + type + " is not known to the MapperRegistry.");
		}
		try {
			Class mapperInterface = mapperProxyFactory.getMapperInterface();
			final MapperProxyExt mapperProxy = new MapperProxyExt(sqlSession, mapperInterface,
					mapperProxyFactory.getMethodCache(), interceptors);
			return (T) Proxy.newProxyInstance(mapperInterface.getClassLoader(), new Class[ ] { mapperInterface },
				mapperProxy);
		} catch ( Exception e ) {
			throw new BindingException("Error getting mapper instance. Cause: " + e, e);
		}
	}
}