[Spring] Bean 조회 2 - 동일한 타입 조회시 처리 (NoUniqueBeanDefinitionException)

 

1. 타입 조회 시 동일한 타입이 있을 때

public class ApplicationContextSameBeanFindTest {
    AnnotationConfigApplicationContext ac = new AnnotationConfigApplicationContext(SameBeanConfig.class);

    @Test
    @DisplayName("타입으로 조회 시 같은 타입이 둘 이상 있으면, 중복 오류가 발생한다.")
    void findBeanByTypeDuplicate() {
        MemberReposiroty bean = ac.getBean(MemberReposiroty.class);
    }

    @Configuration
    static class SameBeanConfig {
        @Bean
        public MemberReposiroty memberReposiroty1() {
            return new MemoryMemberRepository();
        }

        @Bean
        public MemberReposiroty memberReposiroty2() {
            return new MemoryMemberRepository();
        }
    }
}

 

이럴 경우 테스트를 진행하면 에러가 발생합니다.

 

 

왜냐?

 

밑에 SameBeanConfig를 보면

 

MemberRepository  타입의 빈이 2개가 있죠?

 

그러면 MemberRepository 타입으로 조회했을 때 스프링이 둘 중에 뭘 반환해줘야할지 모릅니다.

 

그러면서 NoUniqueBeanDefinitionException 에러를 뱉는거죠.

 

자 그러면 어떻게 처리할지 봅시다.

 

package hello.core.beanfind;

import hello.core.member.MemberReposiroty;
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(MemberReposiroty.class));
    }

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

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


    @Configuration
    static class SameBeanConfig {
        @Bean
        public MemberReposiroty memberRepository1() {
            return new MemoryMemberRepository();
        }

        @Bean
        public MemberReposiroty memberRepository2() {
            return new MemoryMemberRepository();
        }
        
        @Bean
        public DiscountPolicy discountPolicy() {
            return new RateDiscountPolicy();
        }
    }
}

 

0) findBeanByTypeDuplicate : 빈 조회했는데 중복 조회될 때

 

assertThrows를 통해 에러가 발생하는 것을 확인해줍니다.

 

1) findBeanByName : 빈 이름을 지정해서 조회하기

 

타입은 같지만 빈 이름은 각각이 다릅니다.

 

그렇기 때문에 getBean에서 bean 이름을 지정을 해줘서 조회를 하는 겁니다.

 

2) findAllBeanByType : 특정 타입을 모두 조회하기

 

그냥 해당하는 타입의 빈을 모두 뽑아오는 방법입니다.

 

Map을 활용해서 빈의 이름과 타입을 저장해줍니다.

 

그 후에 반복문을 통해 출력을 해주면

 

3개의 빈 중 MemberRepository 타입의 빈만 잘 가져온 것을 확인 할 수 잇습니다.

 

마지막은 그냥 전체를 출력한겁니다. -> 나중에 필요할지도?

 

그러면

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

 

이런식으로 DiscountPolicy 타입의 빈도 조회가 가능하겠죠?