Spring之@EnableAspectJAutoProxy开启AOP功能原理

Posted 2021-10-31 敲代码的小小酥

疑问

在使用AOP功能时，我们需要用@EnableAspectJAutoProxy手动开启AOP功能，否则不生效，为什么写了@EnableAspectJAutoProxy注解就可以使用AOP功能了呢，其中的原理是什么呢？

答案

在Spring中，无论是AOP功能还是其他需要手动开启的功能，实现思路大致如下:
开启功能的注解中，@Import了一个类，这个类都是BeanDefinitionPostprocessor类型的对象，在这个对象中，把开启相关功能的入口类，加入到了Spring容器中，所以就具有了该功能。而没有开启注解的话，相关功能的入口类就没有进入Spring容器，Spring容器也就没有相关功能的判断和处理。

原理解析

我们以AOP的@EnableAspectJAutoProxy为例，分析上述回答的原理。

首先，@EnableAspectJAutoProxy import进了AspectJAutoProxyRegistrar类，我们看这个类:

class AspectJAutoProxyRegistrar implements ImportBeanDefinitionRegistrar {

	/**
	 * Register, escalate, and configure the AspectJ auto proxy creator based on the value
	 * of the @{@link EnableAspectJAutoProxy#proxyTargetClass()} attribute on the importing
	 * {@code @Configuration} class.
	 */
	@Override
	public void registerBeanDefinitions(
			AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {

		AopConfigUtils.registerAspectJAnnotationAutoProxyCreatorIfNecessary(registry);

		AnnotationAttributes enableAspectJAutoProxy =
				AnnotationConfigUtils.attributesFor(importingClassMetadata, EnableAspectJAutoProxy.class);
		if (enableAspectJAutoProxy != null) {
			if (enableAspectJAutoProxy.getBoolean("proxyTargetClass")) {
				AopConfigUtils.forceAutoProxyCreatorToUseClassProxying(registry);
			}
			if (enableAspectJAutoProxy.getBoolean("exposeProxy")) {
				AopConfigUtils.forceAutoProxyCreatorToExposeProxy(registry);
			}
		}
	}

}

可以看到这个类是一个BeanDefinitionRegistrar类，看其registerBeanDefinitions方法，看注册了什么beandefition进来。

看registerAspectJAnnotationAutoProxyCreatorIfNecessary方法，一顿点入方法后，到了如下方法:

private static BeanDefinition registerOrEscalateApcAsRequired(
			Class<?> cls, BeanDefinitionRegistry registry, @Nullable Object source) {

		Assert.notNull(registry, "BeanDefinitionRegistry must not be null");

		if (registry.containsBeanDefinition(AUTO_PROXY_CREATOR_BEAN_NAME)) {
			BeanDefinition apcDefinition = registry.getBeanDefinition(AUTO_PROXY_CREATOR_BEAN_NAME);
			if (!cls.getName().equals(apcDefinition.getBeanClassName())) {
				int currentPriority = findPriorityForClass(apcDefinition.getBeanClassName());
				int requiredPriority = findPriorityForClass(cls);
				if (currentPriority < requiredPriority) {
					apcDefinition.setBeanClassName(cls.getName());
				}
			}
			return null;
		}

		RootBeanDefinition beanDefinition = new RootBeanDefinition(cls);
		beanDefinition.setSource(source);
		beanDefinition.getPropertyValues().add("order", Ordered.HIGHEST_PRECEDENCE);
		beanDefinition.setRole(BeanDefinition.ROLE_INFRASTRUCTURE);
		registry.registerBeanDefinition(AUTO_PROXY_CREATOR_BEAN_NAME, beanDefinition);
		return beanDefinition;
	}

这个方法重点看往register中注册了什么，即如下方法:

通过断点可以看到:

注册了AnnotationAwareAspectJAutoProxyCreator类，我们看这个类的结构图:

所以，这里注册了AOP的入口类，而AOP的入口类又是一个BeanpostProcessor对象，这样，就有了AOP的功能，并听过BeanPostProcessor对相关的bean进行了代理操作。
这就是开启AOP功能的原理。