MyBatis源码系列--6.Mybatis与 Spring 整合分析

参考文档:http://www.mybatis.org/spring/zh/index.html
这里我们以传统的 Spring 为例,因为配置更直观,在 Spring 中使用配置类注解是一样的。

基于之前源码分析的两篇文章,用编程式的方式已经弄清楚了 MyBatis 的工作流程、核
心模块和底层原理。编程式的方式,也就是 MyBatis 的原生 API 里面有三个核心对象:
SqlSessionFactory、SqlSession、MapperProxy。

大部分时候我们不会在项目中单独使用 MyBatis 的工程,而是集成到 Spring 里面使用,但是却没有看到这三个对象在代码里面的出现。我们直接注入了一个 Mapper 接口,调用它的方法。

比如之前是

        InputStream inputStream = Resources.getResourceAsStream("mybatis-config.xml");
        SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);
      
        SqlSession session = sqlSessionFactory.openSession(); 

        BlogMapper mapper = session.getMapper(BlogMapper.class);
         
        Blog blog = mapper.selectBlogById(1);

而我们实际使用的是这样的

    @Autowired
    BlogMapper blogMapper;

直接注入就可以使用了,所以有几个关键的问题,我们要弄清楚:
1、 SqlSessionFactory 是什么时候创建的?
2、 SqlSession 去哪里了?为什么不用它来 getMapper?
3、 为什么@Autowired 注入一个接口,在使用的时候却变成了代理对象?
在 IOC的容器里面我们注入的是什么? 注入的时候发生了什么事情?

关键配置

在 Spring 的 applicationContext.xml 里面配置 SqlSessionFactoryBean,它
是用来帮助我们创建会话的,其中还要指定全局配置文件和 mapper 映射器文件的路径


  
  
  

然后在 applicationContext.xml 配置需要扫描 Mapper 接口的路径
在 Mybatis 里面有3种方式,第一种是配置一个 MapperScannerConfigurer。

 
        
        
    

第二种是配置一个标签:

第三种就是直接用@MapperScan 注解,比如我们在 Spring Boot 的启动类上加上一个注解

@SpringBootApplication
@MapperScan("com.wei.dal.mapper")
public class MybaitsApp {
  public static void main(String[] args) {
    SpringApplication.run(MybaitsApp.class, args);
  }
}

创建会话工厂

Spring 对 MyBatis 的对象进行了管理,但是并不会替换 MyBatis 的核心对象。也就意味着:MyBatis jar 包中的 SqlSessionFactory、SqlSession、MapperProxy 这些都会用到。而 mybatis-spring.jar 里面的类只是做了一些包装或者桥梁的工作。

所以第一步,我们看一下在 Spring 里面,工厂类是怎么创建的。
我们在 Spring 的配置文件中配置了一个 SqlSessionFactoryBean,我们来看一下这个类

public class SqlSessionFactoryBean implements FactoryBean, InitializingBean, ApplicationListener {
    ...

     //它实现了 InitializingBean 接口,所以要实现 afterPropertiesSet()方法,
    //这个方法会在 bean 的属性值设置完的时候被调用
     public void afterPropertiesSet() throws Exception {
        ...
        this.sqlSessionFactory = this.buildSqlSessionFactory();
    }

    //它实现了 FactoryBean 接口,所以它初始化的时候,实际上是调用 getObject()方法,
   //它里面调用的也是 afterPropertiesSet()方法
    public SqlSessionFactory getObject() throws Exception {
        if (this.sqlSessionFactory == null) {
            this.afterPropertiesSet();
        }
        return this.sqlSessionFactory;
    }

}

从上面注释可知,最终还是会调用buildSqlSessionFactory方法来创建会话工厂,我们跟进看看

 protected SqlSessionFactory buildSqlSessionFactory() throws IOException {
        //这里的XMLConfigBuilder 就是mybatis jar包中的对象
        XMLConfigBuilder xmlConfigBuilder = null;
        //这里的Configuration对象就是mybatis jar包中的对象
        Configuration targetConfiguration;
        if (this.configuration != null) {
            targetConfiguration = this.configuration;
            if (targetConfiguration.getVariables() == null) {
                targetConfiguration.setVariables(this.configurationProperties);
            } else if (this.configurationProperties != null) {
                targetConfiguration.getVariables().putAll(this.configurationProperties);
            }
        } else if (this.configLocation != null) {
            xmlConfigBuilder = new XMLConfigBuilder(this.configLocation.getInputStream(), (String)null, this.configurationProperties);
            targetConfiguration = xmlConfigBuilder.getConfiguration();
        } else {
            LOGGER.debug(() -> {
                return "Property 'configuration' or 'configLocation' not specified, using default MyBatis Configuration";
            });
            targetConfiguration = new Configuration();
            Optional.ofNullable(this.configurationProperties).ifPresent(targetConfiguration::setVariables);
        }

       ...
 
        //最终还是调用mybatis jar包中的SqlSessionFactoryBuilder.build方法,返回DefaultSqlSessionFactory
        return this.sqlSessionFactoryBuilder.build(targetConfiguration);
    }

