1、BeanFactoryPostProcessor的使用
下面我们来实际使用一下BeanFactoryPostProcessor，来体会它对BeanDefinition的访问和修改。
先说下需求吧：构造几个Color的类对象，并在 bean 还没有创建之前，把 bean 的名称设置到 bean 的属性上。
1.1 声明bean
咱不搞那么复杂的模型，一个Color的抽象类加两个子类就好：（注意两个子类要标注@Component注解注册到 IOC 容器）
public abstract class Color {
protected String name;
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
}
@Component
public class Red extends Color {
@Override
public String toString() {
return "Red{" + "name='" + name + ''' + "}";
}
}
@Component
public class Green extends Color {
@Override
public String toString() {
return "Green{" + "name='" + name + ''' + "}";
}
}
1.2 编写后置处理器
先理一下思路：后置处理器中，既然拿到的参数是BeanFactory，那就可以取出里面的BeanDefinition，并给它们添加属性值了。
先把架子写出来，由于ConfigurableListableBeanFactory无法通过类型取出指定的 bean ，所以只能取出全部，挨个判断了：
@Component
public class ColorNameSetterFactoryPostProcessor implements BeanFactoryPostProcessor {
@Override
public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
Stream.of(beanFactory.getBeanDefinitionNames()).forEach(beanName -> {
BeanDefinition beanDefinition = beanFactory.getBeanDefinition(beanName);
});
}
}
之后就是判断 BeanDefinition 内部生成 bean 的类型了，由于只能获取到 beanClassName ，没有办法取父类，所以只能借助反射来搞定了：
public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
Stream.of(beanFactory.getBeanDefinitionNames()).forEach(beanName -> {
BeanDefinition beanDefinition = beanFactory.getBeanDefinition(beanName);
if (StringUtils.hasText(beanDefinition.getBeanClassName())) {
if (Class.forName(beanDefinition.getBeanClassName()).getSuperclass().equals(Color.class)) {
}
}
});
}
不过这样写，Class.forName()方法会提示需要捕捉ClassNotFoundException。最简单的方法是在整个 foreach 外层套一个try-catch，不过我们可以利用 SpringFramework 中的工具类ClassUtils来避免异常捕捉：
ClassUtils.resolveClassName(beanDefinition.getBeanClassName(), this.getClass().getClassLoader())
如果这样判断后，bean 的父类确实是 Color ，那就可以添加属性值了：（以下是完整写法）
public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
Stream.of(beanFactory.getBeanDefinitionNames()).forEach(beanName -> {
BeanDefinition beanDefinition = beanFactory.getBeanDefinition(beanName);
if (StringUtils.hasText(beanDefinition.getBeanClassName())) {
if (ClassUtils.resolveClassName(beanDefinition.getBeanClassName(), this.getClass().getClassLoader())
.getSuperclass().equals(Color.class)) {
beanDefinition.getPropertyValues().add("name", beanName);
}
}
});
}
1.3 测试运行
与前面的套路一样，直接使用注解 IOC 容器，包扫描驱动：
public class FactoryProcessorApplication {
public static void main(String[] args) throws Exception {
AnnotationConfigApplicationContext ctx = new AnnotationConfigApplicationContext(
"com.linkedbear.spring.postprocessor.h_factoryprocessor");
Red red = ctx.getBean(Red.class);
System.out.println(red);
}
}
运行 main 方法，控制台打印出 red 的 name 属性，证明 ColorNameSetterFactoryPostProcessor 已经起到了作用。
1.4 替代方案
可能会有小伙伴察觉到哪里不对劲了：哎，这个策略如果用BeanPostProcessor实现，岂不是更简单？
public class ColorNameSetterPostProcessor implements BeanPostProcessor {
@Override
public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
if (bean instanceof Color) {
Color color = (Color) bean;
color.setName(beanName);
}
return bean;
}
}
显然，这种写法更简单是吧。对于这种需求来讲，固然是使用BeanPostProcessor更好。小伙伴辩证的根据需求确定解决方案，这个非常好！赞赏之余，小松架构想多说一句：既然 SpringFramework 中有这个设计，自然也有它的使用场景，而且我们最好对比着来看这两种不同的后置处理器。
