Mybatis的初始化和结合Spring Framework后初始化的源码探究
带着下面的问题进行学习:
(1)Mybatis 框架或 Spring Framework 框架对数据层 Mapper 接口做了代理,那是做了 JDK 动态代理还是 CGLIB 代理?
(2)Mapper 接口使用和不使用 @Mapper 注解有什么区别?
(3)Spring Framework 框架引入 Mybatis 的 jar 包后,Spring Framework 是怎么管理的?
(4)@MapperScan注解的作用是什么?
在探究上面的问题前,先了解什么是 FactoryBean,FactoryBean 和 BeanFactory有什么区别?
BeanFactory 是 Spring Framework 中的一个 Bean 工厂(AnnotationConfigApplicationContext、ClassPathXmlApplicationContext等),可以产生类,它有一个 getBean 方法可以获取到类;
FactoryBean 是一个 Bean,受 Spring Framework 管理的一个对象,定义 Bean 有好几种方式,比如 xml 的 <bean> 标签,注解 @Bean、@Service 等;
下面是案例:实现 FactoryBean 接口,重写方法
public class TempBean { public void query(){ System.out.println("TempBean"); } }
@Configuration("MyFactroyBean") public class MyFactroyBean implements FactoryBean { public void test() { System.out.println("MyFactroyBean MyFactroyBeanTest"); } @Override public Object getObject() throws Exception { return new TempBean(); } @Override public Class<?> getObjectType() { return TempBean.class; } @Override public boolean isSingleton() { return true; } }
public static void main(String[] args) { AnnotationConfigApplicationContext context= new AnnotationConfigApplicationContext(MyFactroyBean.class); MyFactroyBean myFactroyBean = (MyFactroyBean) context.getBean("&MyFactroyBean"); myFactroyBean.test(); TempBean tempBean = (TempBean) context.getBean("MyFactroyBean"); tempBean.query(); }
//=========结果======
MyFactroyBean MyFactroyBeanTest
TempBean
如果你的类实现了 FactoryBean,那么 Spring 存在两个对象:
一个是 getObject() 返回的对象:当前类指定的名字;
一个是当前类:”&”+当前类指定的名字;
当实现了 FactoryBean 后,Spring 在容器实例化过程中,会对项目中的自定义类进行实例化,当前类会被代理成一个 CGLIB 类(使用了@Configuration注解),当你获取当前类时,传递“&xx”,Spring 会进行判断,如果获取是当前类,返回代理类对象,如果是获取 getObject 返回的对象,会直接调用 getObject 方法中的 new xxx();
下面是Spring容器初始化时,在执行一些后置处理器(ConfigurationClassPostProcessor是对项目的 ComponentScan 注解的路径、@Configuration 注解的类进行扫描解析,超级重要的一个类)过程中,对实现了 FactoryBean 接口的子类进行代理:
下面是对实现了 FactoryBean 接口的子类的获取:先判断是不是 FactoryBean 类型,如果是加前缀“&”再进行获取;因为前面已经对子类进行了代理,并且存入了 beanDefinitionMap 中
但即使上面加了前缀”&“,在后面还是会剔除掉,那怎么判断是获取当前类还是获取getObject方法中的类呢?在 Spring 容器过程中,只会对当前类(实现了 FactoryBean 的子类)进行实例化,当获取时,getSingleton 直接从缓存中获取(每个对象实例化后会放入缓存中 singletonObjects)共享对象,然后 getObjectForBeanInstance 获取实例对象返回,在获取实例对象过程中,会判断是不是 FactoryBean 类型,如果是直接将代理对象返回;
下面判断是否是一个FactoryBean对象:
public static boolean isFactoryDereference(@Nullable String name) { return (name != null && name.startsWith(BeanFactory.FACTORY_BEAN_PREFIX));//FACTORY_BEAN_PREFIX=”&" }
如果是获取 getObject() 返回的对象,跟上面一样,但在判断是不是一个 FactoryBean 对象时,由于name没有“&”的前缀,所以走下面的流程,不会直接 return beanInstance;
FactroyBean是一个 Bean(可以当作一个业务类,比如连接数据库做一些业务操作),当一个类的依赖关系很复杂时,只需要提供一个简单的接口给外部使用时(将内部的一些关系进行封装维护好)可以使用 FactroyBean 来实现,比如 mybatis 的 SqlSessionFactoryBean 是一个 FactoryBean 实现类,里面的 getObject 方法有一个 afterPropertiesSet 方法,内部将很多依赖(Configuration 配置、TransactionFactory 事务工厂等)进行了处理封装维护到了 SqlSessionFactoryBean,外部只需要传入一个 DataSource 数据源即可,添加 MapperScan 扫描 xml 文件会自动把 xml 的信息维护到 SqlSessionFactoryBean中;
那么 SqlSessionFactoryBean 什么时候被调用呢?可以看下idea调试的方法调用栈:
从上面几张图片可以看出,当IOC容器Context进行类(Service)初始化时,发现有属性变量(Mapper),会对属性变量(Mapper)进行创建获取自动注入进去,其中会获取到Mapper封装成的一个MapperFactoryBean,执行到 SqlSessionFactoryBean的 afterPropertiesSet 方法对Mapper接口对应的xml进行解析获取;
MapperFactoryBean 也是一个 FactoryBean,每个 Mapper 接口都会转换成一个 MapperFactoryBean,所以 doCreateBean() 创建获取时可以通过 mapperInterface(Mapper接口)获取到对应的 MapperFactoryBean,它的值是存储到 Configuration的 MapperRegistry变量中;
那它是什么时候存储到 Configuration的 MapperRegistry变量中的呢?通过下面的图片可以看出,是在解析完xml后放入的:
当 Mapper 接口在 MapperFactoryBean#getObject() 时,它对Mapper接口进行了代理:MapperRegistry#getMapper()
public <T> T getMapper(Class<T> type, SqlSession sqlSession) { //获得代理对象工厂:里面包含Mapper接口信息 final MapperProxyFactory<T> mapperProxyFactory = (MapperProxyFactory<T>) knownMappers.get(type); if (mapperProxyFactory == null) { throw new BindingException("Type " + type + " is not known to the MapperRegistry."); } try { //代理对象实例化 return mapperProxyFactory.newInstance(sqlSession); } catch (Exception e) { throw new BindingException("Error getting mapper instance. Cause: " + e, e); } }
下面是 MapperProxyFactory#newInstance() 的源码:从中可以看到,Mapper 接口对应的代理对象是使用了 JDK 动态代理产生的;【解决了上文的第一个问题,其实从 Mapper 是接口也可以猜出是使用 JDK 动态实现的】
public T newInstance(SqlSession sqlSession) { final MapperProxy<T> mapperProxy = new MapperProxy<T>(sqlSession, mapperInterface, methodCache); return newInstance(mapperProxy); } protected T newInstance(MapperProxy<T> mapperProxy) { return (T) Proxy.newProxyInstance(mapperInterface.getClassLoader(), new Class[] { mapperInterface }, mapperProxy); }
其中 MapperProxy 是一个 InvocationHandler 实现类,从前文可以得知调用 Mapper.xxx方法 会进入 MapperProxy#invoke(),最后会执行 mapperMethod.execute(sqlSession, args) 进行SQL查询返回结果;
综上所述:每个Mapper接口转化成一个MapperFactoryBean,当调用Mapper接口方法执行JDBC操作时,Mapper 接口通过 MapperFactoryBean#getObject() 对Mapper接口进行了 JDK 动态代理。
对于第三个问题“Spring Framework 框架引入 Mybatis 的 jar 包后,Spring Framework 是怎么管理的?”
从前文得出每个 Mapper 接口都会转换成一个 MapperFactoryBean,而 MapperFactoryBean 继承了 SqlSessionDaoSupport(mybatis-spring-xx.jar),SqlSessionDaoSupport 继承了 DaoSupport(spring-tx.jar),DaoSupport 实现了 InitializingBean接口,所以Spring容器初始化会执行DaoSupport#afterPropertiesSet方法,会执行里面的checkDaoConfig方法,MapperFactoryBean 重写了checkDaoConfig方法,所以最后会执行MapperFactoryBean#checkDaoConfig方法:
public abstract class DaoSupport implements InitializingBean { protected final Log logger = LogFactory.getLog(this.getClass()); public DaoSupport() { } public final void afterPropertiesSet() throws IllegalArgumentException, BeanInitializationException { this.checkDaoConfig(); try { this.initDao(); } catch (Exception var2) { throw new BeanInitializationException("Initialization of DAO failed", var2); } } protected abstract void checkDaoConfig() throws IllegalArgumentException; protected void initDao() throws Exception { } }
public class MapperFactoryBean<T> extends SqlSessionDaoSupport implements FactoryBean<T> { private Class<T> mapperInterface; private boolean addToConfig = true; @Override protected void checkDaoConfig() { super.checkDaoConfig(); notNull(this.mapperInterface, "Property 'mapperInterface' is required"); Configuration configuration = getSqlSession().getConfiguration();//得到配置信息 if (this.addToConfig && !configuration.hasMapper(this.mapperInterface)) { try { configuration.addMapper(this.mapperInterface);//对接口进行xml解析 } catch (Exception e) { logger.error("Error while adding the mapper '" + this.mapperInterface + "' to configuration.", e); throw new IllegalArgumentException(e); } finally { ErrorContext.instance().reset(); } } } }
Configuration#addMapper public <T> void addMapper(Class<T> type) { mapperRegistry.addMapper(type); } MapperRegistry#addMapper public <T> void addMapper(Class<T> type) { if (type.isInterface()) { if (hasMapper(type)) { throw new BindingException("Type " + type + " is already known to the MapperRegistry."); } boolean loadCompleted = false; try { //为每个接口配置一个MapperProxyFactory,供后续Mapper接口进行JDK代理 knownMappers.put(type, new MapperProxyFactory<T>(type)); // It's important that the type is added before the parser is run // otherwise the binding may automatically be attempted by the // mapper parser. If the type is already known, it won't try. //解析类 MapperAnnotationBuilder parser = new MapperAnnotationBuilder(config, type); parser.parse();//XML解析 loadCompleted = true; } finally { if (!loadCompleted) { knownMappers.remove(type); } } } }
@MapperScan 注解是导入了一个 MapperScannerRegistrar(是实现了 ImportBeanDefinitionRegistrar 接口的,作用是动态往 BeanDefinitionMap 添加 BeanDefinition),的作用是对包路径的 Mapper 接口和对应的XML配置信息进行扫描解析,将所有 Mapper 接口的class信息扫描成对应的 BeanDefition, MapperFactoryBean 类型的 BeanDefition,同时为这个BeanDefition提供一个有参构造方法,参数是 class,在后面Mapper接口进行实例化的 JDK 代理时可以根据这个 class 返回对应的代理对象;
@Retention(RetentionPolicy.RUNTIME) @Target(ElementType.TYPE) @Documented @Import(MapperScannerRegistrar.class) public @interface MapperScan { }
public class MapperScannerRegistrar implements ImportBeanDefinitionRegistrar, ResourceLoaderAware { @Override public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) { AnnotationAttributes annoAttrs = AnnotationAttributes.fromMap(importingClassMetadata.getAnnotationAttributes(MapperScan.class.getName())); ClassPathMapperScanner scanner = new ClassPathMapperScanner(registry); // 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(); scanner.doScan(StringUtils.toStringArray(basePackages));//包扫描解析 } }
下面是扫描解析包路径的主要逻辑:
public class ClassPathMapperScanner extends ClassPathBeanDefinitionScanner { /** * Calls the parent search that will search and register all the candidates. * Then the registered objects are post processed to set them as * MapperFactoryBeans */ @Override public Set<BeanDefinitionHolder> doScan(String... basePackages) { //将每个Mapper解析成BeanDefinitionHolder存放到set集合中 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 { //对BeanDefinitionHolder集合进行处理 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 //给每个BeanDefition提供一个有参构造方法,在后面Mapper接口进行实例化的JDK代理时可以根据这个class返回对应的代理对象; definition.getConstructorArgumentValues().addGenericArgumentValue(definition.getBeanClassName()); // issue #59 //Mapper接口转换成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("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() + "'."); } definition.setAutowireMode(AbstractBeanDefinition.AUTOWIRE_BY_TYPE); } } } }
综上所述:在 Spring 容器初始化过程中,在对Mapper接口进行实例化的过程中,@MapperScan 注解主要是解析包路径将 Mapper 接口解析成 MapperFactoryBean 的 BeanDefition,这是 Mapper 接口在实例化之前做的事情,在实例化中和之后的过程中,主要利用 Spring的InitializingBean 接口的特性(每个实现了 InitializingBean 接口的子类有一个 afterPropertiesSet 方法,在实例化过程中会执行)来实现对 Mapper 接口信息的初始化,比如 sql 语句的初始化,将这些信息缓存起来放到一个 Map 中:
上面的是 Mybatis 和 Spring Framework 结合后的初始化流程,那么单独的 Mybatis 初始化流程是怎样的呢?通过下面的案例(前文有详细案例)进行探究:
String resource = "mybatis-config2.xml"; InputStream inputStream = Resources.getResourceAsStream(resource); SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream); SqlSession sqlSession = sqlSessionFactory.openSession(); PersonMapper mapper = sqlSession.getMapper(PersonMapper.class); mapper.list(); mapper.list();
通过 debug 出来的方法调用栈可以看出:通过 SqlSessionFactoryBuilder 对 Mapper 接口和 XML 配置进行解析的信息也是存储到 Configuration 的 mappedStatements 中;
sqlSession.getMapper() 直接从 DefaultSqlSession 中获取,还是跟上面一样在解析完存放到 Configuration 的 MapperRegistry 变量中,代理还是走 JDK 动态代理:
综上所述: Mybatis 和 Spring Framework 结合后对 Mapper 接口的初始化过程中会转换成 MapperFactoryBean 类型,然后利用 Spring的InitializingBean 接口的特性来实现对 Mapper 接口信息的初始化,再将这些信息缓存起来放到 Configuration 的 mappedStatements 中;而Mybatis直接解析完放到 Configuration 的 mappedStatements 中;
最后剩余的一个问题:Mapper 接口使用和不使用 @Mapper 注解有什么区别?
通过查看@Mapper注解的注释,发现这个注解只是一个标识功能,使用与不使用没什么区别,不会影响系统的功能;