从以上代码可知,还是会调用我们之前在mybatis源码分析中的几个对象,比如用XMLConfigBuilder对象解析好配置文件,封装到Configuration对象中,最终调用SqlSessionFactoryBuilder.build方法,返回DefaultSqlSessionFactory

这里用到的 Spring 扩展点总结:


image.png

创建 SqlSession

我们现在已经有一个 DefaultSqlSessionFactory,按照编程式的开发过程,我们接下来就会创建一个 SqlSession 的实现类,但是在 Spring 里面,我们不是直接使用DefaultSqlSession 的,而是对它进行了一个封装,这个 SqlSession 的实现类就是SqlSessionTemplate。这个跟 Spring 封装其他的组件是一样的,比如 JdbcTemplate,RedisTemplate 等等,也是 Spring 跟 MyBatis 整合的最关键的一个类

为什么不用 DefaultSqlSession?它是线程不安全的,注意看类上的注解

*Note that this class is not Thread-Safe

而 SqlSessionTemplate 是线程安全的

* Thread safe, Spring managed, {@code SqlSession} that works with Spring

在编程式的开发中,SqlSession 我们会在每次请求的时候创建一个,但是 Spring 里面只有一个 SqlSessionTemplate(默认是单例的),多个线程同时调用的时候怎么保证线程安全?

SqlSessionTemplate 里面有 DefaultSqlSession 的所有的方法:selectOne()、
selectList()、insert()、update()、delete(),不过它都是通过一个代理对象实现的。
这个代理对象在构造方法里面通过一个代理类创建:

public class SqlSessionTemplate implements SqlSession, DisposableBean {
    private final SqlSessionFactory sqlSessionFactory;
    private final ExecutorType executorType;
    private final SqlSession sqlSessionProxy;
    private final PersistenceExceptionTranslator exceptionTranslator;

    public SqlSessionTemplate(SqlSessionFactory sqlSessionFactory, ExecutorType executorType, PersistenceExceptionTranslator exceptionTranslator) {
        ...
        this.sqlSessionProxy = (SqlSession)Proxy.newProxyInstance(SqlSessionFactory.class.getClassLoader(), new Class[]{SqlSession.class}, new SqlSessionTemplate.SqlSessionInterceptor());
    }

由此可知,所有的方法都会先走到内部代理类 SqlSessionInterceptor 的 invoke()方法

private class SqlSessionInterceptor implements InvocationHandler {
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) 
            throws Throwable {
          //根据Spring的事物上下文来获取事物范围内的sqlSession
        SqlSession sqlSession = getSqlSession(
                SqlSessionTemplate.this.sqlSessionFactory,
                SqlSessionTemplate.this.executorType,
                SqlSessionTemplate.this.exceptionTranslator);
        try {
            //调用从Spring的事物上下文获取事物范围内的sqlSession对象
            Object result = method.invoke(sqlSession, args);
            //然后判断一下当前的sqlSession是否被Spring托管 如果未被Spring托管则自动commit
            if (!isSqlSessionTransactional(sqlSession, 
                    SqlSessionTemplate.this.sqlSessionFactory)) {
                // force commit even on non-dirty sessions because some databases require
                // a commit/rollback before calling close()
                sqlSession.commit(true);
            }
            return result;
        } catch (Throwable t) {
            //如果出现异常则根据情况转换后抛出
            Throwable unwrapped = unwrapThrowable(t);
            if (SqlSessionTemplate.this.exceptionTranslator != null && 
                    unwrapped instanceof PersistenceException) {
                // release the connection to avoid a deadlock if the 
                // translator is no loaded. See issue #22
                closeSqlSession(sqlSession, SqlSessionTemplate.this.sqlSessionFactory);
                sqlSession = null;
                Throwable translated = SqlSessionTemplate.this.exceptionTranslator.
                        translateExceptionIfPossible((PersistenceException) unwrapped);
                if (translated != null) {
                    unwrapped = translated;
                }
            }
            throw unwrapped;
        } finally {
            if (sqlSession != null) {
                //关闭sqlSession,它会根据当前的sqlSession是否在Spring的事物上下文当中来执行具体的关闭动作
                //如果sqlSession被Spring管理 则调用holder.released();
                //使计数器-1,否则才真正的关闭sqlSession
                closeSqlSession(sqlSession, SqlSessionTemplate.this.sqlSessionFactory);
            }
        }
    }
}

