org.beetl.sql.mapper.MapperJavaProxy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sql-mapper Show documentation
The newest version!
package org.beetl.sql.mapper;

import org.beetl.sql.clazz.kit.BeanKit;
import org.beetl.sql.core.SQLManager;
import org.beetl.sql.mapper.annotation.InheritMapper;
import org.beetl.sql.mapper.builder.MapperConfigBuilder;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * Java代理实现.
 * 
 * # 54
 * 封装sqlmanager
 * 
 *
 * @author zhoupan, xiandafu
 */
public class MapperJavaProxy implements InvocationHandler {

	/**
	 * The sql manager.
	 */
	protected SQLManager sqlManager;

	/**
	 * 继承{@code BaseMapper<T>} 接口时给定的泛型T实体类型
	 */
	protected Class entityClass;


	protected MapperConfigBuilder builder;


	/**
	 * 继承了{@link BaseMapper} 的接口
	 */
	protected Class mapperInterface;


	private static final Map PROVIDERS_CACHE = new ConcurrentHashMap();

	/**
	 * The Constructor.
	 */
	public MapperJavaProxy() {

	}

	/**
	 * @param builder
	 * @param sqlManager
	 * @param mapperInterface
	 */
	public MapperJavaProxy(MapperConfigBuilder builder, SQLManager sqlManager, Class mapperInterface) {
		super();
		this.sqlManager = sqlManager;
		this.builder = builder;
		this.mapperInterface(mapperInterface);
		this.mapperInterface = mapperInterface;
	}


	/**
	 * Mapper interface.
	 *
	 * @param mapperInterface the dao2 interface
	 * @return the dao2 proxy
	 */
	public MapperJavaProxy mapperInterface(Class mapperInterface) {
		this.onResolveEntityClassFromMapperInterface(mapperInterface);
		return this;
	}


	/**
	 * Entity class.
	 *
	 * @param entityClass the entity class
	 * @return the dao2 proxy
	 */
	public MapperJavaProxy entityClass(Class entityClass) {
		this.entityClass = entityClass;
		return this;
	}

	/**
	 * Check args.
	 */
	protected void checkArgs() {
	}

	/**
	 * Builds the.
	 *
	 * @return the dao2 proxy
	 */
	public MapperJavaProxy build() {
		this.checkArgs();
		return this;
	}

	/**
	 * 获取 {@code BaseMapper<T>} 接口的泛型T代表的实体类.
	 *
	 * @param mapperInterface 继承BaseMapper的接口
	 */
	protected void onResolveEntityClassFromMapperInterface(Class mapperInterface) {
		if (mapperInterface.isInterface()) {
			this.entityClass = BeanKit.getMapperEntity(mapperInterface);
		} else {
			throw new IllegalArgumentException("mapperInterface is not interface.");
		}
	}


	/**
	 * 获得 Invoke.
	 * @param proxy  the proxy
	 * @param method the method
	 * @param args   the args
	 * @return the object
	 * @throws Throwable the throwable
	 */
	@Override
	public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
		MapperInvoke invoke = null;

		InheritMapper inheritMapper = method.getAnnotation(InheritMapper.class);
		if (inheritMapper == null) {
			//大部分情况下都是使用BaseMapper方法，所以这里的caller 是BaseMapper
			Class caller = method.getDeclaringClass();
			invoke = builder.getAmi(entityClass, caller, method);
		} else {
			/* 不适用method所在的类，而使用传入的mapperInterface构造invoke，这样对应的sqlResource，entity*/
			invoke = builder.getInheritAmi(entityClass, this.mapperInterface, method);
		}
		Object ret = invoke.call(this.sqlManager, this.entityClass, method, args);
		return ret;
	}


}