Commit f5aeac36 authored by Phillip Webb's avatar Phillip Webb

Polish

parent 9fb9a67c
...@@ -134,25 +134,21 @@ public class EmbeddedServletContainerAutoConfiguration { ...@@ -134,25 +134,21 @@ public class EmbeddedServletContainerAutoConfiguration {
if (this.beanFactory == null) { if (this.beanFactory == null) {
return; return;
} }
if (ObjectUtils.isEmpty(this.beanFactory.getBeanNamesForType( registerSyntheticBeanIfMissing(registry,
EmbeddedServletContainerCustomizerBeanPostProcessor.class, true, "embeddedServletContainerCustomizerBeanPostProcessor",
false))) { EmbeddedServletContainerCustomizerBeanPostProcessor.class);
RootBeanDefinition beanDefinition = new RootBeanDefinition( registerSyntheticBeanIfMissing(registry,
EmbeddedServletContainerCustomizerBeanPostProcessor.class); "errorPageRegistrarBeanPostProcessor",
beanDefinition.setSynthetic(true); ErrorPageRegistrarBeanPostProcessor.class);
registry.registerBeanDefinition( }
"embeddedServletContainerCustomizerBeanPostProcessor",
beanDefinition);
} private void registerSyntheticBeanIfMissing(BeanDefinitionRegistry registry,
if (ObjectUtils.isEmpty(this.beanFactory.getBeanNamesForType( String name, Class<?> beanClass) {
ErrorPageRegistrarBeanPostProcessor.class, true, false))) { if (ObjectUtils.isEmpty(
RootBeanDefinition beanDefinition = new RootBeanDefinition( this.beanFactory.getBeanNamesForType(beanClass, true, false))) {
ErrorPageRegistrarBeanPostProcessor.class); RootBeanDefinition beanDefinition = new RootBeanDefinition(beanClass);
beanDefinition.setSynthetic(true); beanDefinition.setSynthetic(true);
registry.registerBeanDefinition("errorPageRegistrarBeanPostProcessor", registry.registerBeanDefinition(name, beanDefinition);
beanDefinition);
} }
} }
......
...@@ -27,6 +27,7 @@ import org.springframework.beans.factory.BeanFactoryAware; ...@@ -27,6 +27,7 @@ import org.springframework.beans.factory.BeanFactoryAware;
import org.springframework.beans.factory.ListableBeanFactory; import org.springframework.beans.factory.ListableBeanFactory;
import org.springframework.beans.factory.config.BeanPostProcessor; import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.core.annotation.AnnotationAwareOrderComparator; import org.springframework.core.annotation.AnnotationAwareOrderComparator;
import org.springframework.util.Assert;
/** /**
* {@link BeanPostProcessor} that applies all {@link EmbeddedServletContainerCustomizer}s * {@link BeanPostProcessor} that applies all {@link EmbeddedServletContainerCustomizer}s
...@@ -45,6 +46,9 @@ public class EmbeddedServletContainerCustomizerBeanPostProcessor ...@@ -45,6 +46,9 @@ public class EmbeddedServletContainerCustomizerBeanPostProcessor
@Override @Override
public void setBeanFactory(BeanFactory beanFactory) { public void setBeanFactory(BeanFactory beanFactory) {
Assert.isInstanceOf(ListableBeanFactory.class, beanFactory,
"EmbeddedServletContainerCustomizerBeanPostProcessor can only be used "
+ "with a ListableBeanFactory");
this.beanFactory = (ListableBeanFactory) beanFactory; this.beanFactory = (ListableBeanFactory) beanFactory;
} }
......
...@@ -27,6 +27,7 @@ import org.springframework.beans.factory.BeanFactoryAware; ...@@ -27,6 +27,7 @@ import org.springframework.beans.factory.BeanFactoryAware;
import org.springframework.beans.factory.ListableBeanFactory; import org.springframework.beans.factory.ListableBeanFactory;
import org.springframework.beans.factory.config.BeanPostProcessor; import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.core.annotation.AnnotationAwareOrderComparator; import org.springframework.core.annotation.AnnotationAwareOrderComparator;
import org.springframework.util.Assert;
/** /**
* {@link BeanPostProcessor} that applies all {@link ErrorPageRegistrar}s from the bean * {@link BeanPostProcessor} that applies all {@link ErrorPageRegistrar}s from the bean
...@@ -45,6 +46,9 @@ public class ErrorPageRegistrarBeanPostProcessor ...@@ -45,6 +46,9 @@ public class ErrorPageRegistrarBeanPostProcessor
@Override @Override
public void setBeanFactory(BeanFactory beanFactory) { public void setBeanFactory(BeanFactory beanFactory) {
Assert.isInstanceOf(ListableBeanFactory.class, beanFactory,
"ErrorPageRegistrarBeanPostProcessor can only be used "
+ "with a ListableBeanFactory");
this.beanFactory = (ListableBeanFactory) beanFactory; this.beanFactory = (ListableBeanFactory) beanFactory;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment