spring cglib实现嵌套方法拦截

使用spring 的拦截器对方法进行拦截，不管是动态代理，还是cglib, 只能拦截到被代理对象的调用方法，对于被调用方法里再调用同一对象里的其他方法就无法拦截到，就是我们说的嵌套拦截，之前文章里提及过加载器改写实现拦截(美团cat方式) , 今天试验出另外一种方法

我们要在spring初始化对象后对其用cglib加强修改，重新注入到容器当中

刚开始想在容器初始化完毕后修改bean, 利用实现ApplicationListener接口

public class InstantiationTracingBeanPostProcessor implements
ApplicationListener<ContextRefreshedEvent> {

@Override
public void onApplicationEvent(ContextRefreshedEvent event) {
//cglib 改写
}

}

参考了另外一篇文章

https://www.cnblogs.com/007sx/p/5785914.html

在做web项目开发中，尤其是企业级应用开发的时候，往往会在工程启动的时候做许多的前置检查。

比如检查是否使用了我们组禁止使用的Mysql的group_concat函数，如果使用了项目就不能启动，并指出哪个文件的xml文件使用了这个函数。

而在Spring的web项目中，我们可以介入Spring的启动过程。我们希望在Spring容器将所有的Bean都初始化完成之后，做一些操作，这个时候我们就可以实现一个接口：

复制代码
package com.yk.test.executor.processor
public class InstantiationTracingBeanPostProcessor implements ApplicationListener<ContextRefreshedEvent> {
@Override
public void onApplicationEvent(ContextRefreshedEvent event) {
//需要执行的逻辑代码，当spring容器初始化完成后就会执行该方法。
}
}
复制代码
同时在Spring的配置文件中，添加注入：

<!-- 当Spring容器启动完成后执行下面的这个Bean -->
<bean class="com.yk.test.executor.processor.InstantiationTracingBeanPostProcessor"/>
但是这个时候，会存在一个问题，在web 项目中（spring mvc），系统会存在两个容器，一个是root application context ,另一个就是我们自己的 projectName-servlet context（作为root application context的子容器）。

这种情况下，就会造成onApplicationEvent方法被执行两次。为了避免上面提到的问题，我们可以只在root application context初始化完成后调用逻辑代码，其他的容器的初始化完成，则不做任何处理，修改后代码

如下：

@Override
public void onApplicationEvent(ContextRefreshedEvent event) {
if(event.getApplicationContext().getParent() == null){//root application context 没有parent，他就是老大.
//需要执行的逻辑代码，当spring容器初始化完成后就会执行该方法。
}
}
 

其实更简单的方法是使用注解：`@PostConstruct`，只需要在需要启动的时候执行的方法上标注这个注解就搞定了。

 

例子：

 

复制代码
    /**
     * 服务启动时就执行--添加超级管理员
     */
    @PostConstruct
    public void addDefaultAdmin() {
        try {
            User user = new User();
            user.setCreateTime(new Date());
            try {
                user.setPassword(Md5.md5Encode("admin"));
            } catch (Exception e) {
                e.printStackTrace();
            }
            user.setUserName("admin");
            user.setRole(FrameConstant.USER_SUPER_ADMIN);
            userDao.save(user);
            log.debug("初始化完毕！");
        } catch (Exception e) {
            log.debug("初始化完毕！");
        }
    }
复制代码

 但是我在改写完bean后，却没有办法把bean重新注入

再换方法，在每个bean初始化完成后，cglib对其修改重新注入,通过实现BeanPostProcessor 来实现

参考文章 https://blog.csdn.net/elim168/article/details/76146351

BeanPostProcessor是Spring中定义的一个接口，其与之前介绍的InitializingBean和DisposableBean接口类似，也是供Spring进行回调的。Spring将在初始化bean前后对BeanPostProcessor实现类进行回调，与InitializingBean和DisposableBean接口不同的是BeanPostProcessor接口将对所有的bean都起作用，即所有的bean初始化前后都会回调BeanPostProcessor实现类，而InitializingBean和DisposableBean接口是针对单个bean的，即只有在对应的bean实现了InitializingBean或DisposableBean接口才会对其进行回调。

