[Spring] 스프링 빈 조회 - 상속관계

스프링 빈을 조회할 때, 빈이 상속관계로 되어있을 경우(=부모 타입을 조회했는데, 자식 타입이 여러 개 있을 경우) 기본적으로 부모 타입을 조회하면 속해있는 자식 빈들이 모두 조회된다. 그래서 모든 자바 객체의 최고 부모인 Object 타입으로 조회하면, 모든 스프링 빈을 조회한다.

---

예제를 통해 확인해보자. 이를 위해 새로운 클래스를 만들고, 그 안에 사용할 Configuration을 하나 만드는데, 부모 타입은 같지만 자식 타입이 다른 두 개의 빈을 만들자.

package hello.core.beanfind;

import hello.core.discount.DiscountPolicy;
import hello.core.discount.FixDiscountPolicy;
import hello.core.discount.RateDiscountPolicy;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

public class ApplicationContextExtendsFindTest {
    AnnotationConfigApplicationContext ac = new AnnotationConfigApplicationContext(TestConfig.class);

    @Test
    @DisplayName("부모 타입으로 조회 시, 자식이 둘 이상이면 중복 오류")
    void findBeanByParentTypeDuplicate(){
        DiscountPolicy bean = ac.getBean(DiscountPolicy.class);
    }

    @Configuration
    static class TestConfig{

        @Bean
        public DiscountPolicy rateDiscountPolicy(){
            return new RateDiscountPolicy();
        }

        @Bean
        public DiscountPolicy fixDiscountPolicy(){
            return new FixDiscountPolicy();
        }
    }
}

이를 실행해보면 오류가 발생하는 것을 확인할 수 있다. 

따라서 이 예외를 처리하기 위해서는 아래의 코드를 입력하면 된다.

Assertions.assertThrows(NoUniqueBeanDefinitionException.class,
            ()->ac.getBean(DiscountPolicy.class));

 

이 문제의 해결 방법 역시 이전과 마찬가지로 빈의 이름을 지정해주면 된다. 코드는 다음과 같다.

...
@Test
@DisplayName("부모 타입으로 조회 시 자식이 둘 이상이면, 빈의 이름을 지정")
void findBeanByParentTypeBeanName(){
    DiscountPolicy rateDiscountPolicy = ac.getBean("rateDiscountPolicy", DiscountPolicy.class);
    assertThat(rateDiscountPolicy).isInstanceOf(RateDiscountPolicy.class);
}
...

---

다음은 부모 타입으로 모두 조회하는 방법이다.

@Test
@DisplayName("Find all bean by parent type")
void findAllBeanByParentType(){
    Map<String, DiscountPolicy> beansOfType = ac.getBeansOfType(DiscountPolicy.class);
    assertThat(beansOfType.size()).isEqualTo(2);
    for(String key:beansOfType.keySet()){
        System.out.println("key = " + key + " | value = " + beansOfType.get(key));
    }
}