创建基于时间的分块枚举

时间:2013-03-09 13:41:56

标签: scala playframework playframework-2.0 akka

我想创建一个Play 2 Enumeratee,它接收值并以x秒/毫秒为单位将它们组合在一起。这样,在具有大量用户输入的多用户websocket环境中,可以限制每秒接收到的帧数。

我知道可以像这样将一定数量的项目组合在一起:

val chunker = Enumeratee.grouped(
  Traversable.take[Array[Double]](5000) &>> Iteratee.consume()
)

是否有基于时间而不是基于项目数量的内置方式来执行此操作?

我正在考虑以预定的Akka工作以某种方式这样做,但初看起来这似乎效率低下,而且我不确定是否会出现相关问题。

2 个答案:

答案 0 :(得分:3)

这样怎么样?我希望这对你有所帮助。

 package controllers

 import play.api._
 import play.api.Play.current
 import play.api.mvc._
 import play.api.libs.iteratee._
 import play.api.libs.concurrent.Akka
 import play.api.libs.concurrent.Promise

 object Application extends Controller {

   def index = Action {
     val queue = new scala.collection.mutable.Queue[String]
     Akka.future {
       while( true ){
         Logger.info("hogehogehoge")
         queue += System.currentTimeMillis.toString
         Thread.sleep(100)
       }
     }
     val timeStream = Enumerator.fromCallback { () =>
       Promise.timeout(Some(queue), 200)
     }
     Ok.stream(timeStream.through(Enumeratee.map[scala.collection.mutable.Queue[String]]({ queue =>
       var str = ""
       while(queue.nonEmpty){
         str += queue.dequeue + ", "
       }
       str
     })))
   }

 }

此文档对您也有帮助。 http://www.playframework.com/documentation/2.0/Enumerators

<强>更新 这是适用于play2.1版本。

 package controllers

 import play.api._
 import play.api.Play.current
 import play.api.mvc._
 import play.api.libs.iteratee._
 import play.api.libs.concurrent.Akka
 import play.api.libs.concurrent.Promise
 import scala.concurrent._
 import ExecutionContext.Implicits.global

 object Application extends Controller {

   def index = Action {
     val queue = new scala.collection.mutable.Queue[String]
     Akka.future {
       while( true ){
         Logger.info("hogehogehoge")
         queue += System.currentTimeMillis.toString
         Thread.sleep(100)
       }
     }
     val timeStream = Enumerator.repeatM{
       Promise.timeout(queue, 200)
     }
     Ok.stream(timeStream.through(Enumeratee.map[scala.collection.mutable.Queue[String]]({ queue =>
       var str = ""
       while(queue.nonEmpty){
         str += queue.dequeue + ", "
       }
       str
     })))
   }

 }

答案 1 :(得分:2)

在这里,我快速定义了一个迭代,它将从输入中获取固定时间长度t的值,以毫秒为单位,以及一个枚举,它允许您对输入流进行分组和进一步处理,这些输入流被分成在此类中构建的段。长度t。它依赖于JodaTime来跟踪自迭代开始以来已经过了多少时间。

def throttledTakeIteratee[E](timeInMillis: Long): Iteratee[E, List[E]] = {
  var startTime = new Instant()

  def step(state: List[E])(input: Input[E]): Iteratee[E, List[E]] = {
    val timePassed = new Interval(startTime, new Instant()).toDurationMillis

    input match {
      case Input.EOF => { startTime = new Instant; Done(state, Input.EOF) }
      case Input.Empty => Cont[E, List[E]](i => step(state)(i))
      case Input.El(e) =>
        if (timePassed >= timeInMillis) { startTime = new Instant; Done(e::state, Input.Empty) }
        else Cont[E, List[E]](i => step(e::state)(i))
    }
  }

  Cont(step(List[E]()))
}

def throttledTake[T](timeInMillis: Long) = Enumeratee.grouped(throttledTakeIteratee[T](timeInMillis))
相关问题