spring動態註冊bean會使AOP失效?

語言: CN / TW / HK

1

前言

本文的素材來自讀者的一個問題,他看過我之前寫的一篇博文 聊聊如何把第三方服務註冊到我們專案的spring容器中 。剛好他專案中也有類似這樣的一個需求,他就採用我文中介紹的第三種方法

呼叫beanFactory.registerSingleton()

一開始專案執行得還可以,後面他在這個第三方服務中使用AOP,發現AOP始終沒有生效。於是他就給我留言了。今天就來聊一下這個話題, 為什麼使用registerSingleton()註冊的bean,無法使AOP生效

2

問題根源

registerSingleton()這個方法直接將bean存放到單例池裡面了。

如果對bean的生命週期有了解的朋友,應該會知道,bean可能會經過一系列的後置處理器後,再存放到單例池裡面。因此這個bean可能是會被增強的,其中當然包括經過AOP增強

而使用registerSingleton()相當於是直接走捷徑,不經過後置處理器,一步到位直接存放到單例池中。如果第三方服務是直接通過new出來的,就是一個普通的物件,因此注入到IOC容器後,也只是一個普通的bean,並沒有任何增強

3

問題修復

01

方案一:不使用registerSingleton(),而是使用BeanDefinition註冊方式

這種方式本質是讓這個物件完整經歷了bean的生命週期

示例:

@Configuration
public class HelloServiceConfiguration implements BeanFactoryPostProcessor {


@Override
public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {
DefaultListableBeanFactory defaultListableBeanFactory = (DefaultListableBeanFactory) configurableListableBeanFactory;
BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition();
AbstractBeanDefinition beanDefinition = beanDefinitionBuilder.getBeanDefinition();
beanDefinition.setBeanClass(HelloService.class);
HelloServiceProperties properties = new HelloServiceProperties();
properties.setBeanName("helloService");
beanDefinition.getConstructorArgumentValues().addIndexedArgumentValue(0,properties);
defaultListableBeanFactory.registerBeanDefinition(properties.getBeanName(),beanDefinition);

}
}

02

方案二、使用registerSingleton(),但注入的物件不是用new出來的,而是直接注入AOP代理物件

主要利用AOP的代理api:AnnotationAwareAspectJAutoProxyCreator

示例

@Configuration
public class HelloServiceWithProxyConfiguration implements BeanFactoryAware, InitializingBean {

private BeanFactory beanFactory;

@Autowired
private AnnotationAwareAspectJAutoProxyCreator annotationAwareAspectJAutoProxyCreator;


@Override
public void setBeanFactory(BeanFactory beanFactory) throws BeansException {
this.beanFactory = beanFactory;
}

@Override
public void afterPropertiesSet() throws Exception {

DefaultListableBeanFactory defaultListableBeanFactory = (DefaultListableBeanFactory)beanFactory;
HelloServiceProperties properties = new HelloServiceProperties();
properties.setBeanName("helloService");
HelloService helloServicePrxoy = (HelloService) annotationAwareAspectJAutoProxyCreator.postProcessAfterInitialization(new HelloService(properties), "helloService$$Prxoy");
defaultListableBeanFactory.registerSingleton(properties.getBeanName(),helloServicePrxoy);

}
}

4

總結

以上兩種方案,建議使用方案一。因為方案一完整經歷過bean的生命週期,這就意味著可以獲取spring提供的各種增強功能。方案二反而更像是硬編碼進去,如果後面要使用spring的其他增強的功能,就還必須呼叫其他API。不過如果可以確定業務不會使用spring提供的各種擴充套件功能。方案二也是可以的

5

demo連結

https://github.com/lyb-geek/springboot-learning/tree/master/springboot-registerSingleton-aop-invalid