Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add BeanFactoryInitializer callback before preInstantiateSingletons

Browse Source 
Closes gh-32836
This commit is contained in:
Juergen Hoeller
2024-06-04 22:50:42 +02:00
parent f10caf6aa6
commit 28eb9aebcf
4 changed files with 56 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
spring-test/src/main/java/org/springframework/test/context/support/DynamicPropertySourceBeanInitializer.java
View File

@@ -19,55 +19,30 @@ package org.springframework.test.context.support;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.BeanFactory;
import org.springframework.beans.factory.BeanFactoryAware;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.BeanFactoryInitializer;
import org.springframework.beans.factory.ListableBeanFactory;
import org.springframework.context.weaving.LoadTimeWeaverAware;
import org.springframework.instrument.classloading.LoadTimeWeaver;
import org.springframework.lang.Nullable;
import org.springframework.test.context.DynamicPropertySource;
/**
* Internal component which eagerly initializes beans created by {@code @Bean}
* factory methods annotated with {@link DynamicPropertySource @DynamicPropertySource}.
*
* <p>This class implements {@link LoadTimeWeaverAware} since doing so is
* currently the only way to have a component eagerly initialized before the
* {@code ConfigurableListableBeanFactory.preInstantiateSingletons()} phase.
*
* @author Sam Brannen
* @since 6.2
*/
class DynamicPropertySourceBeanInitializer implements BeanFactoryAware, InitializingBean, LoadTimeWeaverAware {
class DynamicPropertySourceBeanInitializer implements BeanFactoryInitializer<ListableBeanFactory> {
private static final Log logger = LogFactory.getLog(DynamicPropertySourceBeanInitializer.class);
@Nullable
private BeanFactory beanFactory;
@Override
public void setBeanFactory(BeanFactory beanFactory) {
this.beanFactory = beanFactory;
}
@Override
public void afterPropertiesSet() {
if (!(this.beanFactory instanceof ListableBeanFactory lbf)) {
throw new IllegalStateException("BeanFactory must be set and must be a ListableBeanFactory");
}
for (String name : lbf.getBeanNamesForAnnotation(DynamicPropertySource.class)) {
public void initialize(ListableBeanFactory beanFactory) {
for (String name : beanFactory.getBeanNamesForAnnotation(DynamicPropertySource.class)) {
if (logger.isDebugEnabled()) {
logger.debug("Eagerly initializing @DynamicPropertySource bean '%s'".formatted(name));
}
this.beanFactory.getBean(name);
beanFactory.getBean(name);
}
}
@Override
public void setLoadTimeWeaver(LoadTimeWeaver loadTimeWeaver) {
// no-op
}
}