같은 타입인 여러 개의 빈들을 조회하고 싶을 때

조회한 빈이 모두 필요할 때

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

예를 들어서 할인 서비스를 제공하는데, 클라이언트가 할인의 종류(rate - 비율로 할인, fix - 고정금액 할인)를 선택할 수 있다고

가정해보자.

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

package hello.core.autowired;

import hello.core.AutoAppConfig;
import hello.core.discount.discount.DiscountPolicy;
import hello.core.member.Grade;
import hello.core.member.Member;
//import org.junit.jupiter.api.Assertions;
import org.assertj.core.api.Assertions;
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;

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

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

		//rate일 때
		int rateDiscountPrice = discountService.discount(member, 20000, "rateDiscountPolicy");
		Assertions.assertThat(rateDiscountPrice).isEqualTo(2000);
	}

	static class DiscountService{
		private final Map<String, DiscountPolicy> policyMap; //모든 discountPolicy 받기
		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);
		}
	}
}

로직 분석

DiscountService는 Map으로 모든 DiscountPolicy 를 생성자로 주입받는다. (지금 생성자가 1개이므로 @Autowired 생략 가능)

• 이때 fixDiscountPolicy ,rateDiscountPolicy 가 주입된다.

	static class DiscountService{
		private final Map<String, DiscountPolicy> policyMap; //모든 discountPolicy 받기
		private final List<DiscountPolicy> policies;

		@Autowired // 빈으로 등록 (생성자가 1개이므로 생략 가능)
		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);
	}

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

→ map에서 fixDiscountPolicy 스프링 빈을 찾아서 해당하는 로직을 discountPolicy.discount(member, price); 실행한다.

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

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

주입 분석

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

List<DiscountPolicy> : DiscountPolicy 타입으로 조회한 모든 스프링 빈을 담아준다. 만약 해당하는 타입의 스프링 빈이 없으면, 빈 컬렉션이나 Map을 주입한다.

• 장점
• 동적으로 빈을 받아올 때 다형성 코드를 유지하면서 편리하게 사용할 수 있다.

참고 - 스프링 컨테이너를 생성하면서 스프링 빈 등록하기

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

이 코드는 2가지로 나누어 이해할 수 있다. new AnnotationConfigApplicationContext() 를 통해 스프링 컨테이너를 생성한다. AutoAppConfig.class , DiscountService.class 를 파라미터로 넘기면서 해당 클래스를 자동으로

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

 

 

이 글은 김영한님의 스프링 기본편을 듣고 정리한 글입니다.