Custom Annotation Scan만들기

1. Scan을 할 Annotation Interface를 생성
2. ImportBeanDefinitionRegistrard 구현체 생성
<- 해당 구현체는 Spring @Configration이 선언된 클래스에 대해서 실행된다.
3. 테스트를 진행할 Annotation Interface를 생성
4.TestContext객체생성

테스트 클래스들은 Mybatis mapperScan을 기반으로 생성하였음.

TestScan 클래스

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
@Documented
@Import(TestSannerRegistarar.class)
public @interface TestScan {



  String[] value() default {};

  String[] basePackages() default {};

  Class<?>[] basePackageClasses() default {};

  Class<? extends BeanNameGenerator> nameGenerator() default BeanNameGenerator.class;

  Class<? extends Annotation> annotationClass() default Annotation.class;

  Class<?> markerInterface() default Class.class;

  String sqlSessionTemplateRef() default "";

  String sqlSessionFactoryRef() default "";

}

TestScannerRegistarar 클래스

public class TestSannerRegistarar implements ImportBeanDefinitionRegistrar, ResourceLoaderAware {

  private ResourceLoader resourceLoader;

  /**

   * {@inheritDoc}

   */

  public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {

    AnnotationAttributes annoAttrs = AnnotationAttributes.fromMap(importingClassMetadata.getAnnotationAttributes(TestScan.class.getName()));

    ClassPathMapperScanner scanner = new ClassPathMapperScanner(registry);

    if (resourceLoader != null) { // this check is needed in Spring 3.1

      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));

    }

    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));

  }

  /**

   * {@inheritDoc}

   */

  public void setResourceLoader(ResourceLoader resourceLoader) {

    this.resourceLoader = resourceLoader;

  }

}

TestMapper 클래스

@Target(ElementType.TYPE)

@Retention(RetentionPolicy.RUNTIME)

@Documented

public @interface TestMapper {

}

TestContext 클래스 

@Configuration

@EnableTransactionManagement

@MapperScan(

        annotationClass=TestMapper.class,

        basePackages="com.kg.uppp",

        sqlSessionFactoryRef="sqlSessionFactoryBean")

public class TestContext {

    protected final Logger log = LoggerFactory.getLogger(this.getClass());

    

    @Bean

    public SqlSessionFactoryBean sqlSessionFactoryBean( 

            DataSource dataSource, ApplicationContext applicationContext) throws IOException {

        SqlSessionFactoryBean factoryBean = new SqlSessionFactoryBean();

        // 마이바티스가 사용한 DataSource를 등록

        factoryBean.setDataSource(dataSource);

        

        return factoryBean;

    }

    /**

     * 마이바티스 {@link org.apache.ibatis.session.SqlSession} 빈을 등록한다.

     *

     * SqlSessionTemplate은 SqlSession을 구현하고 코드에서 SqlSession를 대체하는 역할을 한다.

     * 쓰레드에 안전하게 작성되었기 때문에 여러 DAO나 매퍼에서 공유 할 수 있다.

     */

    @Bean

    public SqlSessionTemplate sqlSession(SqlSessionFactory sqlSessionFactory) {

        return new SqlSessionTemplate(sqlSessionFactory);

    }

    

        

}