232일차(모험 141일차) - 조회한 빈이 모두 필요할 때, List, Map

[add] 조회한 빈이 모두 필요할 때, List, Map [#32] #37

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

 

예를 들어서 할인 서비스를 제공하는데,

클라이언트가 할인의 종류(rate, fix)를 선택할 수 있다고 가정해본다.

 

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

 

 

 

* AllBeanTest.java

---

 

Map<String, DiscountPolicy> : map의 키에 스프링 빈의 이름을 넣어주고,

그 값으로 DiscountPolicy 타입으로 조회한 모든 스프링 빈을 담아준다.

 

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

만약 해당하는 타입의 스프링 빈이 없으면, 빈 컬렉션이나 Map을 주입한다.

 

 

 

 

* 생성자 입력 대신 @RequiredArgsConstructor 를 작성해도 된다.

* 생성자가 하나이므로 @Autowired를 생략해도 된다.

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

 

 

DiscountService는 Map으로 모든 DiscountPolicy 를 주입받는다.

 

private final Map<String, DiscountPolicy> policyMap;

이때 fixDiscountPolicy , rateDiscountPolicy 가 주입된다.

 

 

 

 

DiscountService를 스프링 빈으로 등록

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

 

 

 

실행해보면 DiscountService만 스프링 빈으로 등록했기 때문에 아무 값도 뜨지 않는다.

 

 

 

AutoAppConfig와 DiscountService 2개 등록

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

설명 +)

스프링 컨테이너는 생성자에 클래스 정보를 받는다.

여기에 클래스 정보를 넘기면 해당 클래스가 스프링 빈으로 자동 등록된다.

 

new AnnotationConfigApplicationContext(AutoAppConfig.class,DiscountService.class);

 

---

 

 

 

 

자동의존관계 주입이 될 때 AutoAppConfig에서 컴포넌트 스캔을 한다.

FixDiscountPolicy, RateDiscountPolicy 둘 다 스프링 빈에 등록된다.@Component

 

실행 결과 Map은 key : value, List는 value

 

 

 

---

 

비즈니스 로직 만들기

 

discount가 얼마나 되는지 보는 서비스

@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); // 할인 금액이 1,000원이어야함
}

 

 

 

 

discountService.discount에서 discount 코드는 아래와 같다.

 

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

설명+)

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

discount 메소드는 discountCode로 fixDiscountPolicy가 내려오면 

 

 

 

DiscountPolicy discountPolicy = policyMap.get(discountCode);

⬆️ 여기서 찾아서 discountPolicy 에서 스프링 빈을 꺼내서 

 

 

 

private final Map<String, DiscountPolicy> policyMap;

* 현재 Map에 스프링 빈이 넘어 온 상태

 

 

 

return discountPolicy.discount(member, price);  ⬅️ 로직을 호출해준다.

 

 

---

 

 

discount () 메서드는 discountCode로 "fixDiscountPolicy"가 넘어오면

map에서 fixDiscountPolicy 스프링 빈을 찾아서 실행한다.

 

물론 “rateDiscountPolicy”가 넘어오면 rateDiscountPolicy 스프링 빈을 찾아서 실행한다.

 

 

 

 

 

 

실행시키면 성공한다. 추가로 rate도 작성

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); // 할인 금액이 1,000원이어야함


    ////////////// 추가 작성
    int rateDiscountPrice = discountService.discount(member, 20000, "rateDiscountPolicy");
    assertThat(rateDiscountPrice).isEqualTo(2000); // 할인 금액이 2,000원이어야함

}