조회한 빈이 모두 필요할 때 - List, Map 사용

의도적으로 정말 해당 타입의 스프링 빈이 전부 필요한 경우도 있을 수 있다.

예를 들어, 할인 서비스를 제공하는데 클라이언트가 할인 방식(rate, fix)을 선택할 수 있다고 가정해보자.

스프링을 사용해서 이러한 상황을 해결할 수 있는 전략 패턴을 매우 간단하게 구현할 수 있다. 

 

예제

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, "userA", Grade.VIP);
        int discountPrice = discountService.discount(member, 10000, "fixDiscountPolicy");

        assertThat(discountService).isInstanceOf(DiscountService.class);
        assertThat(discountPrice).isEqualTo(1000);

        int rateDiscountPrice = discountService.discount(member, 30000, "rateDiscountPolicy");
        assertThat(rateDiscountPrice).isEqualTo(3000);
    }

    static class DiscountService {
        private final Map<String, DiscountPolicy> policyMap;
        private final List<DiscountPolicy> policies;

        @Autowired
        public DiscountService(Map<String, DiscountPolicy> policyMap, List<DiscountPolicy> policies) {
            this.policyMap = policyMap;
            this.policies = policies;
            System.out.println("policyMap = " + policyMap);
            System.out.println("policies = " + policies);
            /*
            policyMap = {fixDiscountPolicy=hello.core.discount.FixDiscountPolicy@32811494, rateDiscountPolicy=hello.core.discount.RateDiscountPolicy@4795ded0}
            policies = [hello.core.discount.FixDiscountPolicy@32811494, hello.core.discount.RateDiscountPolicy@4795ded0]
             */
        }

        public int discount(Member member, int price, String discountCode) {
            DiscountPolicy discountPolicy = policyMap.get(discountCode);
            return discountPolicy.discount(member, price);
        }
    }
}

< 로직 >

• DiscountService 는 Map 으로 모든 DiscountPolicy 타입의 클래스를 주입받는다. 
• 이때 Map의 key인 String 에는 스프링 빈 이름이 등록된다. 
  • fixDiscountPolicy, rateDiscountPolicy 등등
• discount()
  • discountCode 로 키에 해당하는 스프링 빈 이름을 받아서 해당 스프링 빈을 찾아 실행
  • discountCode 로 "fixDiscountPolicy" 가 넘어오면 map 에서 fixDiscountPolicy 스프링 빈을 찾아서 실행한다. 

 

< 주입 >

• Map<String, DiscountPolciy> : map 의 키에 스프링 빈의 이름을 넣어주고, 그 값으로 DiscountPolicy 타입으로 조회한 모든 스프링 빈을 담아준다. 
• 만약 해당하는 타입의 스프링 빈이 없으면 빈 컬렉션이나 Map 을 주입