SqlSession sqlSession = SqlSessionUtils.getSqlSession(SqlSessionTemplate.this.sqlSessionFactory, SqlSessionTemplate.this.executorType, SqlSessionTemplate.this.exceptionTranslator);
我们跟进这个代码可知,最终还是会生成一个 DefaultSqlSession

Mapper接口的扫描注册

在 Service 层可以使用@Autowired 自动注入的 Mapper 接口,需要保存在 BeanFactory(比如 XmlWebApplicationContext)中。也就是说接口肯定是在 Spring 启动的时候被扫描了,注册过的
1、 什么时候扫描的?
2、 注册的时候,注册的是什么?这个决定了我们拿到的是什么实际对象
我们在applicationContext.xml中配置了 MapperScannerConfigurer,MapperScannerConfigurer 实现了 BeanDefinitionRegistryPostProcessor 接口,BeanDefinitionRegistryPostProcessor 是 BeanFactoryPostProcessor 的子类,可以通过编码的方式修改、新增或者删除某些 Bean 的定义,我们只需要重写 postProcessBeanDefinitionRegistry()方法,在这里面操作 Bean就可以了

public class MapperScannerConfigurer implements BeanDefinitionRegistryPostProcessor, InitializingBean, ApplicationContextAware, BeanNameAware {
  ...
 public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) {
        if (this.processPropertyPlaceHolders) {
            this.processPropertyPlaceHolders();
        }

        ClassPathMapperScanner scanner = new ClassPathMapperScanner(registry);
        scanner.setAddToConfig(this.addToConfig);
        scanner.setAnnotationClass(this.annotationClass);
        scanner.setMarkerInterface(this.markerInterface);
        scanner.setSqlSessionFactory(this.sqlSessionFactory);
        scanner.setSqlSessionTemplate(this.sqlSessionTemplate);
        scanner.setSqlSessionFactoryBeanName(this.sqlSessionFactoryBeanName);
        scanner.setSqlSessionTemplateBeanName(this.sqlSessionTemplateBeanName);
        scanner.setResourceLoader(this.applicationContext);
        scanner.setBeanNameGenerator(this.nameGenerator);
        scanner.registerFilters();
        scanner.scan(StringUtils.tokenizeToStringArray(this.basePackage, ",; \t\n"));
    }
  ...
}

跟进scan 方法

public int scan(String... basePackages) {
        int beanCountAtScanStart = this.registry.getBeanDefinitionCount();
        //scanner.scan() 方 法 是 ClassPathBeanDefinitionScanner 中 的 , 而 它 的 子 类
ClassPathMapperScanner 覆 盖 了 doScan() 方 法
        this.doScan(basePackages);
        if (this.includeAnnotationConfig) {
            AnnotationConfigUtils.registerAnnotationConfigProcessors(this.registry);
        }

        return this.registry.getBeanDefinitionCount() - beanCountAtScanStart;
    }


/scanner.scan() 方 法 是 ClassPathBeanDefinitionScanner 中 的 , 而 它 的 子 类ClassPathMapperScanner 覆 盖 了 doScan() 方 法

public class ClassPathMapperScanner extends ClassPathBeanDefinitionScanner {
   public Set doScan(String... basePackages) {
        Set beanDefinitions = super.doScan(basePackages);
        if (beanDefinitions.isEmpty()) {
            LOGGER.warn(() -> {
                return "No MyBatis mapper was found in '" + Arrays.toString(basePackages) + "' package. Please check your configuration.";
            });
        } else {
            this.processBeanDefinitions(beanDefinitions);
        }

        return beanDefinitions;
    }
}

跟进processBeanDefinitions方法

