使 Enumerator / Iteratee 适应 Stream

Question

我有一个 Spray 应用程序，我在其中使用 reactive-mongo 连接到我的 Mongo 数据库。

val collection: BSONCollection = db("ping")
val cursor: Cursor[Ping] = collection.find(BSONDocument()).cursor[Ping]
val e: Enumerator[Ping] = cursor.enumerate()

ReactiveMongo 的光标允许我按块获取数据。此光标可以转换为 play.api.libs.iteratee.Enumerator ，后者又连接到 play.api.libs.iteratee.Iteratee ，其中为集合的每个元素调用我想要的任何代码。 “Ping”是我的域实体。

使用 Spray 可以返回一个 Stream[Ping]，该 Stream[Ping] 将被编组为 JSON 响应。我想知道是否可以将 Enumerator 或 Iteratee 调整为 Stream，所以我可以返回它。

Answer 1

您可以在 Cursor 实例上使用 collect[Stream] 而不是 enumerate 来执行此操作：

val collection: BSONCollection = db("ping")
val cursor: Cursor[Ping] = collection.find(BSONDocument()).cursor[Ping]
val stream: Stream[Ping] = cursor.toList.toStream

编辑：使用cursor.toList.toStream。这将收集与查询匹配的所有文档，然后生成一个流。