BeanPostProcessor接口的定义如下：

public interface BeanPostProcessor {

    Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException;

    Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException;

}
如你所见，BeanPostProcessor接口中定义了两个方法，其中方法postProcessBeforeInitialization()将在一个bean被完全初始化前进行回调，此时对应的bean已经实例化了，但是对应的属性注入等还没有进行，即在调用InitializingBean的afterPropertiesSet()方法或bean对应的init-method之前；而方法postProcessAfterInitialization()将在bean被完全初始化后进行回调，此时对应的依赖注入已经完成，即在调用InitializingBean的afterPropertiesSet()方法或对应init-method方法之后。两个方法的参数以及返回值对应的意义都是一样的，其中参数bean表示当前状态的bean，参数beanName表示当前bean的名称，而方法对应的返回值即表示需要放入到bean容器中的bean，所以用户如果有需要完全可以在这两个方法中对bean进行修改，即封装自己的bean进行返回。

以下是Spring源码中对bean进行初始化的逻辑，从源码中我们可以看到是先通过applyBeanPostProcessorsBeforeInitialization()方法使用注册的BeanPostProcessor的postProcessBeforeInitialization()方法依次回调，然后是通过invokeInitMethods()方法依次调用当前bean对应的初始化方法，再通过applyBeanPostProcessorsAfterInitialization方法使用注册的BeanPostProcessor的postProcessorAfterInitialization()方法依次进行回调。

    protected Object initializeBean(final String beanName, final Object bean, RootBeanDefinition mbd) {
        if (System.getSecurityManager() != null) {
            AccessController.doPrivileged(new PrivilegedAction<Object>() {
                @Override
                public Object run() {
                    invokeAwareMethods(beanName, bean);
                    return null;
                }
            }, getAccessControlContext());
        }
        else {
            invokeAwareMethods(beanName, bean);
        }

        Object wrappedBean = bean;
        if (mbd == null || !mbd.isSynthetic()) {
            wrappedBean = applyBeanPostProcessorsBeforeInitialization(wrappedBean, beanName);
        }

        try {
            invokeInitMethods(beanName, wrappedBean, mbd);
        }
        catch (Throwable ex) {
            throw new BeanCreationException(
                    (mbd != null ? mbd.getResourceDescription() : null),
                    beanName, "Invocation of init method failed", ex);
        }

        if (mbd == null || !mbd.isSynthetic()) {
            wrappedBean = applyBeanPostProcessorsAfterInitialization(wrappedBean, beanName);
        }
        return wrappedBean;
    }
11.2 注册

BeanPostProcessor的注册是非常简单的，我们只需要把它当做一个普通的bean定义到Spring的bean容器中，Spring将能够自动检测到它，并将它注册到当前的bean容器中。BeanPostProcessor是容器绑定的，即BeanPostProcessor只能对跟它属于同一个bean容器中的bean进行回调，即BeanPostProcessor不能对属于它父容器或子容器中的bean进行回调。

在bean容器中定义了BeanPostProcessor之后，Spring将最先将BeanPostProcessor对应的bean进行实例化，如果我们制定BeanPostProcessor的lazy-initialization=”true”或default-lazy-initialization=”true”，Spring将对其进行忽略，即这些配置对BeanPostProcessor不起作用。这也很好理解，因为只有这样之后在实例化其它bean的时候这些BeanPostProcessor才能派上用场。鉴于这种机制，所以这里有一个问题需要注意，Spring在初始化bean的时候将优先初始化depends-on属性指定的bean，所以当我们的BeanPostProcessor通过depends-on指定了对其它bean的依赖时，其它bean是不会被BeanPostProcessor所回调的，当然这里也包括简介的depends-on对应的bean。此外，在BeanPostProcessor实例化后需要直接或间接的进行注入的bean也由于实例化时间提前不会被BeanPostProcessor回调。还有就是BeanPostProcessor之间不会进行回调，即BeanPostProcessorA不会在BeanPostProcessorB初始化时对其进行回调。

