【Spring源码分析】Spring Scope功能中的动态代理
Bluの妙妙天地 2024-08-07 08:09:12 阅读 90
本文基于Springboot 3.3.2及Springcloud 2023.0.1版本编写。
Spring Scoped Proxy是什么
在使用Spring cloud配置中心动态配置更新功能时,笔者发现在给一个类加上<code>@RefreshScope注解后,其中@Value
注入的字段会被自动更新。起初笔者以为Spring在收到配置更新事件后会自动设置该bean的字段值,但测试后发现配置更新是通过重建整个bean的方式来实现的。实验代码如下:
@RestController
public class TestController {
@Autowired
private RefreshClazz refreshClazz;
@GetMapping("/test/url")
public String getCount() {
return "count=" + refreshClazz.getCount();
}
@Component
@RefreshScope
public static class RefreshClazz {
@Value("${example.config}")code>
private String configStr;
private int count = 0;
@PostConstruct
public void postConstruct() {
System.out.println("POST_CONSTRUCT");
}
@PreDestroy
public void preDestroy() {
System.out.println("PRE_DESTROY");
}
public int getCount() {
return count++;
}
}
}
代码中,RefreshClazz 是一个被标记了@RefreshScope
的 Bean,通过@Autowired
的方式注入到 Controller 中。运行上面的代码,会发现当配置更新后,RefreshClazz 的内部字段 count 被重置到了 0,同时也会输出 PRE_DESTROY 和 POST_CONSTRUCT,说明旧的 Bean 被删除,新的 Bean 被创建了。
那么问题来了,RefreshClazz
是被静态注入到 controller 中的,如何做到自动刷新的呢?原理便是注入的是动态代理对象。Spring 在实现诸多功能(如@Lazy
、@Transactional
、@Cacheable
)时都用到了动态代理,Scope 也是其中一个。Scope 功能用到的动态代理被称为 Scoped Proxy。
如何使用 Scoped Proxy
@RefreshScope
定义代码:
@Target({ ElementType.TYPE, ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
@Scope("refresh")
@Documented
public @interface RefreshScope {
@AliasFor(annotation = Scope.class)
ScopedProxyMode proxyMode() default ScopedProxyMode.TARGET_CLASS;
}
根据@RefreshScope
的定义我们可以发现,它等同于@Scope(value = "refresh", proxyMode = ScopedProxyMode.TARGET_CLASS)
。其关键在于proxyMode = ScopedProxyMode.TARGET_CLASS
。这个参数一共有四个取值:
DEFAULT
:默认值,等同于NO
;NO
:不创建scoped proxy,对于非单例Scope的bean来说此模式通常没用;INTERFACES
:使用JDK动态代理创建一个基于接口实现的动态代理对象;TARGET_CLASS
:使用CGLIB创建一个基于继承的动态代理对象。可见,
@RefreshScope
默认使用了基于继承实现的动态代理对象。这样做有几点好处:- 在使用方注入时既可使用接口注入,也可以使用类型注入。如果
proxyMode = ScopedProxyMode.INTERFACES
,创建出的 scoped proxy 类型并非原Bean的类型,而只是实现了它所实现的接口。由于@Autowired
真正要注入的是 scoped proxy,如果变量定义为 Bean 的类型,Spring 会报No qualifying bean of type '...' available
错误; - JDK实现的动态代理在性能和内存开销上稍大于CGLIB动态代理。
Scoped Proxy 是如何被创建的
Spring Bean 注册
在Spring容器中注册 scoped proxy 的逻辑来自 ScopedProxyUtils#createScopedProxy
方法,该方法的代码如下:
public static BeanDefinitionHolder createScopedProxy(BeanDefinitionHolder definition,
BeanDefinitionRegistry registry, boolean proxyTargetClass) {
// 获取原beanName和bean定义
String originalBeanName = definition.getBeanName();
BeanDefinition targetDefinition = definition.getBeanDefinition();
// 生成被代理bean的beanName
String targetBeanName = getTargetBeanName(originalBeanName);
// 创建动态代理Bean的BeanDefinition
RootBeanDefinition proxyDefinition = new RootBeanDefinition(ScopedProxyFactoryBean.class);
proxyDefinition.setDecoratedDefinition(new BeanDefinitionHolder(targetDefinition, targetBeanName));
proxyDefinition.setOriginatingBeanDefinition(targetDefinition);
proxyDefinition.setSource(definition.getSource());
proxyDefinition.setRole(targetDefinition.getRole());
// 将被代理bean的beanName传给动态代理Bean
proxyDefinition.getPropertyValues().add("targetBeanName", targetBeanName);
if (proxyTargetClass) {
targetDefinition.setAttribute(AutoProxyUtils.PRESERVE_TARGET_CLASS_ATTRIBUTE, Boolean.TRUE);
// ScopedProxyFactoryBean's "proxyTargetClass" default is TRUE, so we don't need to set it explicitly here.
}
else {
proxyDefinition.getPropertyValues().add("proxyTargetClass", Boolean.FALSE);
}
// 将原bean的属性复制到动态代理bean定义中.
proxyDefinition.setAutowireCandidate(targetDefinition.isAutowireCandidate());
proxyDefinition.setPrimary(targetDefinition.isPrimary());
if (targetDefinition instanceof AbstractBeanDefinition abd) {
proxyDefinition.copyQualifiersFrom(abd);
}
// 将底层bean隐藏起来,不参与注入.
targetDefinition.setAutowireCandidate(false);
targetDefinition.setPrimary(false);
// 将底层bean的beanName设置为targetBeanName,注册到容器中.
registry.registerBeanDefinition(targetBeanName, targetDefinition);
// 返回刚生成的动态代理bean的BeanDefinition,beanName设置为原bean的名字.
return new BeanDefinitionHolder(proxyDefinition, originalBeanName, definition.getAliases());
}
在进入这个方法之前,这个被@Scope
修饰的 Bean 和其他普通单例 Bean 并没有区别。但此方法对它进行了一通魔改,最后将原 Bean 的定义改了个名字藏在了 Spring 容器内部,而暴露出了一个新生成的 scoped proxy bean。因为新的 bean 名字和原 bean 名一样,并且可能是 Primary Bean,因此在 @Autowired
注入时默认就注入了这个动态代理 bean。
原 Bean 的名字被改成了什么呢?可以参考 getTargetBeanName()
方法:
private static final String TARGET_NAME_PREFIX = "scopedTarget.";
public static String getTargetBeanName(String originalBeanName) {
return TARGET_NAME_PREFIX + originalBeanName;
}
因此,底层 Bean 的 beanName 为 scopedTarget.<originalBeanName>
。
动态代理对象生成
在 ScopedProxyUtils#createScopedProxy
方法的代码中,我们注意到新生成的动态代理 Bean 类被设置为了 ScopedProxyFactoryBean.class
。这是一个 FactoryBean
,负责具体生成动态代理对象。代码如下:
public class ScopedProxyFactoryBean extends ProxyConfig
implements FactoryBean<Object>, BeanFactoryAware, AopInfrastructureBean {
/** 从Spring容器中获取底层对象的TargetSource. */
private final SimpleBeanTargetSource scopedTargetSource = new SimpleBeanTargetSource();
/** 底层bean的beanName. */
@Nullable
private String targetBeanName;
/** 缓存的单例 Scoped proxy. */
@Nullable
private Object proxy;
/** 构造方法. */
public ScopedProxyFactoryBean() {
setProxyTargetClass(true);
}
/** 设置底层bean的beanName. */
public void setTargetBeanName(String targetBeanName) {
this.targetBeanName = targetBeanName;
this.scopedTargetSource.setTargetBeanName(targetBeanName);
}
/** 创建动态代理对象的主方法. */
@Override
public void setBeanFactory(BeanFactory beanFactory) {
if (!(beanFactory instanceof ConfigurableBeanFactory cbf)) {
throw new IllegalStateException("Not running in a ConfigurableBeanFactory: " + beanFactory);
}
// 为targetSource设置使用的beanFactory
this.scopedTargetSource.setBeanFactory(beanFactory);
// 通过ProxyFactory创建动态代理对象
ProxyFactory pf = new ProxyFactory();
pf.copyFrom(this);
// 使用配置好的targetSource
pf.setTargetSource(this.scopedTargetSource);
Assert.notNull(this.targetBeanName, "Property 'targetBeanName' is required");
Class<?> beanType = beanFactory.getType(this.targetBeanName);
if (beanType == null) {
throw new IllegalStateException("Cannot create scoped proxy for bean '" + this.targetBeanName +
"': Target type could not be determined at the time of proxy creation.");
}
if (!isProxyTargetClass() || beanType.isInterface() || Modifier.isPrivate(beanType.getModifiers())) {
pf.setInterfaces(ClassUtils.getAllInterfacesForClass(beanType, cbf.getBeanClassLoader()));
}
// Add an introduction that implements only the methods on ScopedObject.
ScopedObject scopedObject = new DefaultScopedObject(cbf, this.scopedTargetSource.getTargetBeanName());
pf.addAdvice(new DelegatingIntroductionInterceptor(scopedObject));
// Add the AopInfrastructureBean marker to indicate that the scoped proxy
// itself is not subject to auto-proxying! Only its target bean is.
pf.addInterface(AopInfrastructureBean.class);
// 生成并缓存动态代理对象
this.proxy = pf.getProxy(cbf.getBeanClassLoader());
}
/** 工厂类的获取生成对象方法,获取生成的动态代理对象. */
@Override
@Nullable
public Object getObject() {
if (this.proxy == null) {
throw new FactoryBeanNotInitializedException();
}
return this.proxy;
}
}
这个类使用了 ProxyFactory
创建动态代理对象,它生成的动态代理对象通过接口 TargetSource
来获取代理的底层对象。上面的 ScopedProxyFactoryBean
使用了 SimpleBeanTargetSource
,它的代码如下:
public class SimpleBeanTargetSource extends AbstractBeanFactoryBasedTargetSource {
@Override
public Object getTarget() throws Exception {
return getBeanFactory().getBean(getTargetBeanName());
}
}
逻辑很清晰,通过 beanFactory
来获取名为 targetBeanName
的bean对象作为被代理的对象。而这个 targetBeanName
在Spring容器中注册 scoped proxy 的时候就被生成了,设置的逻辑是:
proxyDefinition.getPropertyValues().add("targetBeanName", targetBeanName);
Scoped Proxy 整体工作逻辑
看了前面的代码,我们可以总结出 scoped proxy 的工作逻辑:
- 被
@Autowired
注入的是一个 scoped proxy,它的 BeanDefinition 的 scope 其实是 singleton; - 调用 bean 方法时,scoped proxy 在 Spring 容器中获取名为
scopedTarget.<originalBeanName>
的被代理 bean,此 bean 的 scope 是 refresh; - scoped proxy 调用被代理 bean 的对应方法。
Scoped Bean 的生命周期
八股文里,Spring 有五种 Scope(singleton、prototype、request、session、globalSession)(这其实是过时的,最新文档里是六种:singleton、prototype、request、session、application、websocket)。那本文里的 refresh
scope 是什么呢?
其实,只需要注入一个实现了 Scope
接口的 Bean,用户便可添加一个自定义的 scope。本文中的 refresh
scope 就是 spring-cloud-context
中定义的。Spring 容器在获取任何不是 singleton 或 prototype 的 bean 时,会先找出该 scope 名所对应的 Scope 对象,再使用 Scope#get
从该对象中获取 bean,代码参考 AbstractBeanFactory#doGetBean
。而 scoped bean 的生命周期便是由具体的 scope 类管理了(实现案例可以参考 GenericScope
)。
声明
本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。