Polish Undertow reactive server support

This commit is contained in:
Rossen Stoyanchev
2016-12-14 14:51:33 -05:00
parent c3621bf153
commit 70f14c14c8
4 changed files with 82 additions and 89 deletions

View File

@@ -17,7 +17,6 @@
package org.springframework.http.server.reactive.bootstrap;
import io.undertow.Undertow;
import io.undertow.server.HttpHandler;
import org.springframework.http.server.reactive.UndertowHttpHandlerAdapter;
import org.springframework.util.Assert;
@@ -34,10 +33,19 @@ public class UndertowHttpServer extends HttpServerSupport implements HttpServer
@Override
public void afterPropertiesSet() throws Exception {
Assert.notNull(getHttpHandler());
HttpHandler handler = new UndertowHttpHandlerAdapter(getHttpHandler());
this.server = Undertow.builder().addHttpListener(getPort(), getHost())
.setHandler(handler).build();
.setHandler(initUndertowHttpHandlerAdapter())
.build();
}
private UndertowHttpHandlerAdapter initUndertowHttpHandlerAdapter() {
if (getHttpHandlerMap() != null) {
return new UndertowHttpHandlerAdapter(getHttpHandlerMap());
}
else {
Assert.notNull(getHttpHandler());
return new UndertowHttpHandlerAdapter(getHttpHandler());
}
}
@Override