今日无事就研究研究mybatis这个mapper实现方式,一直都知道这就是通过java动态代理实现的,但却不清楚具体的流程,今个就一点点debug看看源码理一理。
先来段java动态代理回顾下
- 先定义个UserMapper接口
// ...
public interface UserMapper {
@Select({"select username from user where id = #{id}"})
@ResultType(String.class)
String findUsername(@Param("id") String id);
}
- 然后一个小巧的动态代理
public static void main(String[] args) {
ClassLoader classLoader = UserMapper.class.getClassLoader();
UserMapper instance = (UserMapper) Proxy.newProxyInstance(classLoader, new Class[]{UserMapper.class}, new InvocationHandler() {
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
return args[0];
}
});
System.out.println(instance.findUsername("hello, dynamic proxy"));
}
解释下Proxy.newProxyInstance(ClassLoader,Class[],InvocationHandler)
方法的三个参数哈
ClassLoader
:这是类加载器,用来加载这个类的,一般使用该接口的加载器就行
Class[]
:这是参数是指要动态代理的类所实现的所有接口,这里因为就一个UserMapper接口就直接构建个数组就行
InvocationHandler
:这是个重点。这里需要传入一个实现java.lang.reflect.InvocationHandler
接口的类,该接口里面只有一个invoke
方法,每次通过代理类执行方法时都必然调用该方法。也正因此我们才能利用动态代理实现各种增强。
上面那段代码如果执行的话,会看到打印出hello, dynamic proxy
。这里并没有usermapper接口的实现类,因为并不需要有对应方法的实际操作。以上可以算是mybatis实现将接口方法与xml或注解的sql语句绑定执行的核心之处了。
下面看看mybtias是怎么做的
- 首先mybatis会借助
org.apache.ibatis.binding.MapperRegistry
类将所有的接口注册到容器保存
核心代码
public void addMapper(Class type) {
if (type.isInterface()) { // 判断是不是接口
if (hasMapper(type)) {
throw new BindingException("Type " + type + " is already known to the MapperRegistry.");
}
boolean loadCompleted = false;
try {
knownMappers.put(type, new MapperProxyFactory<>(type)); // 在mapper容器中新加一个MapperProxyFactory对象
MapperAnnotationBuilder parser = new MapperAnnotationBuilder(config, type);
parser.parse();
loadCompleted = true;
} finally {
if (!loadCompleted) {
knownMappers.remove(type);
}
}
}
}
- 通过
org.apache.ibatis.binding.MapperProxyFactory
实例化接口的代理对象
protected T newInstance(MapperProxy mapperProxy) {
return (T) Proxy.newProxyInstance(mapperInterface.getClassLoader(), new Class[] { mapperInterface }, mapperProxy);
}
public T newInstance(SqlSession sqlSession) {
final MapperProxy mapperProxy = new MapperProxy<>(sqlSession, mapperInterface, methodCache); // 该类实现了InvocationHandler接口
return newInstance(mapperProxy);
}
- 在执行接口方法时,实际是调用
MapperProxy
中的invoke方法
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
try { // 判断下被执行的方法是不是Object对象或接口的默认方法(j8引入的)
if (Object.class.equals(method.getDeclaringClass())) {
return method.invoke(this, args);
} else if (isDefaultMethod(method)) {
return invokeDefaultMethod(proxy, method, args);
}
} catch (Throwable t) {
throw ExceptionUtil.unwrapThrowable(t);
}
final MapperMethod mapperMethod = cachedMapperMethod(method);
return mapperMethod.execute(sqlSession, args); // 执行指定的方法
}
- 最终被执行的方法长这样
public Object execute(SqlSession sqlSession, Object[] args) {
Object result;
switch (command.getType()) { // 根据不同的sql语句类型执行对应的方法
case INSERT: {
Object param = method.convertArgsToSqlCommandParam(args);
result = rowCountResult(sqlSession.insert(command.getName(), param));
break;
}
case UPDATE: {
Object param = method.convertArgsToSqlCommandParam(args);
result = rowCountResult(sqlSession.update(command.getName(), param));
break;
}
case DELETE: {
Object param = method.convertArgsToSqlCommandParam(args);
result = rowCountResult(sqlSession.delete(command.getName(), param));
break;
}
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); // 查询出来结果,这里会对语句进行解析,生成有问号?的sql,并将参数传递
if (method.returnsOptional() &&
(result == null || !method.getReturnType().equals(result.getClass()))) {
result = Optional.ofNullable(result);
}
}
break;
case FLUSH:
result = sqlSession.flushStatements();
break;
default:
throw new BindingException("Unknown execution method for: " + command.getName());
}
if (result == null && method.getReturnType().isPrimitive() && !method.returnsVoid()) {
throw new BindingException("Mapper method '" + command.getName()
+ " attempted to return null from a method with a primitive return type (" + method.getReturnType() + ").");
}
return result;
}
在这个过程中会通过执行器org.apache.ibatis.executor.SimpleExecutor
执行sql语句,语句返回的结果通过结果集处理器org.apache.ibatis.executor.resultset.DefaultResultSetHandler
进行处理,最终完成查询以及数据的绑定。