3

我正在使用 play2 和 reactivemongo 从 mongodb 获取结果。结果的每一项都需要转换以添加一些元数据。之后我需要对其进行一些排序。

为了处理我使用的转换步骤enumerate()

def ideasEnumerator = collection.find(query)
    .options(QueryOpts(skipN = page))
    .sort(Json.obj(sortField -> -1))
    .cursor[Idea]
    .enumerate()

然后我创建一个Iteratee,如下所示:

val processIdeas: Iteratee[Idea, Unit] =
  Iteratee.foreach[Idea] { idea =>
    resolveCrossLinks(idea) flatMap { idea =>
      addMetaInfo(idea.copy(history = None))
    }
  }

最后我喂Iteratee:

ideasEnumerator(processIdeas)

现在我被困住了。我看到的每个示例都println在里面做一些foreach事情,但似乎并不关心最终结果。

因此,当所有文档都返回并转换后,我如何获得序列、列表或其他可以进一步处理的数据类型?

4

2 回答 2

2

将 Iteratee 的签名从A 类型更改Iteratee[Idea, Unit]为。Iteratee[Idea, Seq[A]]基本上 Iteratee 的第一个参数是输入类型,第二个参数是输出类型。在您的情况下,您将输出类型指定为单位。

看看下面的代码。它可能无法编译,但它为您提供了基本用法。

ideasEnumerator.run(
  Iteratee.fold(List.empty[MyObject]) { (accumulator, next) => 
    accumulator + resolveCrossLinks(next) flatMap { next => 
      addMetaInfo(next.copy(history = None))
    } 
  }
) // returns Future[List[MyObject]]

如您所见,Iteratee 是一个简单的状态机。只需提取该 Iteratee 部分并将其分配给 val:

val iteratee = Iteratee.fold(List.empty[MyObject]) { (accumulator, next) => 
        accumulator + resolveCrossLinks(next) flatMap { next => 
          addMetaInfo(next.copy(history = None))
        } 
      }

并在任何需要将想法转换为 List[MyObject] 的地方随意使用它

于 2014-07-09T17:49:23.747 回答
0

在你的答案的帮助下,我最终得到了

val processIdeas: Iteratee[Idea, Future[Vector[Idea]]] = 
    Iteratee.fold(Future(Vector.empty[Idea])) { (accumulator: Future[Vector[Idea]], next:Idea) =>
  resolveCrossLinks(next) flatMap { next =>
    addMetaInfo(next.copy(history = None))
  } flatMap (ideaWithMeta => accumulator map (acc => acc :+ ideaWithMeta))
}


val ideas = collection.find(query)
  .options(QueryOpts(page, perPage))
  .sort(Json.obj(sortField -> -1))
  .cursor[Idea]
  .enumerate(perPage).run(processIdeas)

这稍后需要ideas.flatMap(identity)删除 Future of Futures 的返回,但我对此很好,我认为一切看起来都是惯用的和优雅的。

与创建列表并随后对其进行迭代相比,所获得的性能可以忽略不计。

于 2014-07-10T08:58:26.010 回答