Wait until all Future.onComplete callbacks are executed

后端 未结 2 1803
萌比男神i
萌比男神i 2021-01-05 04:25

I am using the Future API from Scala 2.10.X.

Here is my use case:

object Class1 {

  def apply(f: (Int) => Future[String])(i: Int): F         


        
相关标签:
2条回答
  • 2021-01-05 05:09

    With 1 Future and 3 onComplete

    I think you're going to have to go the route of either composing your functions into a single onComplete call or else you'll have to do exactly what you said, use map:

     val fut1 = myFut map func1 // yes, a Future[Unit]
     val fut2 = myFut map func2
     val fut3 = myFut map func3
    

    Follow the next section to find out when they all finish.

    With 3 different Futures

    It's very possible to know when all three Futures will complete. In fact, in Scala Future composes!

     def threeFutures(one: Future[Int], two: Future[Int], three: Future[Int]) {
       val fourth = for {
         _ <- one
         _ <- two
         _ <- three
       } yield 0
    
       fourth onComplete {
         case _ => println("all done")
       }
     }
    

    Now what does this mean? It means that fourth is a Future which does not care about the inputs of the three arguments but that, when they all are complete, it will complete itself. This is pre-packaged and made ready just for you.

    (Side note: In the example I'm also assuming you have all your implicits in scope.)

    0 讨论(0)
  • 2021-01-05 05:26

    If you don't want to use other methods (like a CountDownLatch), then you want to use andThen to know when your operations complete (successfully or not, and whether or not the Future was successful).

    scala> val f = Future(3)
    f: scala.concurrent.Future[Int] = scala.concurrent.impl.Promise$DefaultPromise@4b49ca35
    
    scala> val g = f andThen { case Success(i) => println(i) } andThen { case _ => println("All done") }
    3
    g: scala.concurrent.Future[Int] = scala.concurrent.impl.Promise$DefaultPromise@1939e13
    All done
    

    If the future fails, the mapped function by contrast isn't invoked:

    scala> val f = Future[Int](???)
    f: scala.concurrent.Future[Int] = scala.concurrent.impl.Promise$DefaultPromise@7001619b
    
    scala> val g = f andThen { case t => println(s"stage 1 $t") } andThen { case _ => println("All done") }
    stage 1 Failure(java.util.concurrent.ExecutionException: Boxed Error)
    All done
    g: scala.concurrent.Future[Int] = scala.concurrent.impl.Promise$DefaultPromise@24e1e7e8
    
    scala> val g = f map { case i => println(i) } andThen { case _ => println("All done") }
    All done
    g: scala.concurrent.Future[Unit] = scala.concurrent.impl.Promise$DefaultPromise@5d0f75d6
    
    scala> val g = f map { case i => println(i) } map { case _ => println("All done") }
    g: scala.concurrent.Future[Unit] = scala.concurrent.impl.Promise$DefaultPromise@5aabe81f
    
    scala> g.value
    res1: Option[scala.util.Try[Unit]] = Some(Failure(java.util.concurrent.ExecutionException: Boxed Error))
    

    Similarly, blowing up in a chained handler doesn't break subsequent operations:

    scala> val g = f andThen { case t => null.hashCode } andThen { case _ => Thread.sleep(1000L); println("All done") }
    java.lang.NullPointerException
        at $line26.$read$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$anonfun$1.applyOrElse(<console>:51)
        at $line26.$read$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$iw$$anonfun$1.applyOrElse(<console>:51)
        at scala.concurrent.Future$$anonfun$andThen$1.apply(Future.scala:431)
        at scala.concurrent.Future$$anonfun$andThen$1.apply(Future.scala:430)
        at scala.concurrent.impl.CallbackRunnable.run(Promise.scala:32)
        at scala.concurrent.impl.ExecutionContextImpl$AdaptedForkJoinTask.exec(ExecutionContextImpl.scala:121)
        at scala.concurrent.forkjoin.ForkJoinTask.doExec(ForkJoinTask.java:260)
        at scala.concurrent.forkjoin.ForkJoinPool$WorkQueue.runTask(ForkJoinPool.java:1339)
        at scala.concurrent.forkjoin.ForkJoinPool.runWorker(ForkJoinPool.java:1979)
        at scala.concurrent.forkjoin.ForkJoinWorkerThread.run(ForkJoinWorkerThread.java:107)
    g: scala.concurrent.Future[Int] = scala.concurrent.impl.Promise$DefaultPromise@3fb7bec8
    
    scala> All done
    
    
    scala> g.value
    res1: Option[scala.util.Try[Int]] = Some(Success(3))
    

    For the unfortunate case of needing to wait for it:

    scala> val f = Future[Int](???)
    f: scala.concurrent.Future[Int] = scala.concurrent.impl.Promise$DefaultPromise@859a977
    
    scala> import java.util.concurrent.{ CountDownLatch => CDL }
    import java.util.concurrent.{CountDownLatch=>CDL}
    
    scala> val latch = new CDL(3)
    latch: java.util.concurrent.CountDownLatch = java.util.concurrent.CountDownLatch@11683e9f[Count = 3]
    
    scala> f onComplete { _ => println(1); latch.countDown() }
    1
    
    scala> f onComplete { _ => println(2); latch.countDown() }
    2
    
    scala> f onComplete { _ => println(3); latch.countDown() }
    3
    
    scala> f onComplete { _ => latch.await(); println("All done") }
    All done
    
    0 讨论(0)
提交回复
热议问题