Mybatis源码解析之Spring获取Mapper过程

MyBatis是一款优秀的持久层构架,它支持定制化SQL语句、存储过程以及高级映射,可以通过简单的XML或注解配置和映射原生信息,将POJOs映射成数据库中的记录,从而方便进行增删改查的操作(官网)

问题

Spring如何与MyBatis连接起来的?如何去获取Mybatis中的Mapper,以及获取Mapper时干了哪些骚操作?

为了弄清楚上述问题,菜菜首先贴出开发时主要用到的MyBatis配置,随后进行源码解读,本文采用的是注解式配置。
一般MyBatis的配置如下所示:

@Configuration
@EnableTransactionManagement(proxyTargetClass = true)
@PropertySource({"classpath:jdbc.properties"})
@MapperScan(basePackages = "me.ele.eliter.**.mapper")
public class MyBatisConfig {
  private static final Log LOG = LogFactory.getLog(MyBatisConfig.class);

  @Bean(destroyMethod = "close")
  public DataSource dataSource(
      @Value("${db.driver}") String driverClass,
      @Value("${db.url}") String jdbcUrl,
      @Value("${db.username}") String userName,
      @Value("${db.password}") String passWord) throws Exception {
    ComboPooledDataSource dataSource = new ComboPooledDataSource();
    dataSource.setDriverClass(driverClass);
    dataSource.setJdbcUrl(jdbcUrl);
    dataSource.setUser(userName);
    dataSource.setPassword(passWord);
    return dataSource;
  }

  @Bean
  public PlatformTransactionManager transactionManager(DataSource dataSource) {
    DataSourceTransactionManager dataSourceTransactionManager = new DataSourceTransactionManager(dataSource);
    dataSourceTransactionManager.afterPropertiesSet();
    return dataSourceTransactionManager;
  }

  @Bean
  public SqlSessionFactoryBean sqlSessionFactoryBean(DataSource dataSource, ApplicationContext applicationContext) throws Exception {
    SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
    sessionFactory.setDataSource(dataSource);
    sessionFactory.setConfigLocation(applicationContext.getResource("classpath:mybatis-config.xml"));
    sessionFactory.afterPropertiesSet();
    sessionFactory.setPlugins(new Interceptor[] { new PagerInterceptor()});
    return sessionFactory;
  }

  @Bean
  public TransactionTemplate transactionTemplate(PlatformTransactionManager transactionManager) throws Exception {
    TransactionTemplate transactionTemplate = new TransactionTemplate(transactionManager);
    transactionTemplate.afterPropertiesSet();
    return transactionTemplate;
  }
}

可以看到MyBatisConfig这个类上有几个重要的注解@Configuration与@MapperScan,
@Configuration是Spring中注解,主要说明该类是配置类,Spring在启动时会执行@Bean的方法,并进行Bean的定义,详细信息请自行搜索。
本文主要讲@MapperScan,首先看MapperScan的定义如下

MapperScan

从上可以看到MapperScan中@Import了MapperScannerRegistrar,MapperScannerRegistrar是MaBatis连接Spring的扫描注册的类,从以下MapperScannerRegistrar源码图中可以看到它实现了Spring中的ImportBeanDefinitionRegistrar并重写registerBeanDefinitions方法,registerBeanDefinitions这个方法会在Spring启动里的refresh()->invokeBeanFactoryPostProcessors里一层一层调用到,具体调用链如下,这样在Spring启动时就进入到了MyBatis中进行扫描了
registerBeanDefinitions调用链
MapperScannerRegistrar源码

接下来看MyBatis扫描干了些什么事?以及如何将这些BeanDefinition注册到IOC容器中。
以下是MapperScannerRegistrar实现registerBeanDefinitions的源码

 @Override
  public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {

 //1.获取了注解MapperScan中的信息
    AnnotationAttributes annoAttrs = AnnotationAttributes.fromMap(importingClassMetadata.getAnnotationAttributes(MapperScan.class.getName()));
  /*2.将Spring中的registry放入scanner中,ClassPathMapperScanner继承了Spring的ClassPathBeanDefinitionScanner,
 在new 的时候调用了ClassPathBeanDefinitionScanner的构造方法,将register传入了ClassPathBeanDefinitionScanner中
*/
    ClassPathMapperScanner scanner = new ClassPathMapperScanner(registry);

 //3.一些属性设置操作
    // this check is needed in Spring 3.1
    if (resourceLoader != null) {
      scanner.setResourceLoader(resourceLoader);
    }

    Class<? extends Annotation> annotationClass = annoAttrs.getClass("annotationClass");
    if (!Annotation.class.equals(annotationClass)) {
      scanner.setAnnotationClass(annotationClass);
    }

    Class<?> markerInterface = annoAttrs.getClass("markerInterface");
    if (!Class.class.equals(markerInterface)) {
      scanner.setMarkerInterface(markerInterface);
    }

    Class<? extends BeanNameGenerator> generatorClass = annoAttrs.getClass("nameGenerator");
    if (!BeanNameGenerator.class.equals(generatorClass)) {
      scanner.setBeanNameGenerator(BeanUtils.instantiateClass(generatorClass));
    }

    Class<? extends MapperFactoryBean> mapperFactoryBeanClass = annoAttrs.getClass("factoryBean");
    if (!MapperFactoryBean.class.equals(mapperFactoryBeanClass)) {
      scanner.setMapperFactoryBean(BeanUtils.instantiateClass(mapperFactoryBeanClass));
    }

    scanner.setSqlSessionTemplateBeanName(annoAttrs.getString("sqlSessionTemplateRef"));
    scanner.setSqlSessionFactoryBeanName(annoAttrs.getString("sqlSessionFactoryRef"));

    List<String> basePackages = new ArrayList<String>();
    for (String pkg : annoAttrs.getStringArray("value")) {
      if (StringUtils.hasText(pkg)) {
        basePackages.add(pkg);
      }
    }
    for (String pkg : annoAttrs.getStringArray("basePackages")) {
      if (StringUtils.hasText(pkg)) {
        basePackages.add(pkg);
      }
    }
    for (Class<?> clazz : annoAttrs.getClassArray("basePackageClasses")) {
      basePackages.add(ClassUtils.getPackageName(clazz));
    }
    scanner.registerFilters();
 //4.具体执行扫描
    scanner.doScan(StringUtils.toStringArray(basePackages));
  }

