1818
1919import io .microsphere .logging .Logger ;
2020import io .microsphere .spring .boot .env .PropertySourceLoaders ;
21+ import io .microsphere .spring .context .ConfigurableApplicationContextInitializer ;
2122import io .microsphere .spring .context .event .BeanFactoryListenerAdapter ;
2223import org .springframework .beans .factory .config .ConfigurableListableBeanFactory ;
23- import org .springframework .beans .factory .support .BeanDefinitionRegistry ;
2424import org .springframework .boot .env .OriginTrackedMapPropertySource ;
2525import org .springframework .boot .origin .Origin ;
2626import org .springframework .boot .origin .OriginLookup ;
3838
3939import static io .microsphere .collection .MapUtils .newLinkedHashMap ;
4040import static io .microsphere .logging .LoggerFactory .getLogger ;
41- import static io .microsphere .spring .beans .factory .support .BeanRegistrar .registerBean ;
4241import static org .springframework .boot .origin .OriginTrackedValue .of ;
4342
4443/**
4847 * @see ConfigurableEnvironment
4948 * @since ApplicationContextInitializer
5049 */
51- public class OriginTrackedConfigurationPropertyInitializer implements BeanFactoryListenerAdapter , ApplicationContextInitializer <ConfigurableApplicationContext > {
52-
53- public static final String BEAN_NAME = "originTrackedConfigurationPropertyInitializer" ;
50+ public class OriginTrackedConfigurationPropertyInitializer extends ConfigurableApplicationContextInitializer implements BeanFactoryListenerAdapter {
5451
5552 private static final Logger logger = getLogger (OriginTrackedConfigurationPropertyInitializer .class );
5653
5754 private ConfigurableApplicationContext applicationContext ;
5855
5956 private PropertySourceLoaders propertySourceLoaders ;
6057
58+
6159 /**
6260 * Initializes the application context by storing references and registering this instance
6361 * as a bean in the bean factory. This override sets up the {@link PropertySourceLoaders}
@@ -70,15 +68,13 @@ public class OriginTrackedConfigurationPropertyInitializer implements BeanFactor
7068 * initializer.initialize(applicationContext);
7169 * }</pre>
7270 *
73- * @param applicationContext the {@link ConfigurableApplicationContext} to initialize
71+ * @param context the {@link ConfigurableApplicationContext} to initialize
72+ * @param environment the {@link ConfigurableEnvironment}
7473 */
7574 @ Override
76- public void initialize (ConfigurableApplicationContext applicationContext ) {
77- this .applicationContext = applicationContext ;
78- this .propertySourceLoaders = new PropertySourceLoaders (applicationContext .getClassLoader ());
79- ConfigurableListableBeanFactory beanFactory = applicationContext .getBeanFactory ();
80- BeanDefinitionRegistry registry = (BeanDefinitionRegistry ) beanFactory ;
81- registerBean (registry , BEAN_NAME , this );
75+ protected void initialize (ConfigurableApplicationContext context , ConfigurableEnvironment environment ) {
76+ this .applicationContext = context ;
77+ this .propertySourceLoaders = new PropertySourceLoaders (context .getClassLoader ());
8278 }
8379
8480 /**
0 commit comments