[Spring] 빈 조회하기 - 상속 관계

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;

import java.util.Map;

import static org.assertj.core.api.Assertions.assertThat;

public class ApplicationContextExtendsFindTest {

    AnnotationConfigApplicationContext ac = new AnnotationConfigApplicationContext(TestConfig.class);

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

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

    @Test
    @DisplayName("특정 하위 타입으로 조회")
    void findBeanBySubType() {
        RateDiscountPolicy bean = ac.getBean(RateDiscountPolicy.class);
        assertThat(bean).isInstanceOf(RateDiscountPolicy.class);
    }

    @Test
    @DisplayName("부모 타입으로 모두 조회하기")
    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));
        }
    }

    @Test
    @DisplayName("Object 타입으로 모두 조회")
    void findAllBeaByObjectType() {
        Map<String, Object> beansOfType = ac.getBeansOfType(Object.class);
        for (String key : beansOfType.keySet()) {
            System.out.println("key = key" + " value = " + beansOfType.get(key));
        }

    }
    @Configuration
    static class TestConfig {
        @Bean
        public DiscountPolicy rateDiscountPolicy() {
            return new RateDiscountPolicy();
        }

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

 

위의 코드들로 스프링에서 빈을 조회하는 법을 알아보겠습니다.

 

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

우선 위 코드를 실행하면 

NoUniqueBeanDefinitionException 이라는 오류가 발생하게 되는데 이는 ac.getBean(DiscountPolicy.class)에서

Bean을 꺼내올 때 DiscountPolicy 클래스에 자식이 둘 이상이 있기 때문에 특정한 하나의 빈이 없다고 말해주는 것이다.

 

이때 우리가 할 수 있는 것들은

1. 자식 타입들 중 특정한 자식 타입을 지정해 조회하기

@Test
@DisplayName("특정 하위 타입으로 조회")
void findBeanBySubType() {
    RateDiscountPolicy bean = ac.getBean(RateDiscountPolicy.class);
    assertThat(bean).isInstanceOf(RateDiscountPolicy.class);
}

DiscountPolicy 클래스의 자식인 RateDiscountPolicy와 fixDiscountPolicy 중 하나만 선택하여 지정해주는 것이다.

 

2. 부모 타입으로 모두 조회하기

@Test
@DisplayName("부모 타입으로 모두 조회하기")
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));
    }
}

1번 방법은 특정한 자식 하나만 보는 것이지만 자식을 모두 조회하고 싶다면 이 방법을 이용한다.

 

실행 결과

 

3. Object 타입으로 모두 조회

  @Test
    @DisplayName("Object 타입으로 모두 조회")
    void findAllBeaByObjectType() {
        Map<String, Object> beansOfType = ac.getBeansOfType(Object.class);
        for (String key : beansOfType.keySet()) {
            System.out.println("key = key" + " value = " + beansOfType.get(key));
        }
    }

자바는 Object가 최상위 부모 역할을 하므로 Object를 이용하여 모두 조회할 수 있다.

하지만 이 방식을 이용하면 스프링 내의 모든 bean이 다 조회가 되기 때문에 양이 많아진다.

 

출처 : 인프런 - 스프링 핵심 원리 (기본편)