从上面源码中可以看到在Spring调用registerBeanDefinitions时传入了BeanDefinitionRegistry(registry),有了registry就可以将MyBatis中的Mapper的BeanDefinition信息注册到Spring的IOC容器中,下面具体看下registerBeanDefinitions源码,首先是获取了MapperScan注解中的属性信息,接着将Sping中的register传入scanner,这样可将scanner扫描到的BeanDefinition信息注册到IOC容器中,最后是具体的扫描操作,下面看下ClassPathMapperScanner中具体扫描操作的源码

@Override
  public Set<BeanDefinitionHolder> doScan(String... basePackages) {
/*首先调用了Spring中的ClassPathBeanDefinitionScanner扫描类进行扫描得到了具体的beanDefinitions信息,
这一步也将扫描到的beanDefinitions信息注册到了Spring的IOC容器中,这里就不进入Spring源码看了*/
    Set<BeanDefinitionHolder> beanDefinitions = super.doScan(basePackages);

    if (beanDefinitions.isEmpty()) {
      logger.warn("No MyBatis mapper was found in '" + Arrays.toString(basePackages) + "' package. Please check your configuration.");
    } else {
      //这里是MyBatis对这些BeanDefinition做了一系列操作为MyBatis的特性服务
      processBeanDefinitions(beanDefinitions);
    }

    return beanDefinitions;
  }

  private void processBeanDefinitions(Set<BeanDefinitionHolder> beanDefinitions) {
    GenericBeanDefinition definition;
    for (BeanDefinitionHolder holder : beanDefinitions) {
      definition = (GenericBeanDefinition) holder.getBeanDefinition();

      if (logger.isDebugEnabled()) {
        logger.debug("Creating MapperFactoryBean with name '" + holder.getBeanName() 
          + "' and '" + definition.getBeanClassName() + "' mapperInterface");
      }

      // the mapper interface is the original class of the bean
      // but, the actual class of the bean is MapperFactoryBean
    /*这里获取到的是Mapper的接口,设置这些Mapper接口的实现类默认为MapperFactoryBean,偷天换日,
    (非常重要)下一篇文章中说明为什么Mapper不用写接口实现类就能运行会再次讲到这*/
definition.getConstructorArgumentValues().addGenericArgumentValue(definition.getBeanClassName()); // issue #59
      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("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("Cannot use both: sqlSessionTemplate and sqlSessionFactory together. sqlSessionFactory is ignored.");
        }
        definition.getPropertyValues().add("sqlSessionTemplate", this.sqlSessionTemplate);
        explicitFactoryUsed = true;
      }

      if (!explicitFactoryUsed) {
        if (logger.isDebugEnabled()) {
          logger.debug("Enabling autowire by type for MapperFactoryBean with name '" + holder.getBeanName() + "'.");
        }
    //这里设置Bean中的属性自动根据Type注入
   //(非常重要,后面的一些属性的注入, 如sqlSessionFactory会自动注入全仗于此)
     definition.setAutowireMode(AbstractBeanDefinition.AUTOWIRE_BY_TYPE);
      }
    }
  }

ClassPathMapperScanner继承了Spring的ClassPathBeanDefinitionScanner,doScan方法
1.调用了ClassPathBeanDefinitionScanner里的doScan方法(扫描方式与Spring的扫描一致),以上MapperScannerRegistrar源码注释2说明了ClassPathBeanDefinitionScanner拥有了Spring的register,这样在得到MyBatis的Mapper后,就可将其中的BeanDefinition注册到Spring中。
2.通过调用processBeanDefinitions对BeanDefinition做了一些重要处理
2.1 definition.setBeanClass(this.mapperFactoryBean.getClass())设置Mapper接口的class对象或是类的全限定名为MapperFactoryBean(可以理解为实现是MapperFactoryBean),这里是MyBatis只用写接口不用写实现的第一步,具体做是通过JDK动态代理,具体细节见Mybatis源码解析之MapperProxy

2.2 definition.setAutowireMode(AbstractBeanDefinition.AUTOWIRE_BY_TYPE设置了这个bean中的属性是根据Type自动注入,加上2.1后这个也就是说会自动调用MapperFactoryBean中的set方法,sqlSessionFactory就是通过这种方式注入的,进而得到sqlSesion,这方面的详细信息见Mybatis源码解析之SqlSession来自何方

至此,Spring中已经获取到了MyBatis有哪些mapper,并且这些mapper接口的BeanDefinition信息经过了MaBatis的处理,比如设置接口的默认实现类为MapperFactoryBean等等,下一步就是解析MaBatis中配置的xml信息

如有错误,欢迎各位大佬斧正!
[下一篇]:Mybatis源码解析之配置解析

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 205,132评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,802评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,566评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,858评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,867评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,695评论 1 282
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,064评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,705评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,915评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,677评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,796评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,432评论 4 322
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,041评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,992评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,223评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,185评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,535评论 2 343

推荐阅读更多精彩内容