[Spring] 조회한 빈이 모두 필요한 경우 (List, Map)

의도적으로 해당 타입의 스프링 빈이 다 필요한 경우도 종종 있다.

예를 들어 할인 서비스를 제공하는데, 클라이언트가 할인의 종류를 1개가 아닌 2개 이상 선택할 수 있도록 요구한다.

스프링은 이를 전략 패턴으로 매우 간단하게 구현할 수 있다.

 

public class AllBeanTest {

    @Test
    void findAllBean() {
        ApplicationContext ac = new AnnotationConfigApplicationContext(AutoAppConfig.class, DiscountService.class);
        DiscountService discountService = ac.getBean(DiscountService.class);
        
        Member member = new Member(1L, "user", Grade.VIP);
        int discountPrice = discountService.discount(member, 10000, "fixDiscountPolicy");
        
        assertThat(discountService).isInstanceOf(DiscountService.class);
        assertThat(discountPrice).isEqualTo(1000);
    }
    
    static class DiscountService {
        private final Map<String, DiscountPolicy> policyMap;
        private final List<DiscountPolicy> policies;
        
        public DiscountService(Map<String, DiscountPolicy> policyMap,
        		       List<DiscountPolicy> policies) {
            this.policyMap = policyMap;
            this.policies = policies;
    	}
    
        public int discount(Member member, int price, String discountCode) {
            DiscountPolicy discountPolicy = policyMap.get(discountCode);
            return discountPolicy.discount(member, price);
        }
    }
}

 

fixDiscountPolicy와 rateDiscountPolicy가 스프링 빈에 등록되었다고 가정하자.

DiscountService는 Map으로 모든 DiscountPolicy를 주입받는다. 이 시점에서 fixDiscountPolicy, rateDiscountPolicy가 주입된다. 이후 discountService.discount() 메서드를 통해 원하는 할인 정책을 넣어주면 policyMap에서 원하는 할인 정책(스프링 빈)을 찾아 넣어준다.

 

Map<String, DiscountPolicy>: key에 스프링 빈의 이름을 넣고, value에 맞는 타입으로 조회한 모든 스프링 빈을 담아준다.

List<DiscountPolicy>: DiscountPolicy 타입으로 조회한 모든 스프링 빈을 담아준다.

 

 

스프링 컨테이너는 생성자에 클래스 정보를 받는다. 클래스 정보를 넘기면 해당 클래스가 스프링 빈으로 자동 등록된다. new AnnotationConfigApplicationContext(AutoAppConfig.class,DiscountService.class)는 2가지로 나눠 이해할 수 있다.

new AnnotationConfigApplicationContext() 를 통해 스프링 컨테이너를 생성한다.

AutoAppConfig.class , DiscountService.class 를 파라미터로 넘기면서 해당 클래스를 자동으로 스프링 빈으로 등록한다.

스프링 컨테이너를 생성하면서, 동시에 AutoAppConfig , DiscountService 를 스프링 빈으로 자동 등록한다.

 

 

Reference

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