[스프링 핵심원리] - 3.스프링 컨테이너와 스프링 빈 (3)

이 글은 강의 : 김영한님의 - "스프링 핵심원리 - 기본편"을 듣고 정리한 내용입니다. 😁😁

이번에는 동일한 타입이 둘 이상인 경우의 스프링 빈 조회를 알아볼 것이다.

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

타입으로 조회시 같은 타입의 스프링 빈이 둘 이상이면 오류가 발생한다. 이때는 빈 이름을 지정하자.

• ac.getBean(빈 이름, 타입)으로 조회
• ac.getBeansOfType()으로 해당 타입의 모든 빈을 조회

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

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개를 등록한다.

findBeanByTypeDuplicate() - getBean(타입)으로 조회 시 오류

getBean(타입)으로 조회시 같은 타입이 2개 이상이면 중복오류를 발생시킨다.

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

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

findBeanByName() - 같은 타입이 2개 이상 getBean(빈 이름, 타입)

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

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

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

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개가 맞는지 검증.