Mybatis整合Spring思路
Mybatis整合Spring思路
需要解决的问题和解决思路
1、如何将接口注入到容器
这个在Mybatis其实已经做好了，是通过动态代理的方式，将接口由代理类注入；
但是要采取一种合适的方式注入，我们常用的方式有@component注解，或是通过@configuration和@bean注解，但是这里显然都不能满足需求。
采取的做法是通过FactoryBean和ImportBeanDefinitionRegistrar的方式
beanFactoryPostProcessor
bean工厂的后置处理器，只能修改beanDeinition，不能往里面新增
ImportBeanDefinitionRegister 可以实现新增beanDefinition
2、如果扫描到mapper注解
通过spring的工具类方式，代码里体现
核心代码
实现FactoryBean
/**
* 通过动态代理生成mapper的实体类
* 实现FactoryBean方法，用于配置到ImportBeanDefinitionRegistrar
*/
public class MapperProxy implements FactoryBean<Object> {
private Class<?> clazz;
/**
* 将需要代理的接口通过参数传递进来
*
* @param clazz
*/
public MapperProxy(Class<?> clazz) {
this.clazz = clazz;
}
@Override
public Object getObject() throws Exception {
//动态代理接口
return Proxy.newProxyInstance(MapperProxy.class.getClassLoader(), new Class[]{clazz}, new InvocationHandler() {
@Override
public Object invoke(Object proxy, Method method, Object[] args) {
return method.getAnnotation(Select.class).value();
}
});
}
@Override
public Class<?> getObjectType() {
return clazz;
}
}
实现ImportBeanDefinitionRegistrar
/**
* 实现ImportBeanDefinitionRegistrar的registerBeanDefinitions方法
* 可以通过构造beanDefinition来注入bean
* AnnotationMetadata importingClassMetadata这个变量可以获取主类上的所有注解
*/
@Component
public class BeanRegister implements ImportBeanDefinitionRegistrar {
@Override
public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
Map<String, Object> annotationAttributes = importingClassMetadata.getAnnotationAttributes(MapperScan.class.getName());
String basePackage = (String) annotationAttributes.get("value");
Map<String, Class> map = ScanUtils.getMap(basePackage, Mapper.class);
for (Class clazz : map.values()) {
BeanDefinitionBuilder builder = BeanDefinitionBuilder.genericBeanDefinition();
AbstractBeanDefinition beanDefinition = builder.getBeanDefinition();
beanDefinition.setBeanClassName(MapperProxy.class.getName());
beanDefinition.getConstructorArgumentValues().addGenericArgumentValue(clazz);
registry.registerBeanDefinition(clazz.getSimpleName(), beanDefinition);
}
}
}
主类
需要使用Import注解
/**
* 实现mybatis整合spring的大致思路
* 主要思路为
* 1、通过jdk动态代理生成接口mapper的代理类
* 2、将代理类注入到容器
* 2.1 采用BeanFactory的方式，工厂化生成代理类
* 2.2 采用ImportBeanDefinitionRegistrar，直接定义BeanDefinition来构造bean
* 3、指定包扫描注解（ScanUtils）
* 测试方法在com.lexiaoyao.springmybatis.SpringMybatisApplicationTests
*/
@SpringBootApplication
@Import(BeanRegister.class)//需要通过Import注解来导入BeanRegister
@MapperScan("com.lexiaoyao.springmybatis")
public class SpringMybatisApplication {
public static void main(String[] args) {
SpringApplication.run(SpringMybatisApplication.class, args);
}
}
扫包工具类
public class ScanUtils {
private static final String RESOURCE_PATTERN = "/**/*.class";
/**
* 可以获取指定路径下的带有某个注解的全部类
*
* @param path 指定路径
* @param annoClass 指定要寻找的注解
* @return
*/
public static Map<String, Class> getMap(String path, Class<? extends Annotation> annoClass) {
Map<String, Class> handlerMap = new HashMap<String, Class>();
//spring工具类，可以获取指定路径下的全部类
ResourcePatternResolver resourcePatternResolver = new PathMatchingResourcePatternResolver();
try {
String pattern = ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX +
ClassUtils.convertClassNameToResourcePath(path) + RESOURCE_PATTERN;
Resource[] resources = resourcePatternResolver.getResources(pattern);
//MetadataReader 的工厂类
MetadataReaderFactory readerfactory = new CachingMetadataReaderFactory(resourcePatternResolver);
for (Resource resource : resources) {
//用于读取类信息
MetadataReader reader = readerfactory.getMetadataReader(resource);
//扫描到的class
String classname = reader.getClassMetadata().getClassName();
Class<?> clazz = Class.forName(classname);
//判断是否有指定主解
Annotation annotation = clazz.getAnnotation(annoClass);
if (annotation != null) {
//将注解中的类型值作为key，对应的类作为 value
handlerMap.put(classname, clazz);
}
}
} catch (IOException | ClassNotFoundException e) {
}
return handlerMap;
}
}