private void processBeanDefinitions(Set beanDefinitions) {
        Iterator var3 = beanDefinitions.iterator();

        while(var3.hasNext()) {
            BeanDefinitionHolder holder = (BeanDefinitionHolder)var3.next();
            GenericBeanDefinition definition = (GenericBeanDefinition)holder.getBeanDefinition();
            String beanClassName = definition.getBeanClassName();
            LOGGER.debug(() -> {
                return "Creating MapperFactoryBean with name '" + holder.getBeanName() + "' and '" + beanClassName + "' mapperInterface";
            });
             //mapper接口是bean的原始类,但bean的实际类是MapperFactoryBean
            definition.getConstructorArgumentValues().addGenericArgumentValue(beanClassName);
            //最终把BeanClass 修改成 MapperFactoryBean
            definition.setBeanClass(this.mapperFactoryBean.getClass());
            definition.getPropertyValues().add("addToConfig", this.addToConfig);
            boolean explicitFactoryUsed = false;
            if (StringUtils.hasText(this.sqlSessionFactoryBeanName)) {
                definition.getPropertyValues().add("sqlSessionFactory", new RuntimeBeanReference(this.sqlSessionFactoryBeanName));
                explicitFactoryUsed = true;
            } else if (this.sqlSessionFactory != null) {
                definition.getPropertyValues().add("sqlSessionFactory", this.sqlSessionFactory);
                explicitFactoryUsed = true;
            }

            if (StringUtils.hasText(this.sqlSessionTemplateBeanName)) {
                if (explicitFactoryUsed) {
                    LOGGER.warn(() -> {
                        return "Cannot use both: sqlSessionTemplate and sqlSessionFactory together. sqlSessionFactory is ignored.";
                    });
                }

                definition.getPropertyValues().add("sqlSessionTemplate", new RuntimeBeanReference(this.sqlSessionTemplateBeanName));
                explicitFactoryUsed = true;
            } else if (this.sqlSessionTemplate != null) {
                if (explicitFactoryUsed) {
                    LOGGER.warn(() -> {
                        return "Cannot use both: sqlSessionTemplate and sqlSessionFactory together. sqlSessionFactory is ignored.";
                    });
                }

                definition.getPropertyValues().add("sqlSessionTemplate", this.sqlSessionTemplate);
                explicitFactoryUsed = true;
            }

            if (!explicitFactoryUsed) {
                LOGGER.debug(() -> {
                    return "Enabling autowire by type for MapperFactoryBean with name '" + holder.getBeanName() + "'.";
                });
                definition.setAutowireMode(2);
            }
        }

    }

我们主要关注这两行代码
//mapper接口是bean的原始类,但bean的实际类是MapperFactoryBean
definition.getConstructorArgumentValues().addGenericArgumentValue(beanClassName);
//最终把BeanClass 修改成 MapperFactoryBean
definition.setBeanClass(this.mapperFactoryBean.getClass());
我们打开MapperFactoryBean可知

public class MapperFactoryBean extends SqlSessionDaoSupport implements FactoryBean {

MapperFactoryBean 继 承 了 SqlSessionDaoSupport ,可以拿到SqlSessionTemplate

接口注入使用

我们使用 Mapper 的时候,只需要在加了 Service 注解的类里面使用@Autowired 注入 Mapper 接口就好了

@Service
public class  BlogService {
    @Autowired
    BlogMapper  blogMapper;

    public List< Blog> getAll() {
      return blogMapper.selectAll();
    }
}

我们从spring源码解析那篇文章知道,spring在启动的时候需要去实例化BlogService ,BlogService 依赖了BlogMapper 接口,Spring 会根据 Mapper 的名字从 BeanFactory 中获取它的 BeanDefination,再从BeanDefination 中 获 取 BeanClass , BlogMapper 对 应 的 BeanClass 是
MapperFactoryBean(上一步已经分析过)
接下来就是创建 MapperFactoryBean,它实现了 FactoryBean 接口,同样是调用 getObject()方法

public class MapperFactoryBean extends SqlSessionDaoSupport implements FactoryBean {
    public T getObject() throws Exception {
        return this.getSqlSession().getMapper(this.mapperInterface);
    }
}

因 为 MapperFactoryBean 继 承 了 SqlSessionDaoSupport , 所 以getSqlSession()就是调用父类的方法,返回 SqlSessionTemplate

public abstract class SqlSessionDaoSupport extends DaoSupport {
   public SqlSession getSqlSession() {
        return this.sqlSessionTemplate;
    }
}

SqlSessionTemplate 的 getMapper()方法,里面又有两个方法:

public class SqlSessionTemplate implements SqlSession, DisposableBean {
     public  T getMapper(Class type) {
        return this.getConfiguration().getMapper(type, this);
    }
}

getConfiguration方法通过 DefaultSqlSessionFactory,返回全部配置 Configuration

  public Configuration getConfiguration() {
        return this.sqlSessionFactory.getConfiguration();
    }

public class DefaultSqlSessionFactory implements SqlSessionFactory {
      public Configuration getConfiguration() {
        return this.configuration;
    }
}

回到getMapper方法,

 public  T getMapper(Class type, SqlSession sqlSession) {
        return this.mapperRegistry.getMapper(type, sqlSession);
    }

这一步我们很熟悉,跟编程式使用里面的getMapper一样,通过工厂类MapperProxyFactory 获得一个 MapperProxy 代理对象,也就是说,我们注入到 Service 层的接口,实际上还是一个 MapperProxy 代理对象。
所以最后调用 Mapper 接口的方法,也是执行 MapperProxy 的 invoke()方法,后面的流程就跟编程式的里面一模一样了

总结:


MyBatis源码系列--6.Mybatis与 Spring 整合分析_第1张图片
image.png

——学自咕泡学院

你可能感兴趣的:(MyBatis源码系列--6.Mybatis与 Spring 整合分析)