Spring 조회한 빈이 모두 필요할 때, Map으로 담는다.

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

스프링을 사용하면 소위 말하는 전략 패턴을 매우 간단하게 구현할 수 있다.

 

 

주입 분석

public class AllBeanTest {

    @Test
    void findAllBean() {
        ApplicationContext ac = new AnnotationConfigApplicationContext(FixDiscountPolicy.class, RateDiscountPolicy.class, DiscountService.class);
    }

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

        @Autowired
        public DiscountService(Map<String, DiscountPolicy> policyMap) {
            this.policyMap = policyMap;
            System.out.println("policyMap = " + policyMap);
        }
    }
}

결과
policyMap = {fixDiscountPolicy=hello.core.discount.FixDiscountPolicy@2fd953a6
, rateDiscountPolicy=hello.core.discount.RateDiscountPolicy@a4add54}

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

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

• Map<String, DiscountPolicy> : map의 키에 스프링 빈의 이름을 넣어주고, 그 값으로 DiscountPolicy 타입으로 조회한 모든 스프링 빈을 담아준다.

 

 

로직분석

public class AllBeanTest {

    @Test
    void findAllBean() {
        ApplicationContext ac = new AnnotationConfigApplicationContext(FixDiscountPolicy.class, RateDiscountPolicy.class, DiscountService.class);
        DiscountService discountService = ac.getBean(DiscountService.class);
        Member member = new Member(1L, "Jun", Grade.VIP);
        
        int FixDiscountPrice = discountService.discount(member, 10000, "fixDiscountPolicy");
        Assertions.assertThat(FixDiscountPrice).isEqualTo(1000);

        int RateDiscountPrice = discountService.discount(member, 20000, "rateDiscountPolicy");
        Assertions.assertThat(RateDiscountPrice).isEqualTo(2000);
    }

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

        public DiscountService(Map<String, DiscountPolicy> policyMap) {
            this.policyMap = policyMap;
            System.out.println("policyMap = " + policyMap);
        }

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

            System.out.println("discountCode = " + discountCode);
            System.out.println("discountPolicy = " + discountPolicy);

            return discountPolicy.discount(member, price);
        }
    }
}

• DiscountService는 Map으로 모든 DiscountPolicy 를 주입받는다. 이때 fixDiscountPolicy , rateDiscountPolicy 가 주입된다.

• discount () 메서드는 discountCode로 "fixDiscountPolicy"가 넘어오면 map에서 fixDiscountPolicy 스프링 빈을 찾아서 실행한다. 물론 “rateDiscountPolicy”가 넘어오면 rateDiscountPolicy 스프링 빈을 찾아서 실행한다.

 

 

---

REFERENCE

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