10-注解驱动开发-Aware接口及其实现类

本文知识点来源于尚硅谷，感谢尚硅谷为广大学子提供的优质教育资源，感谢各位老师热情指导，本文仅作为学习笔记使用，记录学习心得，如有不适，请联系作者。

我们自定义组件想要使用Spring容器底层的一些组件（如：ApplicationContext，BeanFactory，xxx）只需要实现xxxAware

1. 例如：我们想要在自定义组件Yellow中获applicationContext

我们需要让Yellow类实现ApplicationContextAware接口，重写setApplicationContext方法

public class Yellow implements ApplicationContextAware{
	private ApplicationContext applicationContext;
	
	public Yellow() {
		System.out.println("Yellow....constructor.....");
	}

	@Override
	public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
		System.out.println("Yellow....执行setApplicationContext...");
		this.applicationContext = applicationContext;
	}

	public ApplicationContext getApplicationContext() {
		return applicationContext;
	}
	
}

配置类使用@Bean注入Yellow组件
测试类：

@Test
	public void test2() {
		AnnotationConfigApplicationContext applicationContext = new AnnotationConfigApplicationContext(MainConfigOfAutowired.class);
		System.out.println("容器创建完成....");
		Yellow bean = applicationContext.getBean(Yellow.class);
		System.out.println(bean.getApplicationContext() == applicationContext);
	}

输出true，表示我们自定义组件中的获取的applicationContext就是我们的容器对象：

Yellow....constructor.....
Yellow....执行setApplicationContext...
容器创建完成....
true

我们发现每一个xxxAware都有一个对应的xxxProcessor
以ApplicationContextAwareProcessor为例：
在Yellow类setApplicationContext（）方法中debug断点调试我们查看调用栈，从框选位置看起，该处调用之前内容建议查看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);
		}

首先在Bean初始化之前进入applyBeanPostProcessorsBeforeInitialization 前置处理器

	@Override
	public Object applyBeanPostProcessorsBeforeInitialization(Object existingBean, String beanName)
			throws BeansException {

		Object result = existingBean;
		for (BeanPostProcessor processor : getBeanPostProcessors()) {
			Object current = processor.postProcessBeforeInitialization(result, beanName);
			if (current == null) {
				return result;
			}
			result = current;
		}
		return result;
	}

遍历容器中的Bean

public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
		if (!(bean instanceof EnvironmentAware || bean instanceof EmbeddedValueResolverAware ||
				bean instanceof ResourceLoaderAware || bean instanceof ApplicationEventPublisherAware ||
				bean instanceof MessageSourceAware || bean instanceof ApplicationContextAware)){
			return bean;
		}

		AccessControlContext acc = null;

		if (System.getSecurityManager() != null) {
			acc = this.applicationContext.getBeanFactory().getAccessControlContext();
		}

		if (acc != null) {
			AccessController.doPrivileged((PrivilegedAction<Object>) () -> {
				invokeAwareInterfaces(bean);
				return null;
			}, acc);
		}
		else {
			invokeAwareInterfaces(bean);
		}

		return bean;
	}

首先判断有没有实现各种Aware接口，然后权限校验，然后执行invokeAwareInterfaces(bean);

	private void invokeAwareInterfaces(Object bean) {
		if (bean instanceof EnvironmentAware) {
			((EnvironmentAware) bean).setEnvironment(this.applicationContext.getEnvironment());
		}
		if (bean instanceof EmbeddedValueResolverAware) {
			((EmbeddedValueResolverAware) bean).setEmbeddedValueResolver(this.embeddedValueResolver);
		}
		if (bean instanceof ResourceLoaderAware) {
			((ResourceLoaderAware) bean).setResourceLoader(this.applicationContext);
		}
		if (bean instanceof ApplicationEventPublisherAware) {
			((ApplicationEventPublisherAware) bean).setApplicationEventPublisher(this.applicationContext);
		}
		if (bean instanceof MessageSourceAware) {
			((MessageSourceAware) bean).setMessageSource(this.applicationContext);
		}
		if (bean instanceof ApplicationContextAware) {
			((ApplicationContextAware) bean).setApplicationContext(this.applicationContext);
		}
	}

}

具体判断实现了那个Aware接口，对应的通过相应set方法。

2. 例如我们要在自定义组件中获取bean名字

在前一个案例基础上实现BeanNameAware接口重写setBeanName()方法，

public class Yellow implements ApplicationContextAware,BeanNameAware{
	private ApplicationContext applicationContext;
	
	public Yellow() {
		System.out.println("Yellow....constructor.....");
	}

	@Override
	public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
		System.out.println("Yellow....执行setApplicationContext...");
		this.applicationContext = applicationContext;
	}

	public ApplicationContext getApplicationContext() {
		return applicationContext;
	}

	@Override
	public void setBeanName(String name) {
		System.out.println("BeanName：" + name);		
	}
	
}

执行测试代码输出当前bean的名字：

Yellow....constructor.....
BeanName：yellow
Yellow....执行setApplicationContext...
容器创建完成....
true

3. 我们想在自定义组件中使用字符解析器
   实现EmbeddedValueResolverAware接口重写setEmbeddedValueResolver方法

public class Yellow implements ApplicationContextAware,BeanNameAware,EmbeddedValueResolverAware{
	private ApplicationContext applicationContext;
	
	public Yellow() {
		System.out.println("Yellow....constructor.....");
	}

	@Override
	public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
		System.out.println("Yellow....执行setApplicationContext...");
		this.applicationContext = applicationContext;
	}

	public ApplicationContext getApplicationContext() {
		return applicationContext;
	}

	@Override
	public void setBeanName(String name) {
		System.out.println("BeanName：" + name);		
	}

	@Override
	public void setEmbeddedValueResolver(StringValueResolver resolver) {
		String string = resolver.resolveStringValue("你好,${os.name},你好,#{2020+1}");
		System.out.println(string);
	}
	
}

输出中 你好,Windows 10,你好,2021被成功解析

Yellow....constructor.....
BeanName：yellow
你好,Windows 10,你好,2021
Yellow....执行setApplicationContext...
容器创建完成....
true