[Spring] 조회한 빈(Bean)을 List, Map에 담기

 

 

 

📌 조회한 빈(Bean)을  List, Map에 담기

 

🔍 조회한 빈(Bean) List, Map에 담기

동적으로 스프링 빈을 선택해야 할 때 Map 또는 List로 받아서 그 때 그 때 바꿀 수 있다.

package hello.core.autowired;

import hello.core.AutoAppConfig;
import hello.core.discount.DiscountPolicy;
import hello.core.member.Grade;
import hello.core.member.Member;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;

import java.util.List;
import java.util.Map;

import static org.assertj.core.api.Assertions.*;

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, 20000, "rateDiscountPolicy");
        assertThat(rateDiscountPrice).isEqualTo(2000);

    }

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

        }

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


}

policyMap = {fixDiscountPolicy=hello.core.discount.FixDiscountPolicy@7cbc3762, rateDiscountPolicy=hello.core.discount.RateDiscountPolicy@3a6f2de3}
policies = [hello.core.discount.FixDiscountPolicy@7cbc3762, hello.core.discount.RateDiscountPolicy@3a6f2de3]

• DiscountService는 Map으로 모든 DisCountPolicy를 주입 받는다. 이 때, fixDiscountPolicy, rateDisCountPolicy가 주입된다.
• discount() 메서드의 discountCode로 "fixDiscountPolicy"를 받아서 policyMap에서 fixDiscountPolicy 스프링 빈을 찾아서 실행한다. 만약 discountCode에  "rateDiscountPolicy"가 넘어오면 rateDiscountPolicy 스프링 빈을 찾는다. 

 

💡 주입 분석

• Map<String, DiscountPolicy> 
  • map의 key에 스프링 빈의 이름을 넣어주고, 그 값으로 DiscountPolicy 타입으로 조회한 모든 스프링 빈을 담는다.
• Lisy<DiscountPolicy>
  • DiscountPolicy 타입으로 조회한 모든 스프링 빈을 담는다.
• new AnnotationConfigApplicationContext(AutoAppConfig.class, DiscountService.class)
  • ① new AnnotationConfigApplicationContext() 를 통해 스프링 컨테이너 생성
  • ② AutoAppConfig.class, DiscountService.class 를 파라미터로 넘기면서 해당 클래스를 자동으로 스프링 빈 등록