Improve subclassing support in ConfClassPostPro

philwebb · philwebb · commit 985ede92f3fe · 2013-05-02T10:24:18.000-07:00
Update ConfigurationClassPostProcessor with an additional
afterLoadConfiguration method that can be overridden by subclasses.

Also add a protected getMetadataReaderFactory() method allowing
subclass access to the MetadataReaderFactory.
diff --git a/spring-context/src/main/java/org/springframework/context/annotation/ConfigurationClassPostProcessor.java b/spring-context/src/main/java/org/springframework/context/annotation/ConfigurationClassPostProcessor.java
@@ -317,6 +317,7 @@ public void processConfigBeanDefinitions(BeanDefinitionRegistry registry) {
 			if (!ConditionalAnnotationHelper.shouldSkip(configurationClass, registry,
 					this.environment, this.importBeanNameGenerator)) {
 				reader.loadBeanDefinitionsForConfigurationClass(configurationClass);
+				afterLoadConfiguration(registry, configurationClass.getBeanName(), configurationClass.getMetadata());
 			}
 		}
 
@@ -332,6 +333,19 @@ public void processConfigBeanDefinitions(BeanDefinitionRegistry registry) {
 		}
 	}
 
+	/**
+	 * Called immediately after a {@code @Configuration} bean has been loaded.  This
+	 * method will be called for all configuration beans, including {@code @Imported}
+	 * configurations.  Subclasses can perform additional processing.
+	 * @param registry the registry
+	 * @param beanName the name of the configuration bean
+	 * @param metadata the configuration meta-data
+	 * @since 4.0
+	 */
+	protected void afterLoadConfiguration(BeanDefinitionRegistry registry,
+			String beanName, AnnotationMetadata metadata) {
+	}
+
 	/**
 	 * Post-processes a BeanFactory in search of Configuration class BeanDefinitions;
 	 * any candidates are then enhanced by a {@link ConfigurationClassEnhancer}.
@@ -375,6 +389,11 @@ public void enhanceConfigurationClasses(ConfigurableListableBeanFactory beanFact
 	}
 
 
+	protected final MetadataReaderFactory getMetadataReaderFactory() {
+		return metadataReaderFactory;
+	}
+
+
 	private static class ImportAwareBeanPostProcessor implements PriorityOrdered, BeanFactoryAware, BeanPostProcessor {
 
 		private BeanFactory beanFactory;
