如何将 Twitter Future[Map[A, Future[C]]] 转换为 Future[Map[A, C]]

Question

我正在尝试解决以下问题：

我有一个 Future[Map[A, B]]。对于所有 B，我需要应用将 B 转换为 Future[C] 的方法，并且我想返回 Future[Map[A, C]]

这是我目前的代码：

def getClients(clientIds: Seq[Int]): Future[Map[Int, ClientData]] = {

  def getClientData(clientInfo: ClientInfo): Future[ClientData] = 
    clientInfo match {
      case ValidInfo(info) => getData(info)
      case _ => throw new Exception
    }

  client.getClients(clientIds) map {
    _.toMap map {
      case (clientId: Int, clientInfo: ClientInfo) =>
        getClientData(clientInfo) map {
          clientData => (clientId, clientData)
      }
    }
  }
}

此代码错误，因为它返回 Iterable[Future[(Int, ClientData)]]

信息 getClients 是一种节俭方法，它返回 Future[Map[A, B]] 其中 Map 是可变的，因此我需要将其转换为不可变先用 toMap 映射。

提前感谢您的帮助！

Answer 1

scala> def f: Future[Map[String, Future[Int]]] = ???
f: Future[Map[String,Future[Int]]]

scala> def x = for {
     |   m <- f
     |   i = m.map{ case (k, fv) => fv.map{ k -> _ } }
     |   l <- Future.sequence(i)
     | } yield l.toMap
x: Future[Map[String,Int]]

一步一步：

将Future[Map[A, Future[B]]] 转换为Future[Iterable[Future[(A, B)]]]：

scala> def x1 = f.map{ _.map{ case (k, fv) => fv.map{ k -> _ } } }
x1: Future[Iterable[Future[(String, Int)]]]

使用flatMap 将Iterable[Future[(A, B)]] 转换为Future[Iterable[(A, B)]] 和flatten Future[Future[...]]：

scala> def x2 = x1.flatMap{ Future.sequence(_) }
x2: Future[immutable.Iterable[(String, Int)]]

将Iterable[(A, B)] 转换为Map[A, B]：

scala> def x = x2.map{ _.toMap }
x: Future[Map[String,Int]]

对于com.twitter.util.Future，您应该在collect 之前使用collect 而不是sequence 和toSeq，因为它接受Seq：

def x = for {
  m <- f
  i = m.map{ case (k, fv) => fv.map{ k -> _ } }
  l <- Future.collect(i.toSeq)
} yield l.toMap