Shutdown http pools

2m · Oct 19, 2017 · ca66312 · ca66312
1 parent f331ed3
commit ca66312
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/core/src/main/scala/Authors.scala b/core/src/main/scala/Authors.scala
@@ -56,17 +56,20 @@ object Authors {
 
     import sys.dispatcher
 
-    val completion =
-      DiffSource(repo, from, to)
-        .via(ActsonReader.instance)
-        .via(ProtocolReader.of(GithubProtocol.compareProto))
-        .via(StatsAggregator())
-        .via(SortingMachine())
-        .via(MarkdownConverter())
-        .runFold("")(_ ++ "\n" ++ _)
-
-    completion.onComplete(_ => sys.terminate())
-    completion
+    DiffSource(repo, from, to)
+      .via(ActsonReader.instance)
+      .via(ProtocolReader.of(GithubProtocol.compareProto))
+      .via(StatsAggregator())
+      .via(SortingMachine())
+      .via(MarkdownConverter())
+      .runFold("")(_ ++ "\n" ++ _)
+      .transformWith { res =>
+        for {
+          _ <- Http().shutdownAllConnectionPools()
+          _ <- sys.terminate()
+          r <- Future.fromTry(res)
+        } yield r
+      }
   }
 
   def gitRepo(path: String): FileRepository =