Merge pull request #133 from buildo/api-130-unbind_port_when_program

#130: Unbind port when program exit (closes #130)
buildo · Jul 25, 2018 · d6ca64e · d6ca64e
2 parents a5a7fe2 + d99a7ca
commit d6ca64e
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/wiro/serverAkkaHttp/src/main/scala/RPCServer.scala b/wiro/serverAkkaHttp/src/main/scala/RPCServer.scala
@@ -49,7 +49,13 @@ class HttpRPCServerActor(
     case Status.Failure(cause) => logger.error(s"Unable to bind to ${config.host}:${config.port}", cause)
   }
 
-  Http()
-    .bindAndHandle(route, config.host, config.port)
-    .pipeTo(self)
+  val bindingFuture = Http()
+        .bindAndHandle(route, config.host, config.port)
+        .pipeTo(self)
+
+  scala.sys.addShutdownHook({
+      logger.info("Wiro unbinding on exit")
+      bindingFuture
+        .flatMap(_.unbind())
+    })
 }