SpringBoot整合Mybatis的核心原理

时间:2023-02-10 22:56:01

0. 前言:

  • SpringBoot整合Mybatis只需添加mybatis-spring-boot-starter的依赖(本文版本2.2.0,对应mybatis版本3.5.7),然后yml进行配置即可
  • 本文对Mybatis一些底层原理进行探究,主要是一些自动配置以及Mapper代理对象的生成过程
spring:
  datasource:
    driver-class-name: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://localhost:3306/cloud?characterEncoding=utf8&autoReconnect=true&useUnicode=true&useSSL=false&serverTimezone=UTC
    username: root
    password: 123456
    
mybatis:
  # mybatis-config.xml 配置文件的路径 与 configuration 不可一起设置
  #config-location: classpath:mapper/mybatis-config.xml
  # sql映射文件的位置
  mapper-locations: classpath:mapper/*.xml
  # 开启驼峰命名转化
  configuration:
    map-underscore-to-camel-case: true
    log-impl: org.apache.ibatis.logging.stdout.StdOutImpl
  # 开启别名
  type-aliases-package: com.example.demo.easy.domain

1. 自动配置类MybatisAutoConfiguration:

1.1. SqlSessionFactory的生成:

  • 其中上述yml的mybatis配置项会被读取封装到properties里,通过SqlSessionFactoryBean来构建SqlSessionFactory
  @Bean
  @ConditionalOnMissingBean
  public SqlSessionFactory sqlSessionFactory(DataSource dataSource) throws Exception {
    SqlSessionFactoryBean factory = new SqlSessionFactoryBean();
    factory.setDataSource(dataSource);
    factory.setVfs(SpringBootVFS.class);
    if 	(StringUtils.hasText(this.properties.getConfigLocation())) {
      factory.setConfigLocation(this.resourceLoader.getResource(this.properties.getConfigLocation()));
    }
  
 	// ... ... 省略一些赋值,详细可看源码
    if (!ObjectUtils.isEmpty(this.properties.resolveMapperLocations())) {
      factory.setMapperLocations(this.properties.resolveMapperLocations());
   }   
    return factory.getObject();
  }

SpringBoot整合Mybatis的核心原理

org.mybatis.spring.SqlSessionFactoryBean#buildSqlSessionFactory

  • 过程中会将mybatis的配置被解析封装成Configuration对象(yml的配置项会被mybatis-config.xml的配置项覆盖)
  • mapper.xml会被解析封装成MappedStatement对象(用于存储要映射的SQL语句的id、参数等信息)
  • 最终会通过this.sqlSessionFactoryBuilder.build(targetConfiguration);去new一个DefaultSqlSessionFactory
 protected SqlSessionFactory buildSqlSessionFactory() throws Exception {

    final Configuration targetConfiguration;

    XMLConfigBuilder xmlConfigBuilder = null;
    if (this.configuration != null) {
 		... ...
    } else if (this.configLocation != null) {
      xmlConfigBuilder = new XMLConfigBuilder(this.configLocation.getInputStream(), null, this.configurationProperties);
      targetConfiguration = xmlConfigBuilder.getConfiguration();
    } else {
 		... ...
    }
	// ... ... yml的配置项赋值targetConfiguration
    if (hasLength(this.typeAliasesPackage)) {
      scanClasses(this.typeAliasesPackage, this.typeAliasesSuperType).stream()
          .filter(clazz -> !clazz.isAnonymousClass()).filter(clazz -> !clazz.isInterface())
          .filter(clazz -> !clazz.isMemberClass()).forEach(targetConfiguration.getTypeAliasRegistry()::registerAlias);
    }
    
	... ...

    if (xmlConfigBuilder != null) {
      try {
      	// 解析mybatis-config.xml配置项并会覆盖yml的配置
        xmlConfigBuilder.parse();
        LOGGER.debug(() -> "Parsed configuration file: '" + this.configLocation + "'");
      } catch (Exception ex) {
        throw new NestedIOException("Failed to parse config resource: " + this.configLocation, ex);
      } finally {
        ErrorContext.instance().reset();
      }
    }
	... ...
    if (this.mapperLocations != null) {
      	  ... ...
          try {
          	// mapper.xml会被解析封装成MappedStatement对象(用于存储要映射的SQL语句的id、参数等信息)
            XMLMapperBuilder xmlMapperBuilder = new XMLMapperBuilder(mapperLocation.getInputStream(),
                targetConfiguration, mapperLocation.toString(), targetConfiguration.getSqlFragments());
            xmlMapperBuilder.parse();
          } 
          ... ...
        }
      }
    } else {
      LOGGER.debug(() -> "Property 'mapperLocations' was not specified.");
    }
	// 创建DefaultSqlSessionFactory
    return this.sqlSessionFactoryBuilder.build(targetConfiguration);
  }

1.2. Mapper的扫描和代理生成:

  • 如果没有使用@MapperScan和手动配置过MapperFactoryBean、MapperScannerConfigurer,默认会扫描启动类所在包路径
  • MapperScannerRegistrarNotFoundConfiguration 这个bean还Import了AutoConfiguredMapperScannerRegistrar
  • AutoConfiguredMapperScannerRegistrar实现了ImportBeanDefinitionRegistrar扩展接口,容器启动时会执行registerBeanDefinitions方法
  • 这个注册器中会定义MapperScannerConfigurer的BeanDefinition,通过addPropertyValue来对内部属性赋值,然后进行注册
  @org.springframework.context.annotation.Configuration
  @Import(AutoConfiguredMapperScannerRegistrar.class)
  @ConditionalOnMissingBean({ MapperFactoryBean.class, MapperScannerConfigurer.class })
  public static class MapperScannerRegistrarNotFoundConfiguration implements InitializingBean {

    @Override
    public void afterPropertiesSet() {
      logger.debug(
          "Not found configuration for registering mapper bean using @MapperScan, MapperFactoryBean and MapperScannerConfigurer.");
    }

  }

SpringBoot整合Mybatis的核心原理

1.2.1. MapperScannerConfigurer
  • 实现了BeanDefinitionRegistryPostProcessor扩展接口,容器启动会执行postProcessBeanDefinitionRegistry方法
  • 引入类路径Mapper扫描器ClassPathMapperScanner,调用scan方法(最终调用doScan方法)进行Mapper接口扫描

SpringBoot整合Mybatis的核心原理

org.mybatis.spring.mapper.ClassPathMapperScanner#doScan

  • ClassPathMapperScanner继承ClassPathBeanDefinitionScanner,调用父类方法来扫描包路径下Mapper
  • 父类的扫描器是Spring定义的,有其自身的扫描规则,最终会将Mapper接口扫描封装到BeanDefinition中
  • 由于Mapper接口是没有实现类的,如果不做处理是无法生成Bean然后放入IOC容器使用的
  • 所以要对BeanDefinition的beanClass做修改,修改成一个MapperFactoryBean,见processBeanDefinitions方法处理

SpringBoot整合Mybatis的核心原理
SpringBoot整合Mybatis的核心原理

1.2.2. MapperFactoryBean
  • 上文修改之后相当于beanDefinitionMap中(mapper,持有mapperClass的MapperFactoryBean的BeanDefinition)
  • MapperFactoryBean实现了FactoryBean,在Bean生命周期管理时会调用getObject方法
  • getObject方法中通过getMapper获取Mapper的代理对象

SpringBoot整合Mybatis的核心原理

1.2.3. getMapper生成代理对象
  • 最终通过MapperProxyFactory来创建Mapper的代理对象MapperProxy,可以看出采用的jdk动态代理
  • 所以最终启动后IOC容器的Map储存(mapper,MapperProxy),通过DI进行注入MapperProxy使用
  • MapperProxy是实现InvocationHandler的,最终调用时会触发代理对象的invoke方法

SpringBoot整合Mybatis的核心原理
SpringBoot整合Mybatis的核心原理

  • MapperProxy的invoke方法就不介绍了
  • 最终会通过SqlSessionFactor 创建的SqlSession去调用Executor执行器(入参:MappedStatement类型的参数),进行数据库操作

2. 小结:

  • 自动配置时,会将mybatis的配置被解析封装成Configuration对象
  •  mapper.xml也会被解析封装成MappedStatement对象(用于存储要映射的SQL语句的id、参数等信息)
  •  然后通过this.sqlSessionFactoryBuilder.build(Configuration)去创建SqlSessionFactor
  • 自动配置的过程中会通过ClassPathMapperScanner扫描器找到Mapper接口,封装成各自的BeanDefinition
  • 然后循环遍历对Mapper的BeanDefinition修改beanClass为MapperFactoryBean
  • MapperFactoryBean实现了FactoryBean,在Bean生命周期管理时会调用getObject方法,通过jdk动态代理生成代理对象MapperProxy
  • Mapper接口请求的时候,执行MapperProxy代理类的invoke方法,执行的过程中通过SqlSessionFactor 创建的SqlSession去调用Executor执行器(入参:MappedStatement类型的参数),进行数据库操作