Spring5.x源码之旅三十七之MergedBeanDefinitionPostProcessor扩展点

在实例化之后&＃xff0c;合并bean定义&＃xff0c;其实就是更新bean定义啦&＃xff0c;这里可以再次修改bean定义&＃xff0c;这里只能修改RootBeanDefinition类型的。前面说过了&＃xff0c;主要还是InitDestroyAnnotationBeanPostProcessor&＃xff0c;CommonAnnotationBeanPostProcessor和AutowiredAnnotationBeanPostProcessor处理有注入注解的属性或者方法。分别去处理生命周期PostConstruct&＃xff0c;PreDestroy注解&＃xff0c;Resource&＃xff0c;WebServiceRef&＃xff0c;EJB注解&＃xff0c;Autowired和Value注解。

其实这个可以和上面那个配合&＃xff0c;比如上面那个做一些处理&＃xff0c;这个后面就可以用&＃xff0c;我们下面来做简单的子类看看。

我想定义一个注解&＃xff0c;注解上有个属性&＃xff0c;就是要打印方法的次数&＃xff0c;我希望能找到所有这个属性定义的方法&＃xff0c;然后打印他们。

MyAnnotation注解

属性count就是打印的次数。

&＃64;Target(ElementType.METHOD)
&＃64;Retention(RetentionPolicy.RUNTIME)
public &＃64;interface MyAnnotation {int count() default 0;
}

MyMergedBean

我们注解一些方法&＃xff0c; 看看他会不会打印。

&＃64;Component
public class MyMergedBean {&＃64;MyAnnotation(count &＃61; 1)public void m1(String msg) {print(msg);}&＃64;MyAnnotation(count &＃61; 2)public void m2(String msg) {print(msg);}&＃64;MyAnnotation()public void m3(String msg) {print(msg);}private void print(String msg) {System.out.println(msg);}
}


MyMergedBeanDefinitionPostProcessor 处理器

这里主要是实现了postProcessMergedBeanDefinition&＃xff0c;做了实例化之后的处理&＃xff0c;然后在初始化后postProcessBeforeInitialization具体进行处理。主要是打印有注解的方法&＃xff0c;根据注解的属性。

&＃64;Component
public class MyMergedBeanDefinitionPostProcessor implements MergedBeanDefinitionPostProcessor {//bean名字对应的注解方法public Map<String,List<Method>> stringMethodMap;&＃64;Nullableprivate Class<? extends Annotation> myAnnotationType;public MyMergedBeanDefinitionPostProcessor(){myAnnotationType&＃61;MyAnnotation.class;stringMethodMap&＃61;new HashMap<>();}&＃64;Overridepublic void postProcessMergedBeanDefinition(RootBeanDefinition beanDefinition, Class<?> beanType, String beanName) {List<Method> list&＃61;new ArrayList<>();ReflectionUtils.doWithLocalMethods(beanType, method -> {if (this.myAnnotationType !&＃61; null && method.isAnnotationPresent(this.myAnnotationType)) {list.add(method);stringMethodMap.put(beanName,list);}});}&＃64;Overridepublic Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {if(stringMethodMap.get(beanName)!&＃61;null){for (Method method : stringMethodMap.get(beanName)) {try {MyAnnotation annotation &＃61; (MyAnnotation) method.getAnnotation(this.myAnnotationType);for (int i &＃61; 0; i < annotation.count(); i&＃43;&＃43;) {method.invoke(bean,new Object[]{method.getName()});}} catch (IllegalAccessException e) {e.printStackTrace();} catch (InvocationTargetException e) {e.printStackTrace();}}}return bean;}
}


测试类

&＃64;Testpublic void MergedBeanDefinitionPostProcessorTest() throws Exception {AnnotationConfigApplicationContext applicationContext &＃61; new AnnotationConfigApplicationContext();applicationContext.register(MyConfig.class);applicationContext.refresh();}


我再改改&＃xff1a;

结果&＃xff1a;

为什么不是按定义顺序执行呢&＃xff0c;好像是因为JDK反射拿出来就是无序的&＃xff0c;如果要有序可以用CGLIB的ClassVisitor来拿&＃xff0c;具体spring源码里有&＃xff0c;可以参考ConfigurationClassParser的retrieveBeanMethodMetadata&＃xff1a;

好了&＃xff0c;MyMergedBeanDefinitionPostProcessor到底有什么用呢&＃xff0c;就看你的业务啦&＃xff0c;只要你想在实例化后做扩展&＃xff0c;就可以尝试用这个&＃xff0c;参与bean的初始化的过程。

好了&＃xff0c;今天就到这里了&＃xff0c;希望对学习理解有帮助&＃xff0c;大神看见勿喷&＃xff0c;仅为自己的学习理解&＃xff0c;能力有限&＃xff0c;请多包涵。