spring源码之模拟mybatis通过BeanDefinition注入Mapper

话不多说，直接上代码。

public class MyImporyBeanDefinitionRegistart implements ImportBeanDefinitionRegistrar {

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

        //这里是模拟获取了所有的Mapper的BeanDefinition
		Map<String, BeanDefinition> map = new HashMap<>();
		BeanDefinitionBuilder beanBuilder = BeanDefinitionBuilder.genericBeanDefinition(TMapper.class);
		map.put("tMapper", beanBuilder.getBeanDefinition());

		beanBuilder = BeanDefinitionBuilder.genericBeanDefinition(AMapper.class);
		map.put("aMapper", beanBuilder.getBeanDefinition());


		for (String key : map.keySet()){
		
		    //取出map中的BeanDefinition对象，强转为AbstractBeanDefinition是因为这个子类中有我们使用的属性
			AbstractBeanDefinition mapperDefinition = (AbstractBeanDefinition)map.get(key);
			
			//这个是获取当前的BeanDefinition对象的类属性，也就是AMapper.class或者TMapper.class
			Class beanClass = mapperDefinition.getBeanClass();

            //这一行代码是把这个类属性变更为MyFactoryBean.class，我们自己写的工厂类
			mapperDefinition.setBeanClass(MyFactoryBean.class);
            
            //这是在MyFactoryBean类中给mapperInterface这个属性注入值，而值就是aMapper或者tMapper
			mapperDefinition.getPropertyValues().add("mapperInterface", beanClass.getName());
			registry.registerBeanDefinition(key, mapperDefinition);
		}

//		System.out.println("--MyFactoryBean to beanDefinition--");
//		BeanDefinitionBuilder beanBuilder = BeanDefinitionBuilder.genericBeanDefinition(MyFactoryBean.class);
//		AbstractBeanDefinition beanDefinition = beanBuilder.getBeanDefinition();
//		beanDefinition.getPropertyValues().add("mapperInterface", "com.spring.batis.dao.AMapper");
//		registry.registerBeanDefinition("a", beanDefinition);
	}
}

如何使用这个类呢？使用Import就可以了

@ComponentScan("com.spring.batis")
@Import(MyImporyBeanDefinitionRegistart.class)
public class BatisConfig {
}

	@Test
	public void ImportBeanDefinitionBatis(){
		AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(BatisConfig.class);
		AMapper aMapper = (AMapper)context.getBean("aMapper");
		aMapper.queryMap("1");

		TMapper tMapper = (TMapper)context.getBean("tMapper");
		tMapper.queryMap("1");
	}

有些同学会说了，为什么myBatis直接使用一个Scan的扫描注解就可以实现了呢。我们可以思考一下，把之前写的MySqlSession类作为mybatis本身的jar包，新写的这个MyImporyBeanDefinitionRegistart类作为mybatis-spring的jar包。那么我们本身的代码里面是不是就不需要注入的代码了？

想要实现一个注解就开启的话也很简单

@Retention(RetentionPolicy.RUNTIME)
@Import(MyImporyBeanDefinitionRegistart.class)
public @interface MyScan {
}

@ComponentScan("com.spring.batis")
//@Import(MyImporyBeanDefinitionRegistart.class)
@MyScan
public class BatisConfig {
}

这样一来，MyScan注解就是开启我们自定义Mybaits的关键注解。

那么最后剩下的问题就是如何在MyImporyBeanDefinitionRegistart类中直接获取所有的mapper文件了。