手写spring（6）-BeanPostProcessor模拟实现

BeanPostProcessor是spring对外提供的bean的扩展机制，用于对bean进行扩展操作，常见的比如AOP就是采用BeanPostProcessor接口实现的。

模拟BeanPostProcessor接口

public interface BeanPostProcessor {
    /**
     * 初始化前执行该方法
     * @param bean
     * @param beanName
     * @return
     */
    Object postProcessBeforeInitialization(Object bean, String beanName);

    /**
     * 初始化后执行该方法
     * @param bean
     * @param beanName
     * @return
     */
    Object postProcessAfterInitialization(Object bean, String beanName);
}

定义ZhouyuBeanPostProcessor类实现BeanPostProcessor接口，

@Component
public class ZhouyuBeanPostProcessor implements BeanPostProcessor {
    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) {
        if ("userService".equals(beanName)) {
            System.out.println("初始化前");
            ((UserServiceImpl) bean).setName("周瑜好帅");
        }
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) {
        System.out.println("初始化后");
        if ("userService".equals(beanName)) {
            Object proxyInstance = Proxy.newProxyInstance(ZhouyuBeanPostProcessor.class.getClassLoader(), bean.getClass().getInterfaces(), new InvocationHandler() {
                @Override
                public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
                    System.out.println("代理逻辑"); // aop找切点pointCut：代理逻辑定义在哪些类上
                    return method.invoke(bean,args);
                }
            });

            // 直接返回代理对象
            return proxyInstance;
        }

        // 返回原始对象
        return bean;
    }
}

在BeanPostProcessor的实现类上也加上了@Component注解，在spring启动过程中也会扫描到该BeanPostProcessor，与普通的Bean不同是会将BeanPostProcessor添加到一个List中，后续会在Bean的生命周期中使用到该BeanPostProcessor。

// 存放beanPostProcessor
private List<BeanPostProcessor> beanPostProcessorList = new ArrayList<>();

// 判断是否是BeanPostProcessor
if (BeanPostProcessor.class.isAssignableFrom(clazz)) {
  BeanPostProcessor instance = (BeanPostProcessor) clazz.getDeclaredConstructor().newInstance();
  beanPostProcessorList.add(instance);
}

...
// TODO beanPostProcessor也会走下面的逻辑，在spring中不是这么实现的

BeanPostProcessor.class.isAssignableFrom(clazz)用于判断当前扫描到类的class是否实现了BeanPostProcessor接口，与instanceof关键字的区别如下：

//父类.class.isAssignableFrom(子类.class)
//
//子类实例 instanceof 父类类型

在bean的生命周期中，在bean初始化前调用beanPostProcessor的postProcessorBeforeInitialization方法，在bean初始化后调用beanPostProcessor的postProcessorAfterInitialization方法。

// postProcessorBeforeInitialization 初始化前方法
for (BeanPostProcessor beanPostProcessor : beanPostProcessorList) {
  instance = beanPostProcessor.postProcessBeforeInitialization(instance, beanName);
}

// 初始化逻辑，实现InitializingBean接口
if (instance instanceof InitializingBean) {
  try {
    ((InitializingBean) instance).afterPropertiesSet();
  } catch (Exception e) {
    e.printStackTrace();
  }
}

// postProcessorAfterInitialization 初始化后方法
for (BeanPostProcessor beanPostProcessor : beanPostProcessorList) {
  // instance引用可能改变，可能返回代理对象的引用
  instance = beanPostProcessor.postProcessAfterInitialization(instance, beanName);
}

参考B站视频：链接