硬核！从设计模式上解读Mybatis源码

妖狐艹你老母 2024-03-26 11:27 71阅读 0赞

Mybatis是一个比较主流的ORM框架，所以在日常工作中接触得很多。我比较喜欢看优秀框架的源码，因为能写出这种框架的作者肯定有其独特之处。如果能看懂源码的一些巧妙构思，一定是受益匪浅的。

所谓万事开头难，看源码也要找到切入的点。设计模式无疑是源码分析一个很好的切入点，废话不多说，那么我们就开始吧。

###  ###

工厂模式

工厂模式属于创建型模式。工厂模式的作用是把创建对象的逻辑封装起来，提供一个接口给外部创建对象，降低类与类之间的耦合。

在Mybatis中，用到工厂模式主要在DataSourceFactory。这是一个负责创建DataSource数据源的工厂。DataSourceFactory是一个接口，有不同的子类实现，根据不同的配置，生成不同的DataSourceFactory实现类。类图如下：

![2ccfe790243f070f7f79434538a38d47.jpeg][]

接着我们看一下DataSourceFactory的源码：

public interface DataSourceFactory { void setProperties(Properties props); DataSource getDataSource(); }

DataSourceFactory接口定义了两个抽象方法，怎么工作的呢，其实是跟dataSource标签的属性type有关。

Mybatis内置的type有三种配置，分别是UNPOOLED，POOLED，JNDI。

UNPOOLED，这个数据源的实现只是每次被请求时打开和关闭连接。

POOLED，这种数据源的使用“池“的思想，避免了创建新的连接实例时所必需的初始化和认证时间。

JNDI，这个数据源的实现是为了能在如 EJB 或应用服务器这类容器中使用，容器可以集中或在外部配置数据源，然后放置一个 JNDI 上下文的引用。

接着看XMLConfigBuilder的dataSourceElement()方法。

private DataSourceFactory dataSourceElement(XNode context) throws Exception \{ if (context != null) \{ //读取配置文件中dataSource标签的属性type String type = context.getStringAttribute("type"); Properties props = context.getChildrenAsProperties(); //根据type属性的值，返回不同的子类，使用接口DataSourceFactory接收，体现了面向接口编程的思想 DataSourceFactory factory = (DataSourceFactory) resolveClass(type).newInstance(); //设置属性值，比如数据库的url，username，password等等 factory.setProperties(props); //返回factory return factory; \} throw new BuilderException("Environment declaration requires a DataSourceFactory."); \}

获得DataSourceFactory之后，就通过getDataSource()方法获取数据源，完事了。

private void environmentsElement(XNode context) throws Exception { if (context != null) { if (environment == null) { environment = context.getStringAttribute("default"); } //这里for循环主要是配置文件可以配置多个数据源 for (XNode child : context.getChildren()) { String id = child.getStringAttribute("id"); if (isSpecifiedEnvironment(id)) { TransactionFactory txFactory = transactionManagerElement(child.evalNode("transactionManager")); DataSourceFactory dsFactory = dataSourceElement(child.evalNode("dataSource")); //dsFactory调动getDataSource()方法，创建dataSource对象 DataSource dataSource = dsFactory.getDataSource(); Environment.Builder environmentBuilder = new Environment.Builder(id) .transactionFactory(txFactory) .dataSource(dataSource); configuration.setEnvironment(environmentBuilder.build()); } } } }

Mybatis使用工厂模式来创建DataSourceFactory，可以做到通过配置去使用不同的DataSourceFactory创建DataSource，非常灵活。在Mybatis中使用到工厂模式还有很多地方，比如SqlSessionFactory，这里就不再展开了，有兴趣的可以自己探索一下。

###  ###

单例模式

单例模式属于创建型设计模式，这个类提供了一种访问其唯一的对象的方式，可以直接访问，不需要实例化该类的对象。保证在应用中只有单一对象被创建。

Mybatis中用到单例模式的地方有很多，这里举个例子是ErrorContext类。这是一个用于记录该线程的执行环境错误信息，所以是在线程范围内的单例。

public class ErrorContext { //使用ThreadLocal保存每个线程中的单例对象 private static final ThreadLocal<ErrorContext> LOCAL = new ThreadLocal<ErrorContext>(); //私有化构造器 private ErrorContext() { } //向外提供唯一的接口获取单例对象 public static ErrorContext instance() { //从LOCAL中取出context对象 ErrorContext context = LOCAL.get(); if (context == null) { //如果为null，new一个 context = new ErrorContext(); //放入到LOCAL中保存 LOCAL.set(context); } //如果不为null，直接返回 return context; } }

###  ###

建造者模式

建造者模式也是属于创建型模式，主要是在创建一个复杂对象时使用，通过一步一步构造最终的对象，将一个复杂对象的构建与它的表示分离。

在Mybatis中，使用到建造者模式的地方体现在ParameterMapping类，这是用于参数映射的一个类。

public class ParameterMapping { private Configuration configuration; private String property; private ParameterMode mode; private Class<?> javaType = Object.class; private JdbcType jdbcType; private Integer numericScale; private TypeHandler<?> typeHandler; private String resultMapId; private String jdbcTypeName; private String expression; //私有化构造器 private ParameterMapping() { } //通过内部类Builder创建对象 public static class Builder { //初始化ParameterMapping实例 private Par

[2ccfe790243f070f7f79434538a38d47.jpeg]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/26/f150423a4bed47998ca3d5bcb1ac94b8.png