munit.internal.FutureCompat.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of munit_2.11 Show documentation

munit

There is a newer version: 1.0.0-M10

package munit.internal

import scala.concurrent.Future
import scala.util.Try
import scala.concurrent.ExecutionContext
import scala.concurrent.Promise

object FutureCompat {
  implicit class ExtensionFuture[T](f: Future[T]) {
    def flattenCompat[S](
        ec: ExecutionContext
    )(implicit ev: T <:< Future[S]): Future[S] =
      f.flatMap(ev)(ec)
    def transformCompat[B](
        fn: Try[T] => Try[B]
    )(implicit ec: ExecutionContext): Future[B] = {
      val p = Promise[B]()
      f.onComplete { t => p.complete(fn(t)) }
      p.future
    }
    def transformWithCompat[B](
        fn: Try[T] => Future[B]
    )(implicit ec: ExecutionContext): Future[B] = {
      val p = Promise[B]()
      f.onComplete { t => p.completeWith(fn(t)) }
      p.future
    }
  }
}