MyBatis源码解读(3)——MapperMethod

时间:2021-12-28 16:29:14

在前面两篇的MyBatis源码解读中,我们一路跟踪到了MapperProxy,知道了尽管是使用了动态代理技术使得我们能直接使用接口方法。为巩固加深动态代理,我们不妨再来回忆一遍何为动态代理。

我相信在初学MyBatis的时候几乎每个人都会发出一个疑问,为什么明明是XXXDao接口,我没有用任何代码实现这个接口,但却能直接使用这个接口的方法。现在清楚了,动态代理。我们来写一个demo小程序来看看。

首先是一个Test.java的接口,只有一个say方法。

 package day_16_proxy;

 /**
* @author 余林丰
*
* 2016年11月16日
*/
public interface Test {
void say();
}

我们现在想像MyBatis那样不用实现它而是直接调用。

test.say();

当然不可能直接实例化一个接口,此时就需要生成一个代理类TestProxy.java,这个类需实现InvocationHandler接口。

 package day_16_proxy;

 import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method; /**
* @author 余林丰
*
* 2016年11月16日
*/
public class TestProxy implements InvocationHandler { /* (non-Javadoc)
* @see java.lang.reflect.InvocationHandler#invoke(java.lang.Object, java.lang.reflect.Method, java.lang.Object[])
*/
@Override
public Object invoke(Object proxy, Method method, Object[] args)
throws Throwable {
if (method.getName().equals("say")){
System.out.println("hello world");
}
return null;
} }

需要实现invoke方法,意思就是“调用”的意思(当然我们想要调用接口中的方法时并不会显示调用invoke)。我们从第19行看到,当调用的方法是say时,输出“hello world”。有了这个TestProxy.java代理类过后,我们再来客户端代码中测试。

 package day_16_proxy;

 import java.lang.reflect.Proxy;

 /**
* @author 余林丰
*
* 2016年11月16日
*/
public class Client { public static void main(String[] args){
Test test = (Test)Proxy.newProxyInstance(Test.class.getClassLoader(), new Class<?>[]{Test.class}, new TestProxy());
test.say();
}
}

在第14行代码中,我们已经能够直接调用Test接口中的say方法了,原因就在于我们通过Proxy.newProxyInstance方法生成了一个代理类实例即TestProxy。

回到我们的MyBatis源码,在上一节中我们知道了一个Dao接口实际上是通过MapperProxyFactory生成了一个MapperProxy代理类。

 //org.apache.ibatis.binding.MapperProxyFactory
protected T newInstance(MapperProxy<T> mapperProxy) {
return (T) Proxy.newProxyInstance(mapperInterface.getClassLoader(), new Class[] { mapperInterface }, mapperProxy);
}

第3行代码是不是和在开头的例子中客户端测试代码如出一辙?生成代理类,这个代理类就是MapperProxy。清楚了MyBatis是如何构造出代理类的算是解决了第一个问题——一个接口怎么能直接调用其方法。

现在抛出第二个问题——接口中每个具体的方法是如何做到一一实现代理的呢?我们再来看看MapperProxy类。这次我们先看MapperProxy类实现的InvocationHanlder.invoke方法。

 //org.apache.ibatis.binding.MapperProxy
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
if (Object.class.equals(method.getDeclaringClass())) {
try {
return method.invoke(this, args);
} catch (Throwable t) {
throw ExceptionUtil.unwrapThrowable(t);
}
}
final MapperMethod mapperMethod = cachedMapperMethod(method);
return mapperMethod.execute(sqlSession, args);
}

第3行做一个判断,判断是否是一个类,如果是一个类,那么久直接传递方法和参数调用即可。但我们知道此时是一个接口(也可以自己实现接口,旧版本通常这样做)。如果不是一个类的话,就会创建一个MapperMethod方法。见名思意:好像就是这个类在执行我们所调用的每一个接口方法。最后返回的是MapperMethod.execute方法。暂时不予理会MapperProxy类中的cachedMapperMethod方法。

来看看MapperMethod类,这个MapperMethod类就不得了了啊,可以说它是统管所有和数据库打交道的方法(当然概括起来也只有insert、delete、update、select四个方法)。所以,不管你的dao层有多少方法,归结起来的sql语句都有且仅有只有insert、delete、update、select,可以预料在MapperMethod的execute方法中首先判断是何种sql语句。

//org.apache.ibatis.binding.MapperMethod
public Object execute(SqlSession sqlSession, Object[] args) {
switch (command.getType()) {
case INSERT: {……}
case UPDATE: {……}
case DELETE: {……}
case SELECT: {……}
case FLUSH: {……}
}
}

这是MepperMethod.execute方法的删减,我们可以看到确实在execute方法内部首先判断是何种sql语句。(注意:在阅读这部分源代码时,我们的主线是MyBatis是如何创建出一个代理类,以及实现其方法的,而暂时忽略其中的细节)

我们选择常见的"SELECT"sql语句来进行解读,而在"SELECT"语句中又会设计到较多的细节问题:

 //org.apache.ibatis.binding
case SELECT:
if (method.returnsVoid() && method.hasResultHandler()) {
executeWithResultHandler(sqlSession, args);
result = null;
} else if (method.returnsMany()) {
result = executeForMany(sqlSession, args);
} else if (method.returnsMap()) {
result = executeForMap(sqlSession, args);
} else if (method.returnsCursor()) {
result = executeForCursor(sqlSession, args);
} else {
Object param = method.convertArgsToSqlCommandParam(args);
result = sqlSession.selectOne(command.getName(), param);
}
break;

我们选取第7行中的executeForMany中的方法来解读试试看。

 //org.apache.ibatis.binding.MapperMethod
private <E> Object executeForMany(SqlSession sqlSession, Object[] args) {
List<E> result;
Object param = method.convertArgsToSqlCommandParam(args);
if (method.hasRowBounds()) {
RowBounds rowBounds = method.extractRowBounds(args);
result = sqlSession.<E>selectList(command.getName(), param, rowBounds);
} else {
result = sqlSession.<E>selectList(command.getName(), param);
}
// issue #510 Collections & arrays support
if (!method.getReturnType().isAssignableFrom(result.getClass())) {
if (method.getReturnType().isArray()) {
return convertToArray(result);
} else {
return convertToDeclaredCollection(sqlSession.getConfiguration(), result);
}
}
return result;
}

第7行和第9行代码就是我们真正执行sql语句的地方,原来兜兜转转它又回到了sqlSession的方法中。在下一节,我们再重新回到重要的SqlSession中。