From 1864d7907710bfad4cdcbd77763a8c1a886af06d Mon Sep 17 00:00:00 2001 From: Andy Wilkinson Date: Tue, 18 Nov 2014 21:00:28 +0000 Subject: [PATCH] Polish Undertow contribution Closes gh-1779 --- .gitignore | 3 +- ...ddedServletContainerAutoConfiguration.java | 4 +- .../autoconfigure/web/ServerProperties.java | 121 +++--- .../web/MultipartAutoConfigurationTests.java | 3 +- ...erverPropertiesAutoConfigurationTests.java | 10 +- spring-boot-dependencies/pom.xml | 17 +- .../src/main/asciidoc/cloud-deployment.adoc | 4 +- spring-boot-docs/src/main/asciidoc/howto.adoc | 55 +++ .../main/asciidoc/spring-boot-features.adoc | 21 +- .../src/main/asciidoc/using-spring-boot.adoc | 3 + .../SampleUndertowSslApplication.java | 2 +- .../undertow/service/HelloWorldService.java | 2 +- .../sample/undertow/web/SampleController.java | 2 +- .../SampleUndertowSslApplicationTests.java | 2 +- .../undertow/SampleUndertowApplication.java | 2 +- .../undertow/service/HelloWorldService.java | 2 +- .../sample/undertow/web/SampleController.java | 2 +- .../spring-boot-starter-undertow/pom.xml | 14 + .../undertow/UndertowBuilderCustomizer.java | 35 ++ .../UndertowEmbeddedServletContainer.java | 58 ++- ...dertowEmbeddedServletContainerFactory.java | 369 ++++++++++++------ .../embedded/undertow/package-info.java | 23 ++ ...edServletContainerMvcIntegrationTests.java | 8 +- ...wEmbeddedServletContainerFactoryTests.java | 106 +++-- 24 files changed, 604 insertions(+), 264 deletions(-) create mode 100644 spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowBuilderCustomizer.java create mode 100644 spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/package-info.java diff --git a/.gitignore b/.gitignore index b4c28f3b0e..58de8c7240 100644 --- a/.gitignore +++ b/.gitignore @@ -28,5 +28,4 @@ overridedb.* *.ipr *.iws .idea -*.jar -.DS_Store +*.jar \ No newline at end of file diff --git a/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/web/EmbeddedServletContainerAutoConfiguration.java b/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/web/EmbeddedServletContainerAutoConfiguration.java index 3461b0a94f..925bf1ca4b 100644 --- a/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/web/EmbeddedServletContainerAutoConfiguration.java +++ b/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/web/EmbeddedServletContainerAutoConfiguration.java @@ -54,6 +54,7 @@ import org.springframework.util.ObjectUtils; * * @author Phillip Webb * @author Dave Syer + * @author Ivan Sopov */ @Order(Ordered.HIGHEST_PRECEDENCE) @Configuration @@ -90,7 +91,7 @@ public class EmbeddedServletContainerAutoConfiguration { } } - + /** * Nested configuration if Undertow is being used. */ @@ -105,7 +106,6 @@ public class EmbeddedServletContainerAutoConfiguration { } } - /** * Registers a {@link EmbeddedServletContainerCustomizerBeanPostProcessor}. Registered diff --git a/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/web/ServerProperties.java b/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/web/ServerProperties.java index e913718638..3f4f3afe12 100644 --- a/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/web/ServerProperties.java +++ b/spring-boot-autoconfigure/src/main/java/org/springframework/boot/autoconfigure/web/ServerProperties.java @@ -53,6 +53,7 @@ import org.springframework.util.StringUtils; * @author Dave Syer * @author Stephane Nicoll * @author Andy Wilkinson + * @author Ivan Sopov */ @ConfigurationProperties(prefix = "server", ignoreUnknownFields = false) public class ServerProperties implements EmbeddedServletContainerCustomizer { @@ -72,7 +73,7 @@ public class ServerProperties implements EmbeddedServletContainerCustomizer { private String servletPath = "/"; private final Tomcat tomcat = new Tomcat(); - + private final Undertow undertow = new Undertow(); private final Map contextParameters = new HashMap(); @@ -220,62 +221,6 @@ public class ServerProperties implements EmbeddedServletContainerCustomizer { return prefix + path; } - public static class Undertow { - private Integer bufferSize; - private Integer buffersPerRegion; - private Integer ioThreads; - private Integer workerThreads; - private Boolean directBuffers; - - public Integer getBufferSize() { - return this.bufferSize; - } - - public void setBufferSize(Integer bufferSize) { - this.bufferSize = bufferSize; - } - - public Integer getBuffersPerRegion() { - return this.buffersPerRegion; - } - - public void setBuffersPerRegion(Integer buffersPerRegion) { - this.buffersPerRegion = buffersPerRegion; - } - - public Integer getIoThreads() { - return this.ioThreads; - } - - public void setIoThreads(Integer ioThreads) { - this.ioThreads = ioThreads; - } - - public Integer getWorkerThreads() { - return this.workerThreads; - } - - public void setWorkerThreads(Integer workerThreads) { - this.workerThreads = workerThreads; - } - - public Boolean getDirectBuffers() { - return this.directBuffers; - } - - public void setDirectBuffers(Boolean directBuffers) { - this.directBuffers = directBuffers; - } - - void customizeUndertow(UndertowEmbeddedServletContainerFactory factory) { - factory.setBufferSize(bufferSize); - factory.setBuffersPerRegion(buffersPerRegion); - factory.setIoThreads(ioThreads); - factory.setWorkerThreads(workerThreads); - factory.setDirectBuffers(directBuffers); - } - } - public static class Tomcat { private String accessLogPattern; @@ -462,4 +407,66 @@ public class ServerProperties implements EmbeddedServletContainerCustomizer { } + public static class Undertow { + + private Integer bufferSize; + + private Integer buffersPerRegion; + + private Integer ioThreads; + + private Integer workerThreads; + + private Boolean directBuffers; + + public Integer getBufferSize() { + return this.bufferSize; + } + + public void setBufferSize(Integer bufferSize) { + this.bufferSize = bufferSize; + } + + public Integer getBuffersPerRegion() { + return this.buffersPerRegion; + } + + public void setBuffersPerRegion(Integer buffersPerRegion) { + this.buffersPerRegion = buffersPerRegion; + } + + public Integer getIoThreads() { + return this.ioThreads; + } + + public void setIoThreads(Integer ioThreads) { + this.ioThreads = ioThreads; + } + + public Integer getWorkerThreads() { + return this.workerThreads; + } + + public void setWorkerThreads(Integer workerThreads) { + this.workerThreads = workerThreads; + } + + public Boolean getDirectBuffers() { + return this.directBuffers; + } + + public void setDirectBuffers(Boolean directBuffers) { + this.directBuffers = directBuffers; + } + + void customizeUndertow(UndertowEmbeddedServletContainerFactory factory) { + factory.setBufferSize(this.bufferSize); + factory.setBuffersPerRegion(this.buffersPerRegion); + factory.setIoThreads(this.ioThreads); + factory.setWorkerThreads(this.workerThreads); + factory.setDirectBuffers(this.directBuffers); + } + + } + } diff --git a/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/web/MultipartAutoConfigurationTests.java b/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/web/MultipartAutoConfigurationTests.java index bec1f66bca..0c93d3796a 100644 --- a/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/web/MultipartAutoConfigurationTests.java +++ b/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/web/MultipartAutoConfigurationTests.java @@ -16,9 +16,7 @@ package org.springframework.boot.autoconfigure.web; -import java.io.IOException; import java.net.URI; -import java.net.URISyntaxException; import javax.servlet.MultipartConfigElement; @@ -65,6 +63,7 @@ import static org.mockito.Mockito.mock; * @author Greg Turnquist * @author Dave Syer * @author Josh Long + * @author Ivan Sopov */ public class MultipartAutoConfigurationTests { diff --git a/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/web/ServerPropertiesAutoConfigurationTests.java b/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/web/ServerPropertiesAutoConfigurationTests.java index cf684455ce..9437009728 100644 --- a/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/web/ServerPropertiesAutoConfigurationTests.java +++ b/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/web/ServerPropertiesAutoConfigurationTests.java @@ -47,6 +47,7 @@ import static org.mockito.Mockito.verify; * Tests for {@link ServerPropertiesAutoConfiguration}. * * @author Dave Syer + * @author Ivan Sopov */ public class ServerPropertiesAutoConfigurationTests { @@ -112,8 +113,7 @@ public class ServerPropertiesAutoConfigurationTests { // factory should take precedence... assertEquals(3000, containerFactory.getPort()); } - - + @Test public void customizeWithUndertowContainerFactory() throws Exception { this.context = new AnnotationConfigEmbeddedWebApplicationContext(); @@ -127,7 +127,6 @@ public class ServerPropertiesAutoConfigurationTests { assertNotNull(server); assertEquals(3000, containerFactory.getPort()); } - @Test public void customizeTomcatWithCustomizer() throws Exception { @@ -186,7 +185,7 @@ public class ServerPropertiesAutoConfigurationTests { } } - + @Configuration protected static class CustomUndertowContainerConfig { @@ -203,9 +202,6 @@ public class ServerPropertiesAutoConfigurationTests { } } - - - @Configuration protected static class CustomizeConfig { diff --git a/spring-boot-dependencies/pom.xml b/spring-boot-dependencies/pom.xml index bdeef0e508..3154aab7ac 100644 --- a/spring-boot-dependencies/pom.xml +++ b/spring-boot-dependencies/pom.xml @@ -127,10 +127,10 @@ 1.2.7 1.3 8.0.15 + 1.1.0.Final 1.7 2.0 1.6.3 - 1.1.0.Final 3.0.2 @@ -540,6 +540,16 @@ metrics-servlets ${dropwizard-metrics.version} + + io.undertow + undertow-core + ${undertow.version} + + + io.undertow + undertow-servlet + ${undertow.version} + javax.cache cache-api @@ -707,11 +717,6 @@ tomcat-jsp-api ${tomcat.version} - - io.undertow - undertow-servlet - ${undertow.version} - org.apache.velocity velocity diff --git a/spring-boot-docs/src/main/asciidoc/cloud-deployment.adoc b/spring-boot-docs/src/main/asciidoc/cloud-deployment.adoc index c9919472a0..5f66b71646 100644 --- a/spring-boot-docs/src/main/asciidoc/cloud-deployment.adoc +++ b/spring-boot-docs/src/main/asciidoc/cloud-deployment.adoc @@ -158,8 +158,8 @@ for our starter REST application: Spring Boot makes `-D` arguments available as properties accessible from a Spring `Environment` instance. The `server.port` configuration property is fed to the embedded -Tomcat or Jetty instance which then uses it when it starts up. The `$PORT` environment -variable is assigned to us by the Heroku PaaS. +Tomcat, Jetty or Undertow instance which then uses it when it starts up. The `$PORT` +environment variable is assigned to us by the Heroku PaaS. Heroku by default will use Java 1.6. This is fine as long as your Maven or Gradle build is set to use the same version (Maven users can use the `java.version` property). If you diff --git a/spring-boot-docs/src/main/asciidoc/howto.adoc b/spring-boot-docs/src/main/asciidoc/howto.adoc index 56f3ce125d..49fe253acc 100644 --- a/spring-boot-docs/src/main/asciidoc/howto.adoc +++ b/spring-boot-docs/src/main/asciidoc/howto.adoc @@ -556,6 +556,61 @@ of ways. Or the nuclear option is to add your own `JettyEmbeddedServletContainer +[[howto-use-undertow-instead-of-tomcat]] +=== Use Undertow instead of Tomcat +Using Undertow instead of Tomcat is very similar to <>. You need to exclude the Tomcat dependencies and include +the Undertow starter instead. + +Example in Maven: + +[source,xml,indent=0,subs="verbatim,quotes,attributes"] +---- + + org.springframework.boot + spring-boot-starter-web + + + org.springframework.boot + spring-boot-starter-tomcat + + + + + org.springframework.boot + spring-boot-starter-undertow + +---- + +Example in Gradle: + +[source,groovy,indent=0,subs="verbatim,quotes,attributes"] +---- + configurations { + compile.exclude module: "spring-boot-starter-tomcat" + } + + dependencies { + compile("org.springframework.boot:spring-boot-starter-web:{spring-boot-version}") + compile("org.springframework.boot:spring-boot-starter-undertow:{spring-boot-version}") + // ... + } +---- + + +[[howto-configure-undertow]] +=== Configure Undertow +Generally you can follow the advice from +_<>_ about +`@ConfigurationProperties` (`ServerProperties` and `ServerProperties.Undertow are the +main ones here), but also look at +`EmbeddedServletContainerCustomizer`. Once you have access to the +`UndertowEmbeddedServletContainerFactory` you can use an `UndertowBuilderCustomizer` to +modify Undertow's configuration to meet your needs. Or the nuclear option is to add your +own `UndertowEmbeddedServletContainerFactory`. + + + [[howto-use-tomcat-8]] === Use Tomcat 8 Tomcat 8 works with Spring Boot, but the default is to use Tomcat 7 (so we can support diff --git a/spring-boot-docs/src/main/asciidoc/spring-boot-features.adoc b/spring-boot-docs/src/main/asciidoc/spring-boot-features.adoc index dc873c8e43..9ad68ac2e1 100644 --- a/spring-boot-docs/src/main/asciidoc/spring-boot-features.adoc +++ b/spring-boot-docs/src/main/asciidoc/spring-boot-features.adoc @@ -810,8 +810,8 @@ possible. [[boot-features-developing-web-applications]] == Developing web applications Spring Boot is well suited for web application development. You can easily create a -self-contained HTTP server using embedded Tomcat or Jetty. Most web applications will -use the `spring-boot-starter-web` module to get up and running quickly. +self-contained HTTP server using embedded Tomcat, Jetty, or Undertow. Most web +applications will use the `spring-boot-starter-web` module to get up and running quickly. If you haven't yet developed a Spring Boot web application you can follow the "Hello World!" example in the @@ -1093,9 +1093,9 @@ asks for them to be scanned in its `Filter` registration). [[boot-features-embedded-container]] === Embedded servlet container support -Spring Boot includes support for embedded Tomcat and Jetty servers. Most developers will -simply use the appropriate '`Starter POM`' to obtain a fully configured instance. By -default both Tomcat and Jetty will listen for HTTP requests on port `8080`. +Spring Boot includes support for embedded Tomcat, Jetty, and Undertow servers. Most +developers will simply use the appropriate '`Starter POM`' to obtain a fully configured +instance. By default the embedded server will listen for HTTP requests on port `8080`. @@ -1121,8 +1121,9 @@ interface. Under the hood Spring Boot uses a new type of `ApplicationContext` for embedded servlet container support. The `EmbeddedWebApplicationContext` is a special type of `WebApplicationContext` that bootstraps itself by searching for a single -`EmbeddedServletContainerFactory` bean. Usually a `TomcatEmbeddedServletContainerFactory` -or `JettyEmbeddedServletContainerFactory` will have been auto-configured. +`EmbeddedServletContainerFactory` bean. Usually a `TomcatEmbeddedServletContainerFactory`, +`JettyEmbeddedServletContainerFactory`, or `UndertowEmbeddedServletContainerFactory` will +have been auto-configured. NOTE: You usually won't need to be aware of these implementation classes. Most applications will be auto-configured and the appropriate `ApplicationContext` and @@ -1176,8 +1177,8 @@ methods. [[boot-features-customizing-configurableembeddedservletcontainerfactory-directly]] ===== Customizing ConfigurableEmbeddedServletContainer directly If the above customization techniques are too limited, you can register the -`TomcatEmbeddedServletContainerFactory` or `JettyEmbeddedServletContainerFactory` bean -yourself. +`TomcatEmbeddedServletContainerFactory`, `JettyEmbeddedServletContainerFactory` or +`UndertowEmbeddedServletContainerFactory` bean yourself. [source,java,indent=0] ---- @@ -1208,6 +1209,8 @@ packaged as an executable archive), there are some limitations in the JSP suppor * Jetty does not currently work as an embedded container with JSPs. +* Undertow does not support JSPs + There is a {github-code}/spring-boot-samples/spring-boot-sample-web-jsp[JSP sample] so you can see how to set things up. diff --git a/spring-boot-docs/src/main/asciidoc/using-spring-boot.adoc b/spring-boot-docs/src/main/asciidoc/using-spring-boot.adoc index 0ef7446862..64f62a23c8 100644 --- a/spring-boot-docs/src/main/asciidoc/using-spring-boot.adoc +++ b/spring-boot-docs/src/main/asciidoc/using-spring-boot.adoc @@ -348,6 +348,9 @@ swap specific technical facets. |`spring-boot-starter-tomcat` |Import Spring Boot's default HTTP engine (Tomcat). + +|`spring-boot-starter-undertow` +|Imports the Undertow HTTP engine (to be used as an alternative to Tomcat) |=== TIP: For a list of additional community contributed starter POMs, see the diff --git a/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/SampleUndertowSslApplication.java b/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/SampleUndertowSslApplication.java index dbeac870e5..dbf58c9406 100644 --- a/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/SampleUndertowSslApplication.java +++ b/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/SampleUndertowSslApplication.java @@ -1,5 +1,5 @@ /* - * Copyright 2012-2013 the original author or authors. + * Copyright 2012-2014 the original author or authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. diff --git a/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/service/HelloWorldService.java b/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/service/HelloWorldService.java index 0865d25d2d..5e5a3a4342 100644 --- a/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/service/HelloWorldService.java +++ b/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/service/HelloWorldService.java @@ -1,5 +1,5 @@ /* - * Copyright 2012-2013 the original author or authors. + * Copyright 2012-2014 the original author or authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. diff --git a/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/web/SampleController.java b/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/web/SampleController.java index 78b5c0c776..3084fbd6e3 100644 --- a/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/web/SampleController.java +++ b/spring-boot-samples/spring-boot-sample-undertow-ssl/src/main/java/sample/undertow/web/SampleController.java @@ -1,5 +1,5 @@ /* - * Copyright 2012-2013 the original author or authors. + * Copyright 2012-2014 the original author or authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. diff --git a/spring-boot-samples/spring-boot-sample-undertow-ssl/src/test/java/sample/undertow/SampleUndertowSslApplicationTests.java b/spring-boot-samples/spring-boot-sample-undertow-ssl/src/test/java/sample/undertow/SampleUndertowSslApplicationTests.java index 3b450065d3..3d5a12a07e 100644 --- a/spring-boot-samples/spring-boot-sample-undertow-ssl/src/test/java/sample/undertow/SampleUndertowSslApplicationTests.java +++ b/spring-boot-samples/spring-boot-sample-undertow-ssl/src/test/java/sample/undertow/SampleUndertowSslApplicationTests.java @@ -39,7 +39,7 @@ import static org.junit.Assert.assertEquals; /** * Basic integration tests for demo application. * - * @author Dave Syer + * @author Ivan Sopov */ @RunWith(SpringJUnit4ClassRunner.class) @SpringApplicationConfiguration(classes = SampleUndertowSslApplication.class) diff --git a/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/SampleUndertowApplication.java b/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/SampleUndertowApplication.java index a81adcd109..c01a048f26 100644 --- a/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/SampleUndertowApplication.java +++ b/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/SampleUndertowApplication.java @@ -1,5 +1,5 @@ /* - * Copyright 2012-2013 the original author or authors. + * Copyright 2012-2014 the original author or authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. diff --git a/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/service/HelloWorldService.java b/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/service/HelloWorldService.java index 0865d25d2d..5e5a3a4342 100644 --- a/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/service/HelloWorldService.java +++ b/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/service/HelloWorldService.java @@ -1,5 +1,5 @@ /* - * Copyright 2012-2013 the original author or authors. + * Copyright 2012-2014 the original author or authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. diff --git a/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/web/SampleController.java b/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/web/SampleController.java index 78b5c0c776..3084fbd6e3 100644 --- a/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/web/SampleController.java +++ b/spring-boot-samples/spring-boot-sample-undertow/src/main/java/sample/undertow/web/SampleController.java @@ -1,5 +1,5 @@ /* - * Copyright 2012-2013 the original author or authors. + * Copyright 2012-2014 the original author or authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. diff --git a/spring-boot-starters/spring-boot-starter-undertow/pom.xml b/spring-boot-starters/spring-boot-starter-undertow/pom.xml index 0bcb12fc6b..72f49cc8e1 100644 --- a/spring-boot-starters/spring-boot-starter-undertow/pom.xml +++ b/spring-boot-starters/spring-boot-starter-undertow/pom.xml @@ -19,9 +19,23 @@ ${basedir}/../.. + + io.undertow + undertow-core + io.undertow undertow-servlet + + + org.jboss.spec.javax.servlet + jboss-servlet-api_3.1_spec + + + + + javax.servlet + javax.servlet-api diff --git a/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowBuilderCustomizer.java b/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowBuilderCustomizer.java new file mode 100644 index 0000000000..75436d2b08 --- /dev/null +++ b/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowBuilderCustomizer.java @@ -0,0 +1,35 @@ +/* + * Copyright 2012-2014 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.springframework.boot.context.embedded.undertow; + +import io.undertow.Undertow.Builder; + +/** + * Callback interface that can be used to customize an Undertow {@link Builder}. + * + * @author Andy Wilkinson + * @since 1.2.0 + * @see UndertowEmbeddedServletContainerFactory + */ +public interface UndertowBuilderCustomizer { + + /** + * @param builder the {@code Builder} to customize + */ + void customize(Builder builder); + +} diff --git a/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainer.java b/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainer.java index 43fb3912f5..dfe4b99f13 100644 --- a/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainer.java +++ b/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainer.java @@ -1,3 +1,19 @@ +/* + * Copyright 2012-2014 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + package org.springframework.boot.context.embedded.undertow; import io.undertow.Handlers; @@ -14,16 +30,29 @@ import org.springframework.boot.context.embedded.EmbeddedServletContainerExcepti import org.springframework.util.StringUtils; /** + * {@link EmbeddedServletContainer} that can be used to control an embedded Undertow + * server. Typically this class should be created using + * {@link UndertowEmbeddedServletContainerFactory} and not directly. + * * @author Ivan Sopov + * @author Andy Wilkinson + * @since 1.2.0 + * @see UndertowEmbeddedServletContainer */ public class UndertowEmbeddedServletContainer implements EmbeddedServletContainer { private final DeploymentManager manager; + private final Builder builder; + private final String contextPath; + private final int port; + private final boolean autoStart; + private Undertow undertow; + private boolean started = false; public UndertowEmbeddedServletContainer(Builder builder, DeploymentManager manager, @@ -40,38 +69,39 @@ public class UndertowEmbeddedServletContainer implements EmbeddedServletContaine if (!this.autoStart) { return; } - if (undertow == null) { + if (this.undertow == null) { try { - HttpHandler servletHandler = manager.start(); - if (StringUtils.isEmpty(contextPath)) { - builder.setHandler(servletHandler); + HttpHandler servletHandler = this.manager.start(); + if (StringUtils.isEmpty(this.contextPath)) { + this.builder.setHandler(servletHandler); } else { - PathHandler pathHandler = Handlers.path().addPrefixPath(contextPath, - servletHandler); - builder.setHandler(pathHandler); + PathHandler pathHandler = Handlers.path().addPrefixPath( + this.contextPath, servletHandler); + this.builder.setHandler(pathHandler); } - undertow = builder.build(); + this.undertow = this.builder.build(); } catch (ServletException ex) { throw new EmbeddedServletContainerException( "Unable to start embdedded Undertow", ex); } } - undertow.start(); - started = true; + this.undertow.start(); + this.started = true; } @Override public synchronized void stop() throws EmbeddedServletContainerException { - if (started) { - started = false; - undertow.stop(); + if (this.started) { + this.started = false; + this.undertow.stop(); } } @Override public int getPort() { - return port; + return this.port; } + } \ No newline at end of file diff --git a/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainerFactory.java b/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainerFactory.java index ad112a6566..d3b471a994 100644 --- a/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainerFactory.java +++ b/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainerFactory.java @@ -1,12 +1,21 @@ +/* + * Copyright 2012-2014 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + package org.springframework.boot.context.embedded.undertow; -import static io.undertow.servlet.Servlets.defaultContainer; -import static io.undertow.servlet.Servlets.deployment; -import static io.undertow.servlet.Servlets.servlet; -import static org.xnio.Options.SSL_CLIENT_AUTH_MODE; -import static org.xnio.SslClientAuthMode.NOT_REQUESTED; -import static org.xnio.SslClientAuthMode.REQUESTED; -import static org.xnio.SslClientAuthMode.REQUIRED; import io.undertow.Undertow; import io.undertow.Undertow.Builder; import io.undertow.UndertowMessages; @@ -27,12 +36,15 @@ import io.undertow.servlet.handlers.DefaultServlet; import io.undertow.servlet.util.ImmediateInstanceHandle; import java.io.File; -import java.io.FileInputStream; import java.io.IOException; import java.net.URL; import java.security.KeyManagementException; import java.security.KeyStore; import java.security.NoSuchAlgorithmException; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.Collection; +import java.util.List; import javax.net.ssl.KeyManager; import javax.net.ssl.KeyManagerFactory; @@ -45,6 +57,7 @@ import javax.servlet.ServletException; import org.springframework.boot.context.embedded.AbstractEmbeddedServletContainerFactory; import org.springframework.boot.context.embedded.EmbeddedServletContainer; +import org.springframework.boot.context.embedded.EmbeddedServletContainerFactory; import org.springframework.boot.context.embedded.ErrorPage; import org.springframework.boot.context.embedded.MimeMappings.Mapping; import org.springframework.boot.context.embedded.ServletContextInitializer; @@ -52,176 +65,178 @@ import org.springframework.boot.context.embedded.Ssl; import org.springframework.boot.context.embedded.Ssl.ClientAuth; import org.springframework.context.ResourceLoaderAware; import org.springframework.core.io.ResourceLoader; +import org.springframework.util.Assert; import org.springframework.util.ResourceUtils; import org.springframework.util.SocketUtils; +import static io.undertow.servlet.Servlets.defaultContainer; +import static io.undertow.servlet.Servlets.deployment; +import static io.undertow.servlet.Servlets.servlet; +import static org.xnio.Options.SSL_CLIENT_AUTH_MODE; +import static org.xnio.SslClientAuthMode.NOT_REQUESTED; +import static org.xnio.SslClientAuthMode.REQUESTED; +import static org.xnio.SslClientAuthMode.REQUIRED; + /** + * {@link EmbeddedServletContainerFactory} that can be used to create + * {@link UndertowEmbeddedServletContainer}s. + *

+ * Unless explicitly configured otherwise, the factory will create containers that listen + * for HTTP requests on port 8080. + * * @author Ivan Sopov + * @author Andy Wilkinson + * @since 1.2.0 + * @see UndertowEmbeddedServletContainer */ public class UndertowEmbeddedServletContainerFactory extends AbstractEmbeddedServletContainerFactory implements ResourceLoaderAware { + private List undertowBuilderCustomizers = new ArrayList(); + private ResourceLoader resourceLoader; private Integer bufferSize; + private Integer buffersPerRegion; + private Integer ioThreads; + private Integer workerThreads; + private Boolean directBuffers; /** - * Create a new - * {@link org.springframework.boot.context.embedded.undertow.UndertowEmbeddedServletContainerFactory} - * instance. + * Create a new {@link UndertowEmbeddedServletContainerFactory} instance. */ public UndertowEmbeddedServletContainerFactory() { super(); + setRegisterJspServlet(false); } /** - * Create a new - * {@link org.springframework.boot.context.embedded.undertow.UndertowEmbeddedServletContainerFactory} - * that listens for requests using the specified port. - * + * Create a new {@link UndertowEmbeddedServletContainerFactory} that listens for + * requests using the specified port. * @param port the port to listen on */ public UndertowEmbeddedServletContainerFactory(int port) { super(port); + setRegisterJspServlet(false); } /** - * Create a new - * {@link org.springframework.boot.context.embedded.undertow.UndertowEmbeddedServletContainerFactory} - * with the specified context path and port. - * + * Create a new {@link UndertowEmbeddedServletContainerFactory} with the specified + * context path and port. * @param contextPath root the context path * @param port the port to listen on */ public UndertowEmbeddedServletContainerFactory(String contextPath, int port) { super(contextPath, port); + setRegisterJspServlet(false); + } + + /** + * Set {@link UndertowBuilderCustomizer}s that should be applied to the Undertow + * {@link Builder}. Calling this method will replace any existing customizers. + * @param undertowBuilderCustomizers the customizers to set + */ + public void setUndertowBuilderCustomizers( + Collection undertowBuilderCustomizers) { + Assert.notNull(undertowBuilderCustomizers, + "undertowBuilderCustomizers must not be null"); + this.undertowBuilderCustomizers = new ArrayList( + undertowBuilderCustomizers); + } + + /** + * Returns a mutable collection of the {@link UndertowBuilderCustomizer}s that will be + * applied to the Undertow {@link Builder} . + * @return the customizers that will be applied + */ + public Collection getUndertowBuilderCustomizers() { + return this.undertowBuilderCustomizers; + } + + /** + * Add {@link UndertowBuilderCustomizer}s that should be used to customize the + * Undertow {@link Builder}. + * @param undertowBuilderCustomizers the customizers to add + */ + public void addUndertowBuilderCustomizers( + UndertowBuilderCustomizer... undertowBuilderCustomizers) { + Assert.notNull(undertowBuilderCustomizers, + "undertowBuilderCustomizers must not be null"); + this.undertowBuilderCustomizers.addAll(Arrays.asList(undertowBuilderCustomizers)); } @Override public EmbeddedServletContainer getEmbeddedServletContainer( ServletContextInitializer... initializers) { - DeploymentInfo servletBuilder = deployment(); - - servletBuilder.addListener(new ListenerInfo( - UndertowSpringServletContextListener.class, - new UndertowSpringServletContextListenerFactory( - new UndertowSpringServletContextListener( - mergeInitializers(initializers))))); + DeploymentManager manager = createDeploymentManager(initializers); - if (resourceLoader != null) { - servletBuilder.setClassLoader(resourceLoader.getClassLoader()); + int port = getPort(); + if (port == 0) { + port = SocketUtils.findAvailableTcpPort(40000); } - else { - servletBuilder.setClassLoader(getClass().getClassLoader()); - } - servletBuilder.setContextPath(getContextPath()); - servletBuilder.setDeploymentName("spring-boot"); - if (isRegisterDefaultServlet()) { - servletBuilder.addServlet(servlet("default", DefaultServlet.class)); - } - if (isRegisterJspServlet()) { - logger.error("JSPs are not supported with Undertow"); - } - for (ErrorPage springErrorPage : getErrorPages()) { - if (springErrorPage.getStatus() != null) { - io.undertow.servlet.api.ErrorPage undertowErrorpage = new io.undertow.servlet.api.ErrorPage( - springErrorPage.getPath(), springErrorPage.getStatusCode()); - servletBuilder.addErrorPage(undertowErrorpage); - } - else if (springErrorPage.getException() != null) { - io.undertow.servlet.api.ErrorPage undertowErrorpage = new io.undertow.servlet.api.ErrorPage( - springErrorPage.getPath(), springErrorPage.getException()); - servletBuilder.addErrorPage(undertowErrorpage); - } - else { - io.undertow.servlet.api.ErrorPage undertowErrorpage = new io.undertow.servlet.api.ErrorPage( - springErrorPage.getPath()); - servletBuilder.addErrorPage(undertowErrorpage); - } - } - servletBuilder.setServletStackTraces(ServletStackTraces.NONE); - File root = getValidDocumentRoot(); - if (root != null && root.isDirectory()) { - servletBuilder.setResourceManager(new FileResourceManager(root, 0)); - } - else if (root != null && root.isFile()) { - servletBuilder.setResourceManager(new JarResourcemanager(root)); - } - else if (resourceLoader != null) { - servletBuilder.setResourceManager(new ClassPathResourceManager(resourceLoader - .getClassLoader(), "")); - } - else { - servletBuilder.setResourceManager(new ClassPathResourceManager(getClass() - .getClassLoader(), "")); - } - for (Mapping mimeMapping : getMimeMappings()) { - servletBuilder.addMimeMapping(new MimeMapping(mimeMapping.getExtension(), - mimeMapping.getMimeType())); - } + Builder builder = createBuilder(port); - DeploymentManager manager = defaultContainer().addDeployment(servletBuilder); - - manager.deploy(); - - manager.getDeployment().getSessionManager() - .setDefaultSessionTimeout(getSessionTimeout()); + return new UndertowEmbeddedServletContainer(builder, manager, getContextPath(), + port, port >= 0); + } + private Builder createBuilder(int port) { Builder builder = Undertow.builder(); - if (bufferSize != null) { - builder.setBufferSize(bufferSize); + if (this.bufferSize != null) { + builder.setBufferSize(this.bufferSize); } - if (buffersPerRegion != null) { - builder.setBuffersPerRegion(buffersPerRegion); + if (this.buffersPerRegion != null) { + builder.setBuffersPerRegion(this.buffersPerRegion); } - if (ioThreads != null) { - builder.setIoThreads(ioThreads); + if (this.ioThreads != null) { + builder.setIoThreads(this.ioThreads); } - if (workerThreads != null) { - builder.setWorkerThreads(workerThreads); + if (this.workerThreads != null) { + builder.setWorkerThreads(this.workerThreads); } - if (directBuffers != null) { - builder.setDirectBuffers(directBuffers); + if (this.directBuffers != null) { + builder.setDirectBuffers(this.directBuffers); } - int realPort = getPort(); - if (realPort == 0) { - realPort = SocketUtils.findAvailableTcpPort(40000); - } if (getSsl() == null) { - builder.addHttpListener(realPort, "0.0.0.0"); + builder.addHttpListener(port, "0.0.0.0"); } else { - try { - Ssl ssl = getSsl(); - SSLContext sslContext = SSLContext.getInstance(ssl.getProtocol()); - sslContext.init(getKeyManagers(), getTrustManagers(), null); - builder.addHttpsListener(realPort, "0.0.0.0", sslContext); - if (ssl.getClientAuth() == ClientAuth.NEED) { - builder.setSocketOption(SSL_CLIENT_AUTH_MODE, REQUIRED); - } - else if (ssl.getClientAuth() == ClientAuth.WANT) { - builder.setSocketOption(SSL_CLIENT_AUTH_MODE, REQUESTED); - } - else { - builder.setSocketOption(SSL_CLIENT_AUTH_MODE, NOT_REQUESTED); - } + configureSsl(port, builder); + } + for (UndertowBuilderCustomizer customizer : this.undertowBuilderCustomizers) { + customizer.customize(builder); + } + return builder; + } + + private void configureSsl(int port, Builder builder) { + try { + Ssl ssl = getSsl(); + SSLContext sslContext = SSLContext.getInstance(ssl.getProtocol()); + sslContext.init(getKeyManagers(), getTrustManagers(), null); + builder.addHttpsListener(port, "0.0.0.0", sslContext); + if (ssl.getClientAuth() == ClientAuth.NEED) { + builder.setSocketOption(SSL_CLIENT_AUTH_MODE, REQUIRED); } - catch (NoSuchAlgorithmException e) { - throw new RuntimeException(e); + else if (ssl.getClientAuth() == ClientAuth.WANT) { + builder.setSocketOption(SSL_CLIENT_AUTH_MODE, REQUESTED); } - catch (KeyManagementException e) { - throw new RuntimeException(e); + else { + builder.setSocketOption(SSL_CLIENT_AUTH_MODE, NOT_REQUESTED); } } - return new UndertowEmbeddedServletContainer(builder, manager, getContextPath(), - realPort, realPort > 0); - + catch (NoSuchAlgorithmException ex) { + throw new RuntimeException(ex); + } + catch (KeyManagementException ex) { + throw new RuntimeException(ex); + } } private KeyManager[] getKeyManagers() { @@ -244,8 +259,8 @@ public class UndertowEmbeddedServletContainerFactory extends keyManagerFactory.init(keyStore, keyPassword); return keyManagerFactory.getKeyManagers(); } - catch (Exception e) { - throw new RuntimeException(e); + catch (Exception ex) { + throw new RuntimeException(ex); } } @@ -271,10 +286,100 @@ public class UndertowEmbeddedServletContainerFactory extends trustManagerFactory.init(trustedKeyStore); return trustManagerFactory.getTrustManagers(); } - catch (Exception e) { - throw new RuntimeException(e); + catch (Exception ex) { + throw new RuntimeException(ex); + } + } + + private DeploymentManager createDeploymentManager( + ServletContextInitializer... initializers) { + DeploymentInfo servletBuilder = deployment(); + + servletBuilder.addListener(new ListenerInfo( + UndertowSpringServletContextListener.class, + new UndertowSpringServletContextListenerFactory( + new UndertowSpringServletContextListener( + mergeInitializers(initializers))))); + + if (this.resourceLoader != null) { + servletBuilder.setClassLoader(this.resourceLoader.getClassLoader()); + } + else { + servletBuilder.setClassLoader(getClass().getClassLoader()); + } + servletBuilder.setContextPath(getContextPath()); + servletBuilder.setDeploymentName("spring-boot"); + if (isRegisterDefaultServlet()) { + servletBuilder.addServlet(servlet("default", DefaultServlet.class)); + } + + configureErrorPages(servletBuilder); + servletBuilder.setServletStackTraces(ServletStackTraces.NONE); + + File root = getValidDocumentRoot(); + if (root != null && root.isDirectory()) { + servletBuilder.setResourceManager(new FileResourceManager(root, 0)); + } + else if (root != null && root.isFile()) { + servletBuilder.setResourceManager(new JarResourcemanager(root)); + } + else if (this.resourceLoader != null) { + servletBuilder.setResourceManager(new ClassPathResourceManager( + this.resourceLoader.getClassLoader(), "")); + } + else { + servletBuilder.setResourceManager(new ClassPathResourceManager(getClass() + .getClassLoader(), "")); + } + for (Mapping mimeMapping : getMimeMappings()) { + servletBuilder.addMimeMapping(new MimeMapping(mimeMapping.getExtension(), + mimeMapping.getMimeType())); } + DeploymentManager manager = defaultContainer().addDeployment(servletBuilder); + + manager.deploy(); + + manager.getDeployment().getSessionManager() + .setDefaultSessionTimeout(getSessionTimeout()); + return manager; + } + + private void configureErrorPages(DeploymentInfo servletBuilder) { + for (ErrorPage errorPage : getErrorPages()) { + if (errorPage.getStatus() != null) { + io.undertow.servlet.api.ErrorPage undertowErrorpage = new io.undertow.servlet.api.ErrorPage( + errorPage.getPath(), errorPage.getStatusCode()); + servletBuilder.addErrorPage(undertowErrorpage); + } + else if (errorPage.getException() != null) { + io.undertow.servlet.api.ErrorPage undertowErrorpage = new io.undertow.servlet.api.ErrorPage( + errorPage.getPath(), errorPage.getException()); + servletBuilder.addErrorPage(undertowErrorpage); + } + else { + io.undertow.servlet.api.ErrorPage undertowErrorpage = new io.undertow.servlet.api.ErrorPage( + errorPage.getPath()); + servletBuilder.addErrorPage(undertowErrorpage); + } + } + } + + /** + * Factory method called to create the {@link UndertowEmbeddedServletContainer}. + * Subclasses can override this method to return a different + * {@link UndertowEmbeddedServletContainer} or apply additional processing to the + * {@link Builder} and {@link DeploymentManager} used to bootstrap Undertow + * + * @param builder the builder + * @param manager the deployment manager + * @param port the port that Undertow should listen on + * @return a new {@link UndertowEmbeddedServletContainer} instance + */ + protected UndertowEmbeddedServletContainer getUndertowEmbeddedServletContainer( + Builder builder, DeploymentManager manager, int port) { + return new UndertowEmbeddedServletContainer(builder, manager, getContextPath(), + port, port >= 0); } @Override @@ -302,6 +407,12 @@ public class UndertowEmbeddedServletContainerFactory extends this.directBuffers = directBuffers; } + @Override + public void setRegisterJspServlet(boolean registerJspServlet) { + Assert.isTrue(!registerJspServlet, "Undertow does not support JSPs"); + super.setRegisterJspServlet(registerJspServlet); + } + private static class JarResourcemanager implements ResourceManager { private final String jarPath; @@ -320,7 +431,7 @@ public class UndertowEmbeddedServletContainerFactory extends @Override public Resource getResource(String path) throws IOException { - URL url = new URL("jar:file:" + jarPath + "!" + path); + URL url = new URL("jar:file:" + this.jarPath + "!" + path); URLResource resource = new URLResource(url, url.openConnection(), path); if (resource.getContentLength() < 0) { return null; @@ -361,7 +472,7 @@ public class UndertowEmbeddedServletContainerFactory extends public InstanceHandle createInstance() throws InstantiationException { return new ImmediateInstanceHandle( - listener); + this.listener); } } @@ -376,14 +487,14 @@ public class UndertowEmbeddedServletContainerFactory extends } @Override - public void contextInitialized(ServletContextEvent sce) { + public void contextInitialized(ServletContextEvent event) { try { - for (ServletContextInitializer initializer : initializers) { - initializer.onStartup(sce.getServletContext()); + for (ServletContextInitializer initializer : this.initializers) { + initializer.onStartup(event.getServletContext()); } } - catch (ServletException e) { - throw new RuntimeException(e); + catch (ServletException ex) { + throw new RuntimeException(ex); } } diff --git a/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/package-info.java b/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/package-info.java new file mode 100644 index 0000000000..c0092c5f73 --- /dev/null +++ b/spring-boot/src/main/java/org/springframework/boot/context/embedded/undertow/package-info.java @@ -0,0 +1,23 @@ +/* + * Copyright 2012-2014 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +/** + * Support for Undertow {@link org.springframework.boot.context.embedded.EmbeddedServletContainer EmbeddedServletContainers}. + * + * @see org.springframework.boot.context.embedded.undertow.UndertowEmbeddedServletContainerFactory + */ +package org.springframework.boot.context.embedded.undertow; + diff --git a/spring-boot/src/test/java/org/springframework/boot/context/embedded/EmbeddedServletContainerMvcIntegrationTests.java b/spring-boot/src/test/java/org/springframework/boot/context/embedded/EmbeddedServletContainerMvcIntegrationTests.java index fd60f366a6..1a7be91bf4 100644 --- a/spring-boot/src/test/java/org/springframework/boot/context/embedded/EmbeddedServletContainerMvcIntegrationTests.java +++ b/spring-boot/src/test/java/org/springframework/boot/context/embedded/EmbeddedServletContainerMvcIntegrationTests.java @@ -49,6 +49,7 @@ import static org.junit.Assert.assertThat; * {@link EmbeddedServletContainer}s running Spring MVC. * * @author Phillip Webb + * @author Ivan Sopov */ public class EmbeddedServletContainerMvcIntegrationTests { @@ -69,7 +70,7 @@ public class EmbeddedServletContainerMvcIntegrationTests { TomcatConfig.class); doTest(this.context, "/hello"); } - + @Test public void jetty() throws Exception { this.context = new AnnotationConfigEmbeddedWebApplicationContext( @@ -83,9 +84,6 @@ public class EmbeddedServletContainerMvcIntegrationTests { UndertowConfig.class); doTest(this.context, "/hello"); } - - - @Test public void advancedConfig() throws Exception { @@ -128,7 +126,7 @@ public class EmbeddedServletContainerMvcIntegrationTests { return new JettyEmbeddedServletContainerFactory(0); } } - + @Configuration @Import(Config.class) public static class UndertowConfig { diff --git a/spring-boot/src/test/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainerFactoryTests.java b/spring-boot/src/test/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainerFactoryTests.java index a25ded4dca..d24337e8c1 100644 --- a/spring-boot/src/test/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainerFactoryTests.java +++ b/spring-boot/src/test/java/org/springframework/boot/context/embedded/undertow/UndertowEmbeddedServletContainerFactoryTests.java @@ -1,34 +1,96 @@ +/* + * Copyright 2012-2014 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + package org.springframework.boot.context.embedded.undertow; + +import io.undertow.Undertow.Builder; + +import java.util.Arrays; + import org.junit.Test; -import org.springframework.boot.context.embedded.*; +import org.mockito.InOrder; +import org.springframework.boot.context.embedded.AbstractEmbeddedServletContainerFactory; +import org.springframework.boot.context.embedded.AbstractEmbeddedServletContainerFactoryTests; +import org.springframework.boot.context.embedded.ErrorPage; +import org.springframework.boot.context.embedded.ExampleServlet; +import org.springframework.boot.context.embedded.ServletRegistrationBean; import org.springframework.http.HttpStatus; import static org.hamcrest.Matchers.equalTo; import static org.junit.Assert.assertThat; - +import static org.mockito.Matchers.anyObject; +import static org.mockito.Mockito.inOrder; +import static org.mockito.Mockito.mock; /** - * Tests for {@link org.springframework.boot.context.embedded.undertow.UndertowEmbeddedServletContainerFactory} and - * {@link org.springframework.boot.context.embedded.undertow.UndertowEmbeddedServletContainer}. + * Tests for {@link UndertowEmbeddedServletContainerFactory} and + * {@link UndertowEmbeddedServletContainer} . * * @author Ivan Sopov + * @author Andy Wilkinson */ -public class UndertowEmbeddedServletContainerFactoryTests extends AbstractEmbeddedServletContainerFactoryTests -{ - - @Override - protected UndertowEmbeddedServletContainerFactory getFactory() - { - return new UndertowEmbeddedServletContainerFactory(); - } - - @Test - public void errorPage404() throws Exception { - AbstractEmbeddedServletContainerFactory factory = getFactory(); - factory.addErrorPages(new ErrorPage(HttpStatus.NOT_FOUND, "/hello")); - this.container = factory.getEmbeddedServletContainer(new ServletRegistrationBean(new ExampleServlet(), "/hello")); - this.container.start(); - assertThat(getResponse("http://localhost:8080/hello"), equalTo("Hello World")); - assertThat(getResponse("http://localhost:8080/not-found"), equalTo("Hello World")); - } +public class UndertowEmbeddedServletContainerFactoryTests extends + AbstractEmbeddedServletContainerFactoryTests { + + @Override + protected UndertowEmbeddedServletContainerFactory getFactory() { + return new UndertowEmbeddedServletContainerFactory(0); + } + + @Test + public void errorPage404() throws Exception { + AbstractEmbeddedServletContainerFactory factory = getFactory(); + factory.addErrorPages(new ErrorPage(HttpStatus.NOT_FOUND, "/hello")); + this.container = factory.getEmbeddedServletContainer(new ServletRegistrationBean( + new ExampleServlet(), "/hello")); + this.container.start(); + assertThat(getResponse(getLocalUrl("/hello")), equalTo("Hello World")); + assertThat(getResponse(getLocalUrl("/not-found")), equalTo("Hello World")); + } + + @Test + public void setNullUndertowBuilderCustomizersThrows() { + UndertowEmbeddedServletContainerFactory factory = getFactory(); + this.thrown.expect(IllegalArgumentException.class); + this.thrown.expectMessage("undertowBuilderCustomizers must not be null"); + factory.setUndertowBuilderCustomizers(null); + } + + @Test + public void addNullContextCustomizersThrows() { + UndertowEmbeddedServletContainerFactory factory = getFactory(); + this.thrown.expect(IllegalArgumentException.class); + this.thrown.expectMessage("undertowBuilderCustomizers must not be null"); + factory.addUndertowBuilderCustomizers((UndertowBuilderCustomizer[]) null); + } + + @Test + public void builderCustomizers() throws Exception { + UndertowEmbeddedServletContainerFactory factory = getFactory(); + UndertowBuilderCustomizer[] customizers = new UndertowBuilderCustomizer[4]; + for (int i = 0; i < customizers.length; i++) { + customizers[i] = mock(UndertowBuilderCustomizer.class); + } + factory.setUndertowBuilderCustomizers(Arrays.asList(customizers[0], + customizers[1])); + factory.addUndertowBuilderCustomizers(customizers[2], customizers[3]); + this.container = factory.getEmbeddedServletContainer(); + InOrder ordered = inOrder((Object[]) customizers); + for (UndertowBuilderCustomizer customizer : customizers) { + ordered.verify(customizer).customize((Builder) anyObject()); + } + } }