BeanPostProcessor在Spring内部也是用的比较多的，尤其是AOP代理部分。包括用户需要自己实现BeanPostProcessor实现代理功能时也需要注意BeanPostProcessor直接或间接关联的bean是不会被回调的，即不会被代理成功的。

11.3 示例

接下来看一个简单的定制自己的BeanPostProcessor的示例，在示例中我们将仅仅简单的实现一个BeanPostProcessor，在postProcessBeforeInitialization()和postProcessAfterInitialization()方法中都直接返回对应的bean，然后在postProcessBeforeInitialization()方法中简单的打印一下对应的bean名称。

public class HelloBeanPostProcessor implements BeanPostProcessor {

    public Object postProcessBeforeInitialization(Object bean, String beanName)
            throws BeansException {
        System.out.println("beanName-----------" + beanName);
        return bean;
    }

    public Object postProcessAfterInitialization(Object bean, String beanName)
            throws BeansException {
        return bean;
    }

}
实现了BeanPostProcessor之后就可以将其定义到bean容器中，其定义方式跟普通bean的定义方式是一样的。

<bean class="com.app.HelloBeanPostProcessor"/>
11.4 回调顺序

在bean容器中我们可以同时定义多个BeanPostProcessor，这样在新实例化一个bean后将依次使用每个BeanPostProcessor回调一遍，当然，如果某一个BeanPostProcessor回调后的返回的bean为null，则不再继续往下回调，将直接返回null，这个时候bean容器中对应beanName对应的bean也是null。当在一个bean容器中同时定义有多个BeanPostProcessor时，默认将根据BeanPostProcessor在bean容器中定义的先后顺序对新实例化的bean进行回调。还有一种定义BeanPostProcessor回调顺序的方法是将我们自定义的BeanPostProcessor实现类同时实现Ordered接口，然后Spring将根据Ordered接口定义的getOrder()方法的返回值来决定BeanPostProcessor回调的先后顺序，getOrder()返回值越小的越先进行回调。此外，实现了Ordered接口的BeanPostProcessor总是比没有实现Ordered接口的BeanPostProcessor先进行回调，为了便于管理，推荐要么都实现Ordered接口，要么都不实现。

以下是一个实现了Ordered接口，并把getOrder()方法的返回值作为一个参数进行配置的示例。

public class HelloBeanPostProcessor implements BeanPostProcessor, Ordered {

    private int order;
    
    public Object postProcessBeforeInitialization(Object bean, String beanName)
            throws BeansException {
        System.out.println("beanName-----------" + beanName);
        return bean;
    }

    public Object postProcessAfterInitialization(Object bean, String beanName)
            throws BeansException {
        return bean;
    }

    public void setOrder(int order) {
        this.order = order;
    }
    
    public int getOrder() {
        return order;
    }

}
之后就可以在配置的时候通过参数order来指定我们的getOrder()方法的返回值。

<bean class="com.app.HelloBeanPostProcessor" p:order="3"/>

public class TraceBeanPostProcessor implements BeanPostProcessor {

    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName)
            throws BeansException {
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName)
            throws BeansException {
        
        System.out.println("#################post bean:"+bean);
　　　　　//对Service注解过的方法实现修改 
        Service service = (Service) bean.getClass().getAnnotation(Service.class);
        if(service!=null){
            Class<?> cls = bean.getClass();
            System.out.println("post @@@@@@@bean"+cls);
            Object newbean = TraceCGLibUtil.createBean(cls);
            return newbean;
            
        }
        return bean;
    }

}

PS: 如果Service注解过的类被其他拦截器加强处理过，这里无法无法通过

Service service = (Service) bean.getClass().getAnnotation(Service.class);
去获取class了，service==null