From 6db15da409ee558a32470bf2b6dfc75c8fa5753e Mon Sep 17 00:00:00 2001 From: Bo Meng Date: Wed, 25 Jul 2018 15:54:35 -0700 Subject: [PATCH 1/2] add terminate --- .../org/kunicki/akka_streams/RandomDataGenerator.scala | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/main/scala/org/kunicki/akka_streams/RandomDataGenerator.scala b/src/main/scala/org/kunicki/akka_streams/RandomDataGenerator.scala index f1dcc90..54ed0c6 100644 --- a/src/main/scala/org/kunicki/akka_streams/RandomDataGenerator.scala +++ b/src/main/scala/org/kunicki/akka_streams/RandomDataGenerator.scala @@ -9,16 +9,16 @@ import akka.stream.scaladsl.{FileIO, Sink, Source} import akka.util.ByteString import com.typesafe.config.ConfigFactory import com.typesafe.scalalogging.LazyLogging -import org.kunicki.akka_streams.model.ValidReading -import scala.concurrent.Await import scala.concurrent.duration.Duration +import scala.concurrent.{Await, ExecutionContextExecutor} import scala.util.Random object RandomDataGenerator extends App with LazyLogging { implicit val system = ActorSystem("random-data-generator") implicit val materializer = ActorMaterializer() + implicit val ec: ExecutionContextExecutor = system.dispatcher val config = ConfigFactory.load() val numberOfFiles = config.getInt("generator.number-of-files") @@ -40,8 +40,7 @@ object RandomDataGenerator extends App with LazyLogging { } .runWith(Sink.ignore) - Source(1 to 100) - Await.ready(f, Duration.Inf) logger.info("Generated random data") + f.onComplete(_ => system.terminate()) } From 6fb452c2fbdcf6e72d872a4e0b3d9c84cf54bc15 Mon Sep 17 00:00:00 2001 From: Bo Meng Date: Wed, 25 Jul 2018 16:02:11 -0700 Subject: [PATCH 2/2] add import --- .../scala/org/kunicki/akka_streams/RandomDataGenerator.scala | 1 + 1 file changed, 1 insertion(+) diff --git a/src/main/scala/org/kunicki/akka_streams/RandomDataGenerator.scala b/src/main/scala/org/kunicki/akka_streams/RandomDataGenerator.scala index 54ed0c6..2bae189 100644 --- a/src/main/scala/org/kunicki/akka_streams/RandomDataGenerator.scala +++ b/src/main/scala/org/kunicki/akka_streams/RandomDataGenerator.scala @@ -9,6 +9,7 @@ import akka.stream.scaladsl.{FileIO, Sink, Source} import akka.util.ByteString import com.typesafe.config.ConfigFactory import com.typesafe.scalalogging.LazyLogging +import org.kunicki.akka_streams.model.ValidReading import scala.concurrent.duration.Duration import scala.concurrent.{Await, ExecutionContextExecutor}