Return Undertow server when calling SimpleServer.start()

billoneil · billoneil · commit 05f411765a0e · 2018-08-16T09:26:39.000-04:00
diff --git a/stubbornjava-common/src/main/java/com/stubbornjava/common/undertow/SimpleServer.java b/stubbornjava-common/src/main/java/com/stubbornjava/common/undertow/SimpleServer.java
@@ -26,7 +26,7 @@ public Undertow.Builder getUndertow() {
         return undertowBuilder;
     }
 
-    public void start() {
+    public Undertow start() {
         Undertow undertow = undertowBuilder.build();
         undertow.start();
         /*
@@ -37,6 +37,7 @@ public void start() {
         undertow.getListenerInfo()
                 .stream()
                 .forEach(listenerInfo -> logger.info(listenerInfo.toString()));
+        return undertow;
     }
 
     public static SimpleServer simpleServer(HttpHandler handler) {
diff --git a/stubbornjava-webapp/src/main/java/com/stubbornjava/webapp/StubbornJavaWebApp.java b/stubbornjava-webapp/src/main/java/com/stubbornjava/webapp/StubbornJavaWebApp.java
@@ -19,6 +19,7 @@
 import com.stubbornjava.webapp.post.PostRoutes;
 import com.stubbornjava.webapp.themes.ThemeRoutes;
 
+import io.undertow.Undertow;
 import io.undertow.server.HttpHandler;
 import io.undertow.server.RoutingHandler;
 import io.undertow.server.handlers.BlockingHandler;
@@ -90,11 +91,11 @@ private static final HttpHandler getBasicRoutes() {
             .setFallbackHandler(timed("notFound", PageRoutes::notFound));
     }
 
-    private static void startServer() {
+    private static Undertow startServer() {
         HttpHandler staticRoutes = new PathHandler(getBasicRoutes())
             .addPrefixPath("/assets", timed("getAssets", CustomHandlers.resource("", (int)TimeUnit.DAYS.toSeconds(30))));
         SimpleServer server = SimpleServer.simpleServer(wrapWithMiddleware(staticRoutes));
-        server.start();
+        return server.start();
     }
 
     public static void main(String[] args) {

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ public Undertow.Builder getUndertow() {`
`26`	`26`	`return undertowBuilder;`
`27`	`27`	`}`
`28`	`28`
`29`		`- public void start() {`
	`29`	`+ public Undertow start() {`
`30`	`30`	`Undertow undertow = undertowBuilder.build();`
`31`	`31`	`undertow.start();`
`32`	`32`	`/*`
`@@ -37,6 +37,7 @@ public void start() {`
`37`	`37`	`undertow.getListenerInfo()`
`38`	`38`	`.stream()`
`39`	`39`	`.forEach(listenerInfo -> logger.info(listenerInfo.toString()));`
	`40`	`+ return undertow;`
`40`	`41`	`}`
`41`	`42`
`42`	`43`	`public static SimpleServer simpleServer(HttpHandler handler) {`