自定义mybatis中的 mapperScanner接口

mybatis_train/src/main/java/com/zhouyuTest/Test.java

@@ -10,7 +10,9 @@ import org.springframework.context.annotation.AnnotationConfigApplicationContext
 
 public class Test {
     public static void main(String[] args) {
-        AnnotationConfigApplicationContext an = new AnnotationConfigApplicationContext(MyConfig.class);
+        AnnotationConfigApplicationContext an = new AnnotationConfigApplicationContext();
+        an.register(MyConfig.class);
+        an.refresh();
         UserService bean = an.getBean(UserService.class);
         bean.test();
     }

mybatis_train/src/main/java/com/zhouyuTest/config/MyConfig.java

@@ -1,6 +1,7 @@
 package com.zhouyuTest.config;
 
 import com.zhouyuTest.spring_mybatis.MyImportBeanDefinitionRegistrar;
+import com.zhouyuTest.spring_mybatis.MyMapperScan;
 import org.apache.ibatis.io.Resources;
 import org.apache.ibatis.session.SqlSessionFactory;
 import org.apache.ibatis.session.SqlSessionFactoryBuilder;
@@ -14,7 +15,7 @@ import java.io.InputStream;
 
 @Configuration
 @ComponentScan({"com.zhouyuTest"})
-@Import(MyImportBeanDefinitionRegistrar.class)
+@MyMapperScan("com.zhouyuTest.mapper")
 public class MyConfig {
     @Bean
     public SqlSessionFactory getFactory() throws IOException {

mybatis_train/src/main/java/com/zhouyuTest/mapper/ManagerMapper.java

@@ -0,0 +1,8 @@
+package com.zhouyuTest.mapper;
+
+import org.apache.ibatis.annotations.Select;
+
+public interface ManagerMapper {
+    @Select("select 'managerOrder'")
+    public String getManager();
+}

mybatis_train/src/main/java/com/zhouyuTest/spring_mybatis/MapperScanner.java

@@ -0,0 +1,30 @@
+package com.zhouyuTest.spring_mybatis;
+
+import org.springframework.beans.factory.annotation.AnnotatedBeanDefinition;
+import org.springframework.beans.factory.config.BeanDefinition;
+import org.springframework.beans.factory.config.BeanDefinitionHolder;
+import org.springframework.beans.factory.support.BeanDefinitionRegistry;
+import org.springframework.context.annotation.ClassPathBeanDefinitionScanner;
+
+import java.util.Set;
+
+public class MapperScanner extends ClassPathBeanDefinitionScanner {
+    public MapperScanner(BeanDefinitionRegistry registry) {
+        super(registry);
+    }
+
+    @Override
+    protected boolean isCandidateComponent(AnnotatedBeanDefinition beanDefinition) {
+        return beanDefinition.getMetadata().isInterface();
+    }
+    @Override
+    protected Set<BeanDefinitionHolder> doScan(String... basePackages) {
+        Set<BeanDefinitionHolder> beanDefinitionHolders = super.doScan(basePackages);
+        for (BeanDefinitionHolder beanDefinitionHolder : beanDefinitionHolders) {
+            BeanDefinition beanDefinition = beanDefinitionHolder.getBeanDefinition();
+            beanDefinition.getConstructorArgumentValues().addGenericArgumentValue(beanDefinition.getBeanClassName());
+            beanDefinition.setBeanClassName(MybatisFactoryBean.class.getName());
+        }
+        return beanDefinitionHolders;
+    }
+}

mybatis_train/src/main/java/com/zhouyuTest/spring_mybatis/MyImportBeanDefinitionRegistrar.java

@@ -8,17 +8,33 @@ import org.springframework.beans.factory.support.BeanDefinitionRegistry;
 import org.springframework.beans.factory.support.BeanNameGenerator;
 import org.springframework.context.annotation.ImportBeanDefinitionRegistrar;
 import org.springframework.core.type.AnnotationMetadata;
+import org.springframework.core.type.classreading.MetadataReader;
+import org.springframework.core.type.classreading.MetadataReaderFactory;
+import org.springframework.core.type.filter.TypeFilter;
+
+import java.io.IOException;
+import java.util.Map;
 
 public class MyImportBeanDefinitionRegistrar implements ImportBeanDefinitionRegistrar {
     @Override
     public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry, BeanNameGenerator importBeanNameGenerator) {
-        AbstractBeanDefinition beanDefinition = BeanDefinitionBuilder.genericBeanDefinition().getBeanDefinition();
-        beanDefinition.setBeanClass(MybatisFactoryBean.class);
-        beanDefinition.getConstructorArgumentValues().addGenericArgumentValue(UserMapper.class);
-        registry.registerBeanDefinition("userMapper",beanDefinition);
-        AbstractBeanDefinition beanDefinition2 = BeanDefinitionBuilder.genericBeanDefinition().getBeanDefinition();
-        beanDefinition2.setBeanClass(MybatisFactoryBean.class);
-        beanDefinition2.getConstructorArgumentValues().addGenericArgumentValue(OrderMapper.class);
-        registry.registerBeanDefinition("orderMapper",beanDefinition2);
+        Map<String, Object> annotationAttributes = importingClassMetadata.getAnnotationAttributes(MyMapperScan.class.getName());
+        String path = ((String) annotationAttributes.get("value"));
+        MapperScanner mapperScanner = new MapperScanner(registry);
+        mapperScanner.addIncludeFilter(new TypeFilter() {
+            @Override
+            public boolean match(MetadataReader metadataReader, MetadataReaderFactory metadataReaderFactory) throws IOException {
+                return true;
+            }
+        });
+        mapperScanner.scan(path);
+//        AbstractBeanDefinition beanDefinition = BeanDefinitionBuilder.genericBeanDefinition().getBeanDefinition();
+//        beanDefinition.setBeanClass(MybatisFactoryBean.class);
+//        beanDefinition.getConstructorArgumentValues().addGenericArgumentValue(UserMapper.class);
+//        registry.registerBeanDefinition("userMapper",beanDefinition);
+//        AbstractBeanDefinition beanDefinition2 = BeanDefinitionBuilder.genericBeanDefinition().getBeanDefinition();
+//        beanDefinition2.setBeanClass(MybatisFactoryBean.class);
+//        beanDefinition2.getConstructorArgumentValues().addGenericArgumentValue(OrderMapper.class);
+//        registry.registerBeanDefinition("orderMapper",beanDefinition2);
     }
 }

mybatis_train/src/main/java/com/zhouyuTest/spring_mybatis/MyMapperScan.java

@@ -0,0 +1,15 @@
+package com.zhouyuTest.spring_mybatis;
+
+import org.springframework.context.annotation.Import;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Target(ElementType.TYPE)
+@Retention(value = RetentionPolicy.RUNTIME)
+@Import(MyImportBeanDefinitionRegistrar.class)
+public @interface MyMapperScan {
+    String value() default "";
+}