Question

我刚开始在我的新项目中使用Scala（Scala 2.10.3，Play2 2.2.1，Reactivemongo 0.10.0），并遇到了一个非常标准的用例，即将MongoDB中的所有用户都流式传输到外部客户。在浏览Enumerator之后，Enumeratee API我没有找到一个可靠的解决方案，所以我通过以下方式解决了这个问题：

    val users = collection.find(Json.obj()).cursor[User].enumerate(Integer.MAX_VALUE, false)
    var first:Boolean = true
    val indexedUsers = (users.map(u => {
        if(first) {
            first = false;
            Json.stringify(Json.toJson(u))
        } else {
            "," + Json.stringify(Json.toJson(u))
        }
    }))

从我的观点来看，这有点棘手 - 主要是因为我需要在元素列表中添加Json Start Array，Json End Array和逗号分隔符，而我无法将其作为纯Json流提供，所以我将它转换为String steam。

什么是标准解决方案，在游戏中使用reactivemongo？

Answer 1

我编写了一个帮助函数来完成你想要实现的目标：

def intersperse[E](e: E, enum: Enumerator[E]): Enumerator[E] = new Enumerator[E] {
  val element = Input.El(e)

  override def apply[A](i1: Iteratee[E, A]): Future[Iteratee[E, A]] = {
    var iter = i1

    val loop: Iteratee[E, Unit] = {
      lazy val contStep = Cont(step)

      def step(in: Input[E]): Iteratee[E, Unit] = in match {
        case Input.Empty ⇒ contStep
        case Input.EOF ⇒ Done((), Input.Empty)
        case e @ Input.El(_) ⇒
          iter = Iteratee.flatten(iter.feed(element).flatMap(_.feed(e)))
          contStep
      }

      lazy val contFirst = Cont(firstStep)

      def firstStep(in: Input[E]): Iteratee[E, Unit] = in match {
        case Input.EOF ⇒ Done((), Input.Empty)
        case Input.Empty ⇒
          iter = Iteratee.flatten(iter.feed(in))
          contFirst
        case Input.El(x) ⇒
          iter = Iteratee.flatten(iter.feed(in))
          contStep
      }

      contFirst
    }

    enum(loop).map { _ ⇒ iter }
  }
}

用法：

val prefix = Enumerator("[")
val suffix = Enumerator("]")
val asStrings = Enumeratee.map[User] { u => Json.stringify(Json.toJson(u)) }
val result = prefix >>> intersperse(",", users &> asStrings) >>> suffix

Ok.chunked(result)

逗号分隔列表与Enumerator

1 个答案: