🔥 TIL - Day 81 BeanPostProcessor 빈 후처리기

설정파일을 이용해서 직접 Bean을 등록하는 경우가 아닌 컴포넌트 스캔을 이용해서 스프링 컨테이너에 Bean을 등록하는 경우 등록되는 Bean에 대해 조작할 수가 없다. (컴포넌트 스캔에 의해 자동으로 등록되기 때문)

컴포넌트 스캔을 통해 등록되는 Bean을 대신해서 프록시 객체를 등록하고 싶은 경우 불가능한 것이다.

빈 후처리기는 스프링 컨테이너에 등록되는 모든 빈에 대해 조작이 가능하도록 도와준다.
등록되는 Bean에 대해서 추가적인 작업도 가능하고 동일한 Bean이름으로 아예 다른 Bean을 등록하는 것도 가능하다.

---

📌 빈 후처리기 예제

예제 시나리오

• 두 개 클래스를 생성하고 (A클래스, B클래스) A클래스만 스프링 컨테이너에 Bean으로 등록한다.
• 빈 후처리기를 구현하고 빈 후처리기에서 A타입으로 등록되는 Bean을 B타입으로 바꿔서 등록한다.
• 결과적으로 Bean이 A이름으로 B타입 객체가 Bean을 등록되는 것과 A타입으로 등록된 Bean이 없는 것을 확인한다.

두 개 클래스 생성

    @Slf4j
    static class A {
        public void call() {
            log.info("Test A");
        }
    }

    @Slf4j
    static class B {
        public void call() {
            log.info("Test B");
        }
    }

A클래스의 객체만 Bean으로 등록

    @Configuration
    static class BeanPostProcessorTestConfig {
        @Bean // A만 빈으로 등록
        public A a() {
            return new A();
        }
    }

빈후처리기 구현
빈후처리기는 BeanPostProcessor 인터페이스를 구현하고 구현한 인터페이스를 Bean으로 등록하는 것으로 동작한다.

@Slf4j
static class AToBPostProcessor implements BeanPostProcessor {
    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        log.info("bean={}, beanName={}", bean, beanName);

        // 등록되는 빈이 A 타입이라면 B의 객체를 생성해서 등록하도록 처리 (Hooking)
        // 등록되는 빈의 이름은 그대로 'a' 이지만 빈의 실제 객체는 B 타입이 된다.
        if (bean instanceof A) return new B();
        return bean;
    }
}

구현된 빈후처리기를 Bean으로 등록한다.

    @Configuration
    static class BeanPostProcessorTestConfig {
        @Bean // A만 빈으로 등록
        public A a() {
            return new A();
        }

        @Bean // 빈후처리기 등록
        public AToBPostProcessor postProcessor() {
            return new AToBPostProcessor();
        }
    }

테스트

• a로 등록된 Bean을 B클래스 타입으로 조회가 가능해야 한다.
• A클래스 타입으로 등록된 Bean 자체가 없어야 한다. (A타입 Bean이 등록될 때 B타입 Bean을 등록했기 때문)

@Test
void basicConfig() {
    ApplicationContext context = new AnnotationConfigApplicationContext(BeanPostProcessorTestConfig.class);
    B b = context.getBean("a", B.class);
    b.call();

    Assertions.assertThrows(NoSuchBeanDefinitionException.class, () -> context.getBean(A.class));
}

빈 후처리기를 이용해서 컴포넌트 스캔에 의해 자동으로 등록되는 Bean에 대해서도 조작이 가능하게 됐다. 이 기능을 이용해서 컴포넌트 스캔으로 등록되는 Bean에 대해서도 프록시 객체가 대신해서 등록되도록 하는 것이 가능하다.

---

📌 참고

인프런 - 스프링 핵심 원리 고급편 (김영한 님)