如何根据合并规则将2个枚举器合并为一个

时间:2014-04-16 12:54:13

标签: java scala playframework-2.0 enumerator iterate

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

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

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

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

处理此事的反应方式是什么?

1 个答案:

答案 0 :(得分:2)

以下解决方案适用于使用元素上任意Enumerator的任意数量的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]]等等。)