[스프링 핵심 원리] 04. 스프링 컨테이너와 스프링 빈 - 스프링 빈 조회(동일한 타입이 둘 이상)

(인프런) 김영한님의 스프링 핵심 원리-기본편을 공부하고 리뷰한 글입니다.

4. 스프링 빈 조회(동일한 타입이 둘 이상)

ac.get(타입)으로 조회할 때, 같은 타입의 스프링 빈이 둘 이상이면 NoUniqueBeanDefinitionException 예외가 발생한다. 이럴 경우,

1) ac.get(빈 이름, 타입) 으로 1개의 빈을 조회한다.

2) ac.getBeansOfType() 으로 해당 타입의 모든 빈을 조회한다.

 

1. ApplicationContextSameBeanFindTest 예제 코드 

package hello.core.beanfind;

import hello.core.member.MemberRepository;
import hello.core.member.MemoryMemberRepository;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.NoUniqueBeanDefinitionException;
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;
import static org.junit.jupiter.api.Assertions.assertThrows;

public class ApplicationContextSameBeanFindTest {
    // 스프링 컨테이너 생성
    AnnotationConfigApplicationContext ac = new AnnotationConfigApplicationContext(SameBeanConfig.class);
    
    // 실패 테스트
    @Test
    @DisplayName("타입으로 조회시 같은 타입이 둘 이상 있으면, 중복 오류가 발생한다")
    void findBeanByTypeDuplicate() {
        // 오른쪽 로직이 수행될 때 왼쪽 예외가 터져야 테스트 성공
        assertThrows(NoUniqueBeanDefinitionException.class,
                () -> ac.getBean(MemberRepository.class));
    }

    // 성공 테스트
    @Test
    @DisplayName("타입으로 조회시 같은 타입이 둘 이상 있으면, 빈 이름을 지정하면 된다")
    void findBeanByName() {
        MemberRepository memberRepository = ac.getBean("memberRepository1", MemberRepository.class);
        // memberRepository가 MemberRepository의 인스턴스인지 검증
        assertThat(memberRepository).isInstanceOf(MemberRepository.class);
    }
    
    @Test
    @DisplayName("특정 타입을 모두 조회하기")
    void findAllBeanByType() {
        // MemberRepository 타입의 모든 빈을 Map으로 반환
        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);
        // MemberRepository 타입의 빈이 2개인지 확인
        assertThat(beansOfType.size()).isEqualTo(2);
    }

    // static: ApplicationContextSameBeanFindTest 클래스에서만 쓰겠다
    @Configuration
    static class SameBeanConfig {
        // 빈의 이름이 다르고 객체 인스턴스 타입이 같음(파라미터 값이 다를 수 있음)
        @Bean
        public MemberRepository memberRepository1() {
            return new MemoryMemberRepository();
        }

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

1) SameBeanConfig - 설정 정보 클래스

이 테스트 클래스내에서만 사용하기 위해 만든 Config 클래스이기때문에 static으로 선언하였다.

// static: ApplicationContextSameBeanFindTest 클래스에서만 쓰겠다
@Configuration
static class SameBeanConfig {
    // 빈의 이름이 다르고 객체 인스턴스 타입이 같음(파라미터 값이 다를 수 있음)
    @Bean
    public MemberRepository memberRepository1() {
        return new MemoryMemberRepository();
    }

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

• 타입 = MemberRepository인 빈 2개를 등록한다.

2) findBeanByTypeDuplicate() - getBean(타입)으로 조회시 같은 타입이 2개 이상이면 중복 오류 발생

// 실패 테스트
@Test
@DisplayName("타입으로 조회시 같은 타입이 둘 이상 있으면, 중복 오류가 발생한다")
void findBeanByTypeDuplicate() {
    // 오른쪽 로직이 수행될 때 왼쪽 예외가 터져야 테스트 성공
    assertThrows(NoUniqueBeanDefinitionException.class,
            () -> ac.getBean(MemberRepository.class));
}

• 타입 = MemberRepository 인 빈을 조회한다.
• 타입이 MemberRepository인 빈이 2개이므로 NoUniqueBeanDefinitionException 예외가 발생한다.

3) findBeanByName() - 같은 타입이 2개 이상이어도 getBean(빈 이름, 타입)으로 조회하면 성공

// 성공 테스트
@Test
@DisplayName("타입으로 조회시 같은 타입이 둘 이상 있으면, 빈 이름을 지정하면 된다")
void findBeanByName() {
    MemberRepository memberRepository = ac.getBean("memberRepository1", MemberRepository.class);
    // memberRepository가 MemberRepository의 인스턴스인지 검증
    assertThat(memberRepository).isInstanceOf(MemberRepository.class);
}

• 이름 = memberRepository1, 타입 = MemberRepository 인 빈을 조회한다.
• 조회한 빈의 객체 타입이 MemberRepository 인지 검증한다. 

4) findAllBeanByType() - 특정 타입을 가진 모든 빈 조회

@Test
@DisplayName("특정 타입을 모두 조회하기")
void findAllBeanByType() {
    // MemberRepository 타입의 모든 빈을 Map으로 반환
    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);
    // MemberRepository 타입의 빈이 2개인지 확인
    assertThat(beansOfType.size()).isEqualTo(2);
}

• 타입 = MemberRepository 인 모든 빈을 조회한다.
• Map으로 반환된 모든 빈의 개수가 2개가 맞는지 검증한다.