Mybatis技术原理理——整体流程理解
前言:2018年,是最杂乱的一年!所以你看我的博客,是不是很空!
网上有很多关于Mybatis原理介绍的博文,这里介绍两篇我个人很推荐的博文 Mybatis3.4.x技术内幕 和 MyBaits源码分析!让我找到了学习的入口,当然另外你必须要看的官方文档 MyBatis学习。那么有了这些知识,就让我们愉快的吃鸡之路吧!
一:你首先得知道的知识点。
1.1 JDBC
在个人看来, Mybatis的核心就是对SQL语句的管理!那么在JAVA环境下,对SQL的管理或者其他任何的实现,肯定是离不开JAVA的数据库操作接口,包括Connrction接口、Statement接口、PreparadStatement接口以及ResultSet接口等等的属性,你可以先通过JDBC来操作一次或者更多次的数据库。这个就不多做赘述了!
1.2 动态代理
不知道你最初有没有和我一样,有这样的疑问。Mybatis的mapper层明明就是一个接口,都没有实现类!即使是在TMapper.xml中做了映射,也没有看到任何关于接口的实现类,那他是怎么被实例化的呢,又是怎么实现方法的功能的呢?动态代理会告诉你答案!
先来看看jdk动态代理的一个小例子:
建立一个实体
public class TestDemo { private Integer id; private String name; public Integer getId() { return id; } public void setId(Integer id) { this.id = id; } public String getName() { return name; } public void setName(String name) { this.name = name; } }
建立一个mapper
public interface TestDemoMapper { TestDemo getById(Integer id); }
建立一个Mapper的代理类,需要继承一个 InvocationHandler 即可
public class DemoMapperProxy<T> implements InvocationHandler { @SuppressWarnings("unchecked") public T newInstance(Class<T> clz) { return (T)Proxy.newProxyInstance(clz.getClassLoader(), new Class[] { clz }, this); } @Override public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { TestDemo testDemo = new TestDemo(); testDemo.setId(1); testDemo.setName("proxyName"); return testDemo; } }
测试类
public static void main(String[] args) { DemoMapperProxy<TestDemoMapper> demoMapperProxy = new DemoMapperProxy<TestDemoMapper>(); TestDemoMapper testDemoMapper = demoMapperProxy.newInstance(TestDemoMapper.class); TestDemo byId = testDemoMapper.getById(1); System.out.println(byId); }
结果:TestDemo{id=1, name='proxyName'},
所以这就应该大致能够说明Mybatis中的mapper是怎么工作的了吧!
二:关于mapper的整体流程理解
2.1 先来看看官方网站的一个例子吧
在官方文档中的 Mybatis 3 的入门介绍中,介绍了怎么配置一个mybatis的测试类
mybatis-config.xml 中的一个简单配置:
<configuration>
<environments default="development">
<environment id="development">
<transactionManager type="JDBC" />
<dataSource type="POOLED">
<property name="driver" value="com.mysql.jdbc.Driver" />
<property name="url" value="jdbc:mysql://172.20.139.237:3306/rubber-fruit?useUnicode=true&characterEncoding=utf-8&autoReconnect=true" />
<property name="username" value="user123" />
<property name="password" value="u123" />
</dataSource>
</environment>
</environments>
<mappers>
<mapper resource="UserMapper.xml" />
</mappers>
</configuration>
然后在写一个测试类:
public static void main(String[] args) throws Exception { String resouse = "mybatis-config.xml"; InputStream stream = Resources.getResourceAsStream(resouse); SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(stream); SqlSession sqlSession = sqlSessionFactory.openSession(); UserMapper mapper = sqlSession.getMapper(UserMapper.class); User byId = mapper.getById(1); sqlSession.close(); System.out.println(byId); }
其实这个逻辑很简单,首先是拿到mabatis-config.xml 中的配置对象,生成SqlSessionFactory,然后通过sqlSession拿到mapper,通过动态代理完成方法的执行,并返回结果!那么我们就可以简单的拆分成两个部分,第一:配置文件的初始化,第二:sql的执行,后面的文章也会更具这个大模块来更具细化的讲解
其实上面的第四行代码 可以更换成config对象的,可能看的更清楚一些:
String resouse = "mybatis-config.xml"; InputStream stream = Resources.getResourceAsStream(resouse); XMLConfigBuilder configBuilder = new XMLConfigBuilder(stream,null,null); Configuration parse = configBuilder.parse(); SqlSessionFactory build = new SqlSessionFactoryBuilder().build(parse);
其实Mybatis的前期初始化的一个重要的对象就是 Configuration对象,后面会慢慢揭开它的面纱!
2.2 整体流程图
通过上面的这个测试方法,我们通过源码流程,大致可以拆分出这样的一个图:
其实从这个图中我们已经可以看出Mybatis划分先两个部分:
1:初始化过程
首先是通过XMLConfigBuilder解析 mybatis-config.xml 对象来进行初始化,拿到Configuration对象,然后SqlSessionFactoryBuilder通过初始化的Configuration对象,生成SqlSession,SqlSession中调用getMapper对象方法,其实也就是Configuration中的getMapper方法 通过MapperProxy代理对象生成一个mapper。到这里位置就是一个简单的初初始化思路了!当然Mybatis的初始化远远不止这些。
2:mybatia中执行一个sql的完整流程
从图中也是能够很较清晰的看出这个执行流程的。因为Mapper是一个接口,所以不能直接实例化的!那么MapperProxy的作用,就是通过JDK的动态代理,来间接的对mapper进行实例化,不了解的可以看上面的1.2中的例子。那么我们可以看看org.apache.ibatis.binding.MapperProxy源码中的对象方法:
@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); }
通过代理的对象方法方法,拿到了一个MapperMethod对象,并对mapperMethod对象进行来缓存。为啥要缓存呢?这个时候可以看看MapperMethod对象是什么?org.apache.ibatis.binding.MapperMethod中对两个私有方法
private final SqlCommand command; private final MethodSignature method; public MapperMethod(Class<?> mapperInterface, Method method, Configuration config) { this.command = new SqlCommand(config, mapperInterface, method); this.method = new MethodSignature(config, mapperInterface, method); }
MapperMethod中的SqlCommand对象中的两个私有方法,
public static class SqlCommand { private final String name; private final SqlCommandType type; }
public enum SqlCommandType { UNKNOWN, INSERT, UPDATE, DELETE, SELECT, FLUSH; }
而MapperMethod中的MethodSignature对象呢?
private final boolean returnsMany; private final boolean returnsMap; private final boolean returnsVoid; private final boolean returnsCursor; private final Class<?> returnType; private final String mapKey; private final Integer resultHandlerIndex; private final Integer rowBoundsIndex; private final ParamNameResolver paramNameResolver;
所以这些很清晰了,MapperMethod对象是把mapper中的sql进行了封装,获取了sql的执行类型和返回值。
MapperMethod中的另外一个重要的方法:execute() 则担任这路由的重要任务:
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; }
通过执行的sql类型 选取不通的sqlSession进行执行。其实可以看出,转了一圈,最后的最后还是落在了SqlSession当中。而SqlSession又把这个重要的操作交个了执行器Executor。最后又到了StatementHandler来负责执行最后的sql,ResultSetHandler放回执行的结果。
2.3 记一个知识点
看到这里,突然是想问一下一个问题的,Mapper的方法是否支持重载呢?
答案是不能的!mybatis是使用package+Mapper+method 全名称作为Key值 去xml中寻找一个唯一sql来执行的那么,重载方法时将导致矛盾。对于Mapper接口,Mybatis禁止方法重载。通过代码断点可以看到。
org.apache.ibatis.session.Configuration对象中有一个方法,addMappedStatement()
protected final Map<String, MappedStatement> mappedStatements = new StrictMap<MappedStatement>("Mapped Statements collection");
/**中间部分省略掉**
public void addMappedStatement(MappedStatement ms) {
mappedStatements.put(ms.getId(), ms);
}
而这个地方put的ms.getId 就是通过mapper的配置文件组合成的一个唯一的key值。那我们在看看StrictMap 中的put方法源码,那么就一目了然了
@SuppressWarnings("unchecked") public V put(String key, V value) { if (containsKey(key)) { throw new IllegalArgumentException(name + " already contains value for " + key); } if (key.contains(".")) { final String shortKey = getShortName(key); if (super.get(shortKey) == null) { super.put(shortKey, value); } else { super.put(shortKey, (V) new Ambiguity(shortKey)); } } return super.put(key, value); }
如果有相同的key的,那么会抛出异常,这也就是为啥mapper不能重载的原因!