Spring整合MyBatis之底層原理

作者 | 波哥

審校 | 孫淑娟

如果老鐵們對Spring架構足夠熟悉，整合MyBatis其實很容易了解，當然這裡假定老鐵們也已經熟悉了MyBatis架構。

在我們正常的應用開發過程中，使用MyBatis一般分為如下幾個步驟：

1.在配置類上增加MapperScan注解，例如：@MapperScan(basePackages = {"com.test.dao"},annotationClass = Mapper.class)；

2.在basePackages指定的目錄下建立待MyBatis讀取的接口檔案，例如：

@Mapper
public interface TestMapper {
......
}1.2.3.4.           

3.在Service或者其他地方使用該Mapper來操作資料庫。

使用起來是很簡單的，但是有沒有老鐵想過，為什麼做了這麼一個簡單的配置，這個Mapper就能操作資料庫了？按理說這個Mapper是個接口，應該是不能被建立才對啊！如果你有這個疑問，證明你是個愛思考的好童鞋。

咱們直接進入主題。Spring要與MyBatis整合，簡單來說隻要解決如下兩個問題：

一、Spring如何知道哪些類應該被管理？

要讓Spring去管理Bean的生命周期，首先需要對應的類被Spring掃描到，并且生成DeanDefinition，然後基于BeanDefinition生成Bean。下面對Spring生成BeanDefinition的方式做個小總結：

• 包含Component、Configuration、ComponentScan、Import、ImportResource注解的類；
• Import注解中指定的類、被Bean注解标注的方法所在的類；
• 實作了ImportBeanDefinitionRegistrar接口，并且在registerBeanDefinitions方法中調用registry直接注冊的類；
• 實作了ImportSelector接口，并且在selectImports方法中傳回的字元串對應的類；
• 直接調用register方法；
• 另外Spring還提供了一個擴充，可以讓開發者自己指定需要被管理的類對應的類型：通過往includeFilters中添加注解類類型。

我們分析源碼，第一步得找到它的入口，Spring整合MyBatis的入口，毫無疑問是MapperScan這個注解，在MapperScan注解上包含Import(MapperScannerRegistrar.class)注解，Spring整合MyBatis正是用了Import和ImportBeanDefinitionRegistrar的方式。我們先通過一張流程圖來了解下整體流程，然後再慢慢品。

我們來看MapperScannerRegistrar這個類的繼承關系圖：

MapperScannerRegistrar是ImportBeanDefinitionRegistrar的實作類，Spring會去調用這個類的registerBeanDefinitions方法添加beanDefinition，這個方法中具體做了些什麼呢：

擷取MapperScan注解的配置資訊，比如basePackages、annotationClass，basePackages表示需要掃描的路徑，annotationClass則是指定了增加了這種注解類的類需要被Spring進行管理，比如增加了Mapper注解的類需要被Spring管理。

生成MapperScannerConfigurer這個類型的beanDefinition，并且把MapperScan注解的配置資訊添加到該beanDefinition的屬性集合中。

後續Spring就會基于這個MapperScannerConfigurer做一系列文章，看下它的繼承關系：

它是BeanDefinitionRegistryPostProcessor的實作類，是一個BeanFactory後置處理器，Spring會調用該類的postProcessBeanDefinitionRegistry方法來添加beanDefinition的操作，MapperScannerConfigurer這個類中具體實作如下：

它定義了ClassPathMapperScanner這個掃描器，然後使用這個掃描器來掃描類，掃描哪些類呢？掃描有Mapper注解的類，看它的關系知道，它是ClassPathBeanDefinitionScanner的子類，而spring則是使用ClassPathBeanDefinitionScanner來進行掃描的。

為什麼ClassPathMapperScanner能夠掃描到帶有Mapper注解的類呢？看上面代碼，就是通過調用registerFilters方法來添加includeFilter（實際類型是：TypeFilter），這個就是Spring提供的擴充點，讓咱們自己來指定需要被掃描的類，這裡使用的是MappScan注解中annotationClass屬性配置的注解類型，我們這裡配置了Mapper，是以調用scan方法開啟掃描後，Spring就會将包含Mapper注解的類掃描為BeanDefinition。注意這裡的掃描能力還是調用Spring的掃描器來實作的，ClassPathMapperScanner并沒有修改，隻是當掃描完成後，ClassPathMapperScanner會對掃描出的BeanDefinition進行重新處理，主要是把原來的BeanClass修改成了MapperFactoryBean.class：

而這個MapperFactoryBean是FactoryBean的實作類，老鐵們，FactoryBean這種Bean有什麼特點？這個可是面試的高發點哦。

做個小小的總結：Spring掃描到有Mapper注解的類，生成BeanDefinition，并且将這一類BeanDefinition的BeanClass的值修改為MapperFactoryBean，也就是說它的類型不再是咱們自己編寫的Mapper接口了，而是一個FactoryBean，這樣Spring就能做妖了。

二、Mapper注解的類是接口

那如何執行個體化呢？

到這一步，其實老鐵們也大概清楚了，Spring在執行個體化Mapper執行個體時，實際上首先會執行個體化MapperFactoryBean，然後再調用它的getObject方法。我們知道在Java裡面接口是肯定不能被執行個體化的，那這個被執行個體化的對象隻能是一個代理對象，是以我們有理由猜想這個getObject方法應該是用來建立代理對象的。要建立代理對象，得從以下兩個方面着手：

1.準備工作

這裡Spring準備的是接口類型和建立代理對象的代理工廠。具體如何準備的呢？來看上述MapperFactoryBean類型的整體繼承關系：

它實作了InitializingBean，于是可以知道，在MapperFactoryBean初始化完成後，Spring會調用它的afterPropertiesSet方法，進而會執行到checkDaoConfig方法：

在該方法中調用configuration的addMapper方法，這個方法裡面到底做了啥？

看出門道了嗎？其實就是使用Mapper的接口類型作為key，MapperProxyFactory做為value，然後添加到mapperRegistry對象的Map集合中，注意這個type同時也是MapperProxyFactory對象的構造參數哦。

2.執行個體化

上述動作已經準備好了，接下來就應該是建立了。Spring在建立完成MapperFactoryBean對象後，最終會調用它的getObject方法來獲得真實的對象：

getObject方法中，會調用getMapper方法，該方法中從knowMappers這個Map集合中拿到MapperProxyFactory對象，這個對象不就是我們在準備階段添加的嘛！它就是用來建立代理對象的工廠。

從上面代碼中也不難看出，确實是為咱們自己的接口建立了代理對象，而代理類的處理類則是MapperProxy對象，也就是說對所有接口對象的調用，都會進入MapperProxy的Invoke方法，至此Spring成功對接MyBatis。

作者介紹

波哥，互聯行業從業10餘年，先後擔任項目總監及架構師。目前專攻技術，喜歡研究技術原理。技術全面，主攻java，精通JVM底層機制及Spring全家桶底層架構原理，熟練掌握目前主流的中間件、服務網格等技術原理。