3

我们在 Playframework 上有一个小型 Scala 项目。我正在尝试做所有反应,并偶然发现了一个问题。

我有两个 Enumerator[A] 实例,代表按日期排序的数据库中的值。我需要将它们作为单个 Enumerator[A] 保持日期排序返回。我在 Enumerator[A] 中没有找到任何解决方案,所以我在单个集合中累积 A,然后对它们进行排序。

case class A(
   created: Date,
   data: String
)

val as: Enumerator[A] = findByAOrderedByCreated()
val bs: Enumerator[A] = findByBOrderedByCreated()

什么是应对这种情况的反应方式?

4

1 回答 1

2

这是一个解决方案,适用于在元素上Enumerator使用 any 的任意数量的值:Ordering

import play.api.libs.iteratee._
import scala.concurrent._

object MergeEnums {
  def apply[E: Ordering](enums: Enumerator[E]*)(implicit executor: ExecutionContext) = new Enumerator[E] {
    def apply[A](iter: Iteratee[E, A]) = {
      case class IterateeReturn(o: Option[(Promise[Promise[IterateeReturn]], E)])

      val failP = Promise()
      val failPF = failP.future
      val initState = Future.traverse(enums) { enum =>
        val p = Promise[IterateeReturn]()
        enum.run(Iteratee.foldM(p) { (oldP: Promise[IterateeReturn], elem: E) =>
          val p = Promise[Promise[IterateeReturn]]()
          oldP success IterateeReturn(Some(p, elem))
          p.future
        } map { promise =>
          promise success IterateeReturn(None)
        }) onFailure { case t => failP failure t }
        p.future
      } map (_.map(_.o).flatten.toList)

      Enumerator.unfoldM(initState) { fstate =>
        Future.firstCompletedOf(Seq(fstate, failPF)) map { state =>
          state.sortBy(_._2) match {
            case Nil => None
            case (oldP, elem) :: tail =>
              val p = Promise[IterateeReturn]()
              oldP success p
              val newState = p.future.map(_.o.map(_ :: tail).getOrElse(tail))
              Some(newState, elem)
          }
        }
      } apply iter
    }
  }
}

它创建一个Iteratee应用于Enumerator传入的每个元素,并创建一个Enumerator给排序的元素。Iteratee实例和实例通过相互Enumerator发送Promise实例进行通信(因此是Promise[Promise[IterateeReturn]]这样的)。

于 2014-04-16T20:22:47.867 回答