Mybatis mapper动态代理的原理详解
在开始动态代理的原理讲解以前，我们先看一下集成 mybatis 以后 dao 层不使用动态代理以及使用动态代理的两种实现方式，通过对比我们自己实现 dao 层接口以及 mybatis 动态代理可以更加直观的展现出 mybatis 动态代理替我们所做的工作，有利于我们理解动态代理的过程，讲解完以后我们再进行动态代理的原理解析，此讲解基于 mybatis 的环境已经搭建完成，并且已经实现了基本的用户类编写以及用户类的 Dao 接口的声明，下面是 Dao 层的接口代码
1 public interface UserDao { 2 /* 3 查询所有用户信息 4 */ 5 List<User> findAll(); 6 7 /** 8 * 保存用户 9 * @param user 10 */ 11 void save(User user); 12 13 /** 14 * 更新用户 15 * @return 16 */ 17 void update(User user); 18 /** 19 * 删除用户 20 */ 21 void delete(Integer userId); 22 23 /** 24 * 查找一个用户 25 * @param userId 26 * @return 27 */ 28 User findOne(Integer userId); 29 30 /** 31 * 根据名字模糊查询 32 * @param name 33 * @return 34 */ 35 List<User> findByName(String name); 36 /** 37 * 根据组合对象进行模糊查询 38 * @param vo 39 * @return 40 */ 41 List<User> findByQueryVo(QueryVo vo); 42 }
一、Mybatis dao 层两种实现方式的对比
1.dao 层不使用动态代理
dao 层不使用动态代理的话，就需要我们自己实现 dao 层的接口，为了简便起见，我只是实现了 Dao 接口中的 findAll 方法，以此方法为例子来展现我们自己实现 Dao 的方式的情况，让我们来看代码：
1 public class UserDaoImpl implements UserDao{ 2 private SqlSessionFactory factory; 3 public UserDaoImpl(SqlSessionFactory factory){ 4 this.factory = factory; 5 } 6 public List<User> findAll() { 7 //1. 获取 sqlSession 对象 8 SqlSession sqlSession = factory.openSession(); 9 //2. 调用 selectList 方法 10 List<User> list = sqlSession.selectList("com.example.dao.UserDao.findAll"); 11 //3. 关闭流 12 sqlSession.close(); 13 return list; 14 } 15 16 public void save(User user) { 17 18 } 19 20 public void update(User user) { 21 22 } 23 24 public void delete(Integer userId) { 25 26 } 27 28 public User findOne(Integer userId) { 29 return null; 30 } 31 32 public List<User> findByName(String name) { 33 return null; 34 } 35 36 public List<User> findByQueryVo(QueryVo vo) { 37 return null; 38 }
这里的关键代码 List<User> list = sqlSession.selectList("com.example.dao.UserDao.findAll")，需要我们自己手动调用 SqlSession 里面的方法，基于动态代理的方式最后的目标也是成功的调用到这里。
注意：如果是添加，更新或者删除操作的话需要在方法中增加事务的提交。
2.dao 层使用 Mybatis 的动态代理
使用动态代理的话 Dao 层的接口声明完成以后只需要在使用的时候通过 SqlSession 对象的 getMapper 方法获取对应 Dao 接口的代理对象，关键代码如下：
//3. 获取 SqlSession 对象
SqlSession session = factory.openSession();
//4. 获取 dao 的代理对象
UserDao mapper = session.getMapper(UserDao.class);
//5. 执行查询所有的方法
List<User> list = mapper.findAll();
获取到 dao 层的代理对象以后通过代理对象调用查询方法就可以实现查询所有用户列表的功能。
二、Mybatis 动态代理实现方式的原理解析
动态代理中最重要的类：SqlSession、MapperProxy、MapperMethod，下面开始从入口方法到调用结束的过程分析。
- 调用方法的开始:
//4. 获取 dao 的代理对象
UserDao mapper = session.getMapper(UserDao.class); 因为 SqlSesseion 为接口，所以我们通过 Debug 方式发现这里使用的实现类为 DefaultSqlSession。 - 找到 DeaultSqlSession 中的 getMapper 方法，发现这里没有做其他的动作，只是将工作继续抛到了 Configuration 类中，Configuration 为类不是接口，可以直接进入该类的 getMapper 方法中
@Override public <T> T getMapper(Class<T> type) { return configuration.<T>getMapper(type, this);}
3. 找到 Configuration 类的 getMapper 方法，这里也是将工作继续交到 MapperRegistry 的 getMapper 的方法中，所以我们继续向下进行。
public <T> T getMapper(Class<T> type, SqlSession sqlSession) { return mapperRegistry.getMapper(type, sqlSession); }
4. 找到 MapperRegistry 的 getMapper 的方法, 看到这里发现和以前不一样了，通过 MapperProxyFactory 的命名方式我们知道这里将通过这个工厂生成我们所关注的 MapperProxy 的代理类，然后我们通过 mapperProxyFactory.newInstance(sqlSession); 进入 MapperProxyFactory 的 newInstance 方法中
public <T> T getMapper(Class<T> type, SqlSession sqlSession) { final MapperProxyFactory<T> mapperProxyFactory = (MapperProxyFactory<T>)knownMappers.get(type); if (mapperProxyFactory == null) { throw new BindingException("Type" + type + "is not known to the MapperRegistry.");} try { return mapperProxyFactory.newInstance(sqlSession); } catch (Exception e) { throw new BindingException("Error getting mapper instance. Cause:" + e, e);} }
5. 找到 MapperProxyFactory 的 newIntance 方法，通过参数类型 SqlSession 可以得知，上面的调用先进入第二个 newInstance 方法中并创建我们所需要重点关注的 MapperProxy 对象，第二个方法中再调用第一个 newInstance 方法并将 MapperProxy 对象传入进去，根据该对象创建代理类并返回。这里已经得到需要的代理类了，但是我们的代理类所做的工作还得继续向下看 MapperProxy 类。
protected T newInstance(MapperProxy<T> mapperProxy) { return (T) Proxy.newProxyInstance(mapperInterface.getClassLoader(), new Class[] { mapperInterface}, mapperProxy);}public T newInstance(SqlSession sqlSession) {
final MapperProxy<T> mapperProxy = new MapperProxy<T>(sqlSession, mapperInterface, methodCache);
return newInstance(mapperProxy);
}
6. 找到 MapperProxy 类，发现其确实实现了 JDK 动态代理必须实现的接口 InvocationHandler，所以我们重点关注 invoke()方法，这里看到在 invoke 方法里先获取 MapperMethod 类，然后调用 mapperMethod.execute()，所以我们继续查看 MapperMethod 类的 execute 方法。
public class MapperProxy<T> implements InvocationHandler, Serializable {private static final long serialVersionUID = -6424540398559729838L;
private final SqlSession sqlSession;
private final Class<T> mapperInterface;
private final Map<Method, MapperMethod> methodCache;public MapperProxy(SqlSession sqlSession, Class<T> mapperInterface, Map<Method, MapperMethod> methodCache) {
this.sqlSession = sqlSession;
this.mapperInterface = mapperInterface;
this.methodCache = methodCache;
}@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
try {
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);
}private MapperMethod cachedMapperMethod(Method method) {
MapperMethod mapperMethod = methodCache.get(method);
if (mapperMethod == null) {
mapperMethod = new MapperMethod(mapperInterface, method, sqlSession.getConfiguration());
methodCache.put(method, mapperMethod);
}
return mapperMethod;
}@UsesJava7
private Object invokeDefaultMethod(Object proxy, Method method, Object[] args)
throws Throwable {
final Constructor<MethodHandles.Lookup> constructor = MethodHandles.Lookup.class
.getDeclaredConstructor(Class.class, int.class);
if (!constructor.isAccessible()) {
constructor.setAccessible(true);
}
final Class<?> declaringClass = method.getDeclaringClass();
return constructor
.newInstance(declaringClass,
MethodHandles.Lookup.PRIVATE | MethodHandles.Lookup.PROTECTED
| MethodHandles.Lookup.PACKAGE | MethodHandles.Lookup.PUBLIC)
.unreflectSpecial(method, declaringClass).bindTo(proxy).invokeWithArguments(args);
}/**
- Backport of java.lang.reflect.Method#isDefault()
*/
private boolean isDefaultMethod(Method method) {
return ((method.getModifiers()
& (Modifier.ABSTRACT | Modifier.PUBLIC | Modifier.STATIC)) == Modifier.PUBLIC)
&& method.getDeclaringClass().isInterface();
}
}
7. 找到类 MapperMethod 类的 execute 方法，发现 execute 中通过调用本类中的其他方法获取并封装返回结果，我们来看一下 MapperMethod 整个类。
public Object execute(SqlSession sqlSession, Object[] args) { Object result; switch (command.getType()) { 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);} 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; }
8. MapperMethod 类是整个代理机制的核心类，对 SqlSession 中的操作进行了封装使用。 该类里有两个内部类 SqlCommand 和 MethodSignature。 SqlCommand 用来封装 CRUD 操作，也就是我们在 xml 中配置的操作的节点。每个节点都会生成一个 MappedStatement 类。MethodSignature 用来封装方法的参数以及返回类型，在 execute 的方法中我们发现在这里又回到了 SqlSession 中的接口调用，和我们自己实现 UerDao 接口的方式中直接用 SqlSession 对象调用 DefaultSqlSession 的实现类的方法是一样的，经过一大圈的代理又回到了原地，这就是整个动态代理的实现过程了。
1 public class MapperMethod { 2 3 private final SqlCommand command; 4 private final MethodSignature method; 5 6 public MapperMethod(Class<?> mapperInterface, Method method, Configuration config) { 7 this.command = new SqlCommand(config, mapperInterface, method); 8 this.method = new MethodSignature(config, mapperInterface, method); 9 } 10 11 public Object execute(SqlSession sqlSession, Object[] args) { 12 Object result; 13 switch (command.getType()) { 14 case INSERT: { 15 Object param = method.convertArgsToSqlCommandParam(args); 16 result = rowCountResult(sqlSession.insert(command.getName(), param)); 17 break; 18 } 19 case UPDATE: { 20 Object param = method.convertArgsToSqlCommandParam(args); 21 result = rowCountResult(sqlSession.update(command.getName(), param)); 22 break; 23 } 24 case DELETE: { 25 Object param = method.convertArgsToSqlCommandParam(args); 26 result = rowCountResult(sqlSession.delete(command.getName(), param)); 27 break; 28 } 29 case SELECT: 30 if (method.returnsVoid() && method.hasResultHandler()) { 31 executeWithResultHandler(sqlSession, args); 32 result = null; 33 } else if (method.returnsMany()) { 34 result = executeForMany(sqlSession, args); 35 } else if (method.returnsMap()) { 36 result = executeForMap(sqlSession, args); 37 } else if (method.returnsCursor()) { 38 result = executeForCursor(sqlSession, args); 39 } else { 40 Object param = method.convertArgsToSqlCommandParam(args); 41 result = sqlSession.selectOne(command.getName(), param); 42 } 43 break; 44 case FLUSH: 45 result = sqlSession.flushStatements(); 46 break; 47 default: 48 throw new BindingException("Unknown execution method for:" + command.getName()); 49 } 50 if (result == null && method.getReturnType().isPrimitive() && !method.returnsVoid()) { 51 throw new BindingException("Mapper method'" + command.getName() 52 + "attempted to return null from a method with a primitive return type (" + method.getReturnType() + ")."); 53 } 54 return result; 55 } 56 57 private Object rowCountResult(int rowCount) { 58 final Object result; 59 if (method.returnsVoid()) { 60 result = null; 61 } else if (Integer.class.equals(method.getReturnType()) || Integer.TYPE.equals(method.getReturnType())) { 62 result = rowCount; 63 } else if (Long.class.equals(method.getReturnType()) || Long.TYPE.equals(method.getReturnType())) { 64 result = (long)rowCount; 65 } else if (Boolean.class.equals(method.getReturnType()) || Boolean.TYPE.equals(method.getReturnType())) { 66 result = rowCount > 0; 67 } else { 68 throw new BindingException("Mapper method'"+ command.getName() +"'has an unsupported return type:" + method.getReturnType()); 69 } 70 return result; 71 } 72 73 private void executeWithResultHandler(SqlSession sqlSession, Object[] args) { 74 MappedStatement ms = sqlSession.getConfiguration().getMappedStatement(command.getName()); 75 if (void.class.equals(ms.getResultMaps().get(0).getType())) { 76 throw new BindingException("method" + command.getName() 77 + "needs either a @ResultMap annotation, a @ResultType annotation," 78 + "or a resultType attribute in XML so a ResultHandler can be used as a parameter."); 79 } 80 Object param = method.convertArgsToSqlCommandParam(args); 81 if (method.hasRowBounds()) { 82 RowBounds rowBounds = method.extractRowBounds(args); 83 sqlSession.select(command.getName(), param, rowBounds, method.extractResultHandler(args)); 84 } else { 85 sqlSession.select(command.getName(), param, method.extractResultHandler(args)); 86 } 87 } 88 89 private <E> Object executeForMany(SqlSession sqlSession, Object[] args) { 90 List<E> result; 91 Object param = method.convertArgsToSqlCommandParam(args); 92 if (method.hasRowBounds()) { 93 RowBounds rowBounds = method.extractRowBounds(args); 94 result = sqlSession.<E>selectList(command.getName(), param, rowBounds); 95 } else { 96 result = sqlSession.<E>selectList(command.getName(), param); 97 } 98 // issue #510 Collections & arrays support 99 if (!method.getReturnType().isAssignableFrom(result.getClass())) { 100 if (method.getReturnType().isArray()) { 101 return convertToArray(result); 102 } else { 103 return convertToDeclaredCollection(sqlSession.getConfiguration(), result); 104 } 105 } 106 return result; 107 } 108 109 private <T> Cursor<T> executeForCursor(SqlSession sqlSession, Object[] args) { 110 Cursor<T> result; 111 Object param = method.convertArgsToSqlCommandParam(args); 112 if (method.hasRowBounds()) { 113 RowBounds rowBounds = method.extractRowBounds(args); 114 result = sqlSession.<T>selectCursor(command.getName(), param, rowBounds); 115 } else { 116 result = sqlSession.<T>selectCursor(command.getName(), param); 117 } 118 return result; 119 } 120 121 private <E> Object convertToDeclaredCollection(Configuration config, List<E> list) { 122 Object collection = config.getObjectFactory().create(method.getReturnType()); 123 MetaObject metaObject = config.newMetaObject(collection); 124 metaObject.addAll(list); 125 return collection; 126 } 127 128 @SuppressWarnings("unchecked") 129 private <E> Object convertToArray(List<E> list) { 130 Class<?> arrayComponentType = method.getReturnType().getComponentType(); 131 Object array = Array.newInstance(arrayComponentType, list.size()); 132 if (arrayComponentType.isPrimitive()) { 133 for (int i = 0; i < list.size(); i++) { 134 Array.set(array, i, list.get(i)); 135 } 136 return array; 137 } else { 138 return list.toArray((E[])array); 139 } 140 } 141 142 private <K, V> Map<K, V> executeForMap(SqlSession sqlSession, Object[] args) { 143 Map<K, V> result; 144 Object param = method.convertArgsToSqlCommandParam(args); 145 if (method.hasRowBounds()) { 146 RowBounds rowBounds = method.extractRowBounds(args); 147 result = sqlSession.<K, V>selectMap(command.getName(), param, method.getMapKey(), rowBounds); 148 } else { 149 result = sqlSession.<K, V>selectMap(command.getName(), param, method.getMapKey()); 150 } 151 return result; 152 } 153 154 public static class ParamMap<V> extends HashMap<String, V> { 155 156 private static final long serialVersionUID = -2212268410512043556L; 157 158 @Override 159 public V get(Object key) { 160 if (!super.containsKey(key)) { 161 throw new BindingException("Parameter'"+ key +"'not found. Available parameters are" + keySet()); 162 } 163 return super.get(key); 164 } 165 166 } 167 168 public static class SqlCommand { 169 170 private final String name; 171 private final SqlCommandType type; 172 173 public SqlCommand(Configuration configuration, Class<?> mapperInterface, Method method) { 174 final String methodName = method.getName(); 175 final Class<?> declaringClass = method.getDeclaringClass(); 176 MappedStatement ms = resolveMappedStatement(mapperInterface, methodName, declaringClass, 177 configuration); 178 if (ms == null) { 179 if (method.getAnnotation(Flush.class) != null) { 180 name = null; 181 type = SqlCommandType.FLUSH; 182 } else { 183 throw new BindingException("Invalid bound statement (not found):" 184 + mapperInterface.getName() + "." + methodName); 185 } 186 } else { 187 name = ms.getId(); 188 type = ms.getSqlCommandType(); 189 if (type == SqlCommandType.UNKNOWN) { 190 throw new BindingException("Unknown execution method for:" + name); 191 } 192 } 193 } 194 195 public String getName() { 196 return name; 197 } 198 199 public SqlCommandType getType() { 200 return type; 201 } 202 203 private MappedStatement resolveMappedStatement(Class<?> mapperInterface, String methodName, 204 Class<?> declaringClass, Configuration configuration) { 205 String statementId = mapperInterface.getName() + "." + methodName; 206 if (configuration.hasStatement(statementId)) { 207 return configuration.getMappedStatement(statementId); 208 } else if (mapperInterface.equals(declaringClass)) { 209 return null; 210 } 211 for (Class<?> superInterface : mapperInterface.getInterfaces()) { 212 if (declaringClass.isAssignableFrom(superInterface)) { 213 MappedStatement ms = resolveMappedStatement(superInterface, methodName, 214 declaringClass, configuration); 215 if (ms != null) { 216 return ms; 217 } 218 } 219 } 220 return null; 221 } 222 } 223 224 public static class MethodSignature { 225 226 private final boolean returnsMany; 227 private final boolean returnsMap; 228 private final boolean returnsVoid; 229 private final boolean returnsCursor; 230 private final Class<?> returnType; 231 private final String mapKey; 232 private final Integer resultHandlerIndex; 233 private final Integer rowBoundsIndex; 234 private final ParamNameResolver paramNameResolver; 235 236 public MethodSignature(Configuration configuration, Class<?> mapperInterface, Method method) { 237 Type resolvedReturnType = TypeParameterResolver.resolveReturnType(method, mapperInterface); 238 if (resolvedReturnType instanceof Class<?>) { 239 this.returnType = (Class<?>) resolvedReturnType; 240 } else if (resolvedReturnType instanceof ParameterizedType) { 241 this.returnType = (Class<?>)((ParameterizedType) resolvedReturnType).getRawType(); 242 } else { 243 this.returnType = method.getReturnType(); 244 } 245 this.returnsVoid = void.class.equals(this.returnType); 246 this.returnsMany = (configuration.getObjectFactory().isCollection(this.returnType) || this.returnType.isArray()); 247 this.returnsCursor = Cursor.class.equals(this.returnType); 248 this.mapKey = getMapKey(method); 249 this.returnsMap = (this.mapKey != null); 250 this.rowBoundsIndex = getUniqueParamIndex(method, RowBounds.class); 251 this.resultHandlerIndex = getUniqueParamIndex(method, ResultHandler.class); 252 this.paramNameResolver = new ParamNameResolver(configuration, method); 253 } 254 255 public Object convertArgsToSqlCommandParam(Object[] args) { 256 return paramNameResolver.getNamedParams(args); 257 } 258 259 public boolean hasRowBounds() { 260 return rowBoundsIndex != null; 261 } 262 263 public RowBounds extractRowBounds(Object[] args) { 264 return hasRowBounds()? (RowBounds) args[rowBoundsIndex] : null; 265 } 266 267 public boolean hasResultHandler() { 268 return resultHandlerIndex != null; 269 } 270 271 public ResultHandler extractResultHandler(Object[] args) { 272 return hasResultHandler()? (ResultHandler) args[resultHandlerIndex] : null; 273 } 274 275 public String getMapKey() { 276 return mapKey; 277 } 278 279 public Class<?> getReturnType() { 280 return returnType; 281 } 282 283 public boolean returnsMany() { 284 return returnsMany; 285 } 286 287 public boolean returnsMap() { 288 return returnsMap; 289 } 290 291 public boolean returnsVoid() { 292 return returnsVoid; 293 } 294 295 public boolean returnsCursor() { 296 return returnsCursor; 297 } 298 299 private Integer getUniqueParamIndex(Method method, Class<?> paramType) { 300 Integer index = null; 301 final Class<?>[] argTypes = method.getParameterTypes(); 302 for (int i = 0; i < argTypes.length; i++) { 303 if (paramType.isAssignableFrom(argTypes[i])) { 304 if (index == null) { 305 index = i; 306 } else { 307 throw new BindingException(method.getName() + "cannot have multiple" + paramType.getSimpleName() + "parameters"); 308 } 309 } 310 } 311 return index; 312 } 313 314 private String getMapKey(Method method) { 315 String mapKey = null; 316 if (Map.class.isAssignableFrom(method.getReturnType())) { 317 final MapKey mapKeyAnnotation = method.getAnnotation(MapKey.class); 318 if (mapKeyAnnotation != null) { 319 mapKey = mapKeyAnnotation.value(); 320 } 321 } 322 return mapKey; 323 } 324 }