(스프링 기본) 9 - 스프링 빈 조회

■ 기본

스프링 컨테이너에서 스프링 빈을 찾는 가장 기본적인 방법은 두 가지가 있다. 

 

1. ac.getBean(빈이름, 빈타입)

2. ac.getBean(빈타입)

 

만약 조회하는 스프링 빈이 없을 경우 NoSuchBeanDefinitionException 이 발생한다. 

 

AnnotationConfigApplicationContext ac = new AnnotationConfigApplicationContext(AppConfig.class);

@Test
@DisplayName("빈 이름으로 조회")
void findBeanByName() {
    MemberService memberService = ac.getBean("memberService", MemberService.class);
    assertThat(memberService).isInstanceOf(MemberServiceImpl.class);
}

@Test
@DisplayName("이름 없이 타입으로만 조회")
void findBeanByType() {
    MemberService memberService = ac.getBean(MemberService.class);
    assertThat(memberService).isInstanceOf(MemberServiceImpl.class);
}

@Test
@DisplayName("구체 타입으로 조회")
void findBeanByName2() {
    MemberServiceImpl memberService = ac.getBean("memberService", MemberServiceImpl.class);
    assertThat(memberService).isInstanceOf(MemberServiceImpl.class);
}

 

빈 이름과, 타입으로 각각 조회할 수 있다. 

생성된 객체가 MemberServiceImpl 로 잘 생성했는지 비교해본다.  

 

구체 타입으로 조회는 할 수 있지만 유연성이 떨어져서 권장하지 않는다. 

 

    @Test
    @DisplayName("빈 이름으로 조회X")
    void findBeanByNameX() {
//        ac.getBean("xxxxx", MemberService.class);
        assertThrows(NoSuchBeanDefinitionException.class,
                () -> ac.getBean("xxxxx", MemberService.class));
    }

 

없는 빈 이름으로 조회할 때는 junit 의 Assertions 를 import 하고,

assertThrows 를 통해 예외가 잘 터지는지 람다 식으로 확인해보면 된다. 

 

■ 동일 타입한 타입이 둘 이상

 

@Configuration
static class SameBeanConfig {

    @Bean
    public MemberRepository memberRepository1() {
        return new MemoryMemberRepository();
    }

    @Bean
    public MemberRepository memberRepository2() {
        return new MemoryMemberRepository();
    }

}

 

테스트를 위해 임시로 SameBeanConfig class 를 만들어 해당 클래스에서 사용한다.

 

※ class 앞에서 class 생성 시 static 붙이는 이유는 해당 클래스에서만 사용하고,

굳이 밖에 class 객체 생성 후 사용하지 않아도 되게 한다. 

 

AnnotationConfigApplicationContext ac = new AnnotationConfigApplicationContext(SameBeanConfig.class);

@Test
@DisplayName("타입으로 조회 시 같은 타입이 둘 이상 있으면, 중복 오류가 발생한다.")
void findBeanByTypeDuplication() {
    assertThrows(NoUniqueBeanDefinitionException.class,
            () -> ac.getBean(MemberRepository.class));
}

 

동일한 타입이 둘 이상 있으면 NoUniqueBeanDefinitionException 에러가 발생한다. 

여기서는 memberRepository1 와 memberRepository2 가 모두 MemoryMemberRepository 같은 타입 객체를 반환해서 

둘 이상 타입이 들어가 오류가 발생한다. 

 

@Test
@DisplayName("타입으로 조회 시 같은 타입이 둘 이상 있으면, 빈 이름을 지정하면 된다.")
void findBeanByName() {
    MemberRepository memberRepository = ac.getBean("memberRepository1", MemberRepository.class);
    assertThat(memberRepository).isInstanceOf(MemoryMemberRepository.class);
}

 

따라서 이럴 경우 빈 이름을 지정해서 문제를 없애면 된다. 

 

@Test
@DisplayName("특정 타입을 모두 조회하기")
void findAllBeanByType() {
    Map<String, MemberRepository> beansOfType = ac.getBeansOfType(MemberRepository.class);
    for (String key : beansOfType.keySet()) {
        System.out.println("key = " + key + " value = " + beansOfType.get(key));
    }
    System.out.println("beansOfType = " + beansOfType);
    assertThat(beansOfType.size()).isEqualTo(2);
}

 

getBeansOfType(타입) 을 통해 MemberRepository 에 해당하는 타입 모두 조회가 가능하다.

메서드는 Map 으로 생성되며 iter 반복자를 통해 출력할 수 있다. 

 

우리가 넣은 것은 총 2개 이므로 size() 함수를 통해 2개가 맞는지 검증도 가능하다. 

 

■ 상속 관계  

 

 

부모 타입으로 빈 조회 시, 자식 타입도 함께 조회된다. 

따라서 자바의 최고 부모인 Object 타입으로 조회 시, 모든 스프링 빈을 조회한다. 

 

@Configuration
static class TestConfig{

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

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

 

여기서 예시 TestConfjg 를 만들어준다. 

 

AnnotationConfigApplicationContext ac = new AnnotationConfigApplicationContext(TestConfig.class);

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

 

부모 타입 DiscountPolicy 로 조회 시 자식이 rateDiscountPolicy 와 fixDiscountPolicy

두 개 이상이여서 NoUniqueBeanDefinitionException 에러가 난다. 

 

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

 

그럴 경우 마찬가지로 빈 이름을 지정해서 조회하면 된다. 

 

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

 

특정 하위 타입으로 조회 방법도 있지만 이 방법은 권장하지 않는다. 

 

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

 

부모 타입으로 모두 조회를 원하는 경우 getBeansOfType 을 통해 맵 형태로 조회도 가능하다. 

 

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

 

Object 는 자바 객체의 최상위 부모로 Object.class 로 찾을 경우 전체 빈을 불러오게 된다. 

 

 

출처 : 김영한 스프링 핵심 원리 - 기본편

https://www.inflearn.com/course/%EC%8A%A4%ED%94%84%EB%A7%81-%ED%95%B5%EC%8B%AC-%EC%9B%90%EB%A6%AC-%EA%B8%B0%EB%B3%B8%ED%8E%B8

스프링 핵심 원리 - 기본편 강의 | 김영한 - 인프런