将期货列表合并到 Scala 中的地图中

Question

我有两个案例类P(id: String, ...) 和Q(id: String, ...)，以及两个返回期货的函数：

1. 在给定 id-s 列表的情况下检索对象列表： def retrieve(ids: Seq[String]): Future[Seq[P]] = Future { ... }

   如果没有找到所有 id-s，结果的长度可能会比输入短。

2. 进一步将P 转换为其他类型Q： def transform(p: P): Future[Q] = Future { ... }

我最后想要的是，以下。给定ids: Seq[String]，计算一个Future[Map[String, Option[Q]]]。

ids 中的每个id 都应该是映射中的一个键，当它被成功检索（即出现在retrieve 的结果中）并且也成功转换时，id -> Some(q) 应该是一个键。否则，地图应包含id -> None 或Empty。

我怎样才能做到这一点？

Answer 1

P 或 Q 上是否有 .id 属性？您需要一个来创建地图。像这样？

for {
  ps <- retrieve(ids)
  qs <- Future.sequence(ps.map(p => transform(p))
} yield ids.map(id => id -> qs.find(_.id == id)).toMap

请记住，Map[String,Option[X]] 通常不是必需的，因为如果您有 Map[String,X]，则地图上的 .get 方法将为您提供 Option[X]。

Answer 2

编辑：现在假设P 有一个成员id 等于原始id-String，否则ids 和ps 之间的连接在retrieve 之后丢失。

def consolidatedMap(ids: Seq[String]): Future[Map[String, Option[Q]]] = {
  for {
    ps <- retrieve(ids)
    qOpts <- Future.traverse(ps){
      p => transform(p).map(Option(_)).recover {
        // TODO: don't sweep `Throwable` under the 
        // rug in your real code
        case t: Throwable => None 
      }
    }
  } yield {
    val qMap = (ps.map(_.id) zip qOpts).toMap
    ids.map{ id => (id, qMap.getOrElse(id, None)) }.toMap
  }
}

从检索到的Ps 构建中间Map 并转换Qs，以便ids-to-q-Options 映射的构建在线性时间内工作。