Question

我的旧代码看起来像下面所有db调用阻塞。

我需要帮助将其转换为使用期货。

def getUserPoints(username: String): Option[Long]
    db.getUserPoints(username) match {
        case Some(userPoints) => Some(userPoints.total)
        case None => {
            if (db.getSomething("abc").isEmpty) {
                db.somethingElse("asdf") match {
                    case Some(pointId) => {
                        db.setPoints(pointId, username)
                        db.findPointsForUser(username)
                    }
                    case _ => None
                }
            } else {
                db.findPointsForUser(username)
            }
        }       
    }
}

我的新API位于我返回期货的位置。

db.getUserPoints(username: String): Future[Option[UserPoints]]
db.getSomething(s: String): Future[Option[Long]]
db.setPoints(pointId, username): Future[Unit]
db.findPointsForUser(username): Future[Option[Long]]

如何转换上述内容以使用我使用未来的新API。

我尝试使用for-compr，但开始遇到像Future [Nothing]这样的错误。

var userPointsFut: Future[Long] = for {
  userPointsOpt <- db.getUserPoints(username)
  userPoints <- userPointsOpt
} yield userPoints.total

但是对于所有的分支和if条款并尝试将其转换为期货，它会变得有点棘手。

Answer 1

我认为这个设计的第一个问题是对Future的阻塞调用的端口不应该包装Option类型：

拦截电话： def giveMeSomethingBlocking(for:Id): Option[T] 应该成为： def giveMeSomethingBlocking(for:Id): Future[T] 不： def giveMeSomethingBlocking(for:Id): Future[Option[T]]

阻止调用会给出值Some(value)或None，非阻塞的Future版本会提供Success(value)或Failure(exception)，它会完全保留Option语义以非阻塞的方式。

考虑到这一点，我们可以使用Future上的组合器对相关过程进行建模。我们来看看如何：

首先，让我们将API重构为我们可以使用的东西：

type UserPoints = Long
object db {
  def getUserPoints(username: String): Future[UserPoints] = ???
  def getSomething(s: String): Future[UserPoints] = ???
  def setPoints(pointId:UserPoints, username: String): Future[Unit] = ???
  def findPointsForUser(username: String): Future[UserPoints] = ???
}
class PointsNotFound extends Exception("bonk")
class StuffNotFound extends Exception("sthing not found")

然后，过程看起来像：

def getUserPoints(username:String): Future[UserPoints] = {
  db.getUserPoints(username)
  .map(userPoints => userPoints /*.total*/)
  .recoverWith{ 
    case ex:PointsNotFound => 
    (for {
      sthingElse <- db.getSomething("abc")
      _ <- db.setPoints(sthingElse, username)
      points <- db.findPointsForUser(username)
    } yield (points))
    .recoverWith{
      case ex: StuffNotFound => db.findPointsForUser(username)
    }
  }
}

正确检查哪种类型。

修改

鉴于API是一成不变的，处理嵌套monadic类型的方法是定义MonadTransformer。简单来说，让Future[Option[T]]成为一个新的monad，让它称之为FutureO，可以与其他类型的monad组合。 [1]

case class FutureO[+A](future: Future[Option[A]]) {
  def flatMap[B](f: A => FutureO[B])(implicit ec: ExecutionContext): FutureO[B] = {
    val newFuture = future.flatMap{
      case Some(a) => f(a).future
      case None => Future.successful(None)
    }
    FutureO(newFuture)
  }

  def map[B](f: A => B)(implicit ec: ExecutionContext): FutureO[B] = {
    FutureO(future.map(option => option map f))
  }
  def recoverWith[U >: A](pf: PartialFunction[Throwable, FutureO[U]])(implicit executor: ExecutionContext): FutureO[U] = {
    val futOtoFut: FutureO[U] => Future[Option[U]] = _.future
    FutureO(future.recoverWith(pf andThen futOtoFut))
  }

  def orElse[U >: A](other: => FutureO[U])(implicit executor: ExecutionContext): FutureO[U] = {
      FutureO(future.flatMap{
        case None => other.future
        case _ => this.future
      }) 
    }
  }

现在我们可以重新编写我们的流程，保留与基于未来的组合相同的结构。

type UserPoints = Long 
object db { 
  def getUserPoints(username: String): Future[Option[UserPoints]] = ???
  def getSomething(s: String): Future[Option[Long]] = ???
  def setPoints(pointId: UserPoints, username:String): Future[Unit] = ???
  def findPointsForUser(username: String): Future[Option[Long]] = ???
}
class PointsNotFound extends Exception("bonk")
class StuffNotFound extends Exception("sthing not found")

def getUserPoints2(username:String): Future[Option[UserPoints]] = {
  val futureOpt = FutureO(db.getUserPoints(username))
  .map(userPoints => userPoints /*.total*/)
  .orElse{ 
    (for {
      sthingElse <- FutureO(db.getSomething("abc"))
      _ <- FutureO(db.setPoints(sthingElse, username).map(_ => Some(())))
      points <- FutureO(db.findPointsForUser(username))
    } yield (points))
    .orElse{
      FutureO(db.findPointsForUser(username))
    }
  }
  futureOpt.future
}

[1]对http://loicdescotte.github.io/posts/scala-compose-option-future/

的致谢

将阻止代码转换为使用scala期货

1 个答案:

修改