使用Scala节流器更改每条消息之间的时间间隔

使用Scala节流器更改每条消息之间的时间间隔,scala,akka,akka-stream,Scala,Akka,Akka Stream,我正在学习Akka,并编写了以下代码来限制发送给TradeActionactor的消息数量。每秒最多发送3条消息。是否可以修改节流阀,以便设置每条消息之间的时间?例如,message1和message2之间延迟2秒,message2和message3之间延迟1秒 import akka.NotUsed import akka.actor.{Actor, ActorRef, ActorSystem, Props} import akka.stream.{ActorMaterializer, Ove

我正在学习Akka,并编写了以下代码来限制发送给
TradeAction
actor的消息数量。每秒最多发送3条消息。是否可以修改节流阀,以便设置每条消息之间的时间?例如,message1和message2之间延迟2秒,message2和message3之间延迟1秒

import akka.NotUsed
import akka.actor.{Actor, ActorRef, ActorSystem, Props}
import akka.stream.{ActorMaterializer, OverflowStrategy}
import akka.stream.scaladsl.{Sink, Source}
import systemconfig.ActorSystemConfig

import scala.concurrent.duration.DurationInt

case class Action(side: String)

object PlaceTrade {

  val actorSystem = ActorSystem("firstActorSystem")
  println(actorSystem.name)

  object TradeAction {
    def props(action : String) = Props(new TradeAction(action))
  }

  class TradeAction(actorName: String) extends Actor {
    override def receive: Receive = {
      case "Buy" => {
        val r = requests.get("http://www.google.com")
        println("r status code is "+r.statusCode)
        println("Buy")
        println("")
      }
      case "Sell" => {
        val r = requests.get("http://www.google.com")
        println("r status code is "+r.statusCode)
        println("Sell")
        println("")
      }
      case _ =>
    }
  }

  implicit val materializer = ActorMaterializer.create(ActorSystemConfig.getActorSystem)

  def getThrottler(ac: ActorRef) = Source.actorRef(bufferSize = 1000, OverflowStrategy.dropNew)
    .throttle(1, 3.second)
    .to(Sink.actorRef(ac, NotUsed))
    .run()

  def main(args: Array[String]): Unit = {
    val tradeAction = actorSystem.actorOf(TradeAction.props("TradeAction"))
    val throttler = getThrottler(tradeAction)

    val l = List(Action("Buy"),Action("Buy"),Action("Buy"),Action("Sell"))
    l.foreach(action => {
      throttler ! action.side
    })
  }


}

您可以通过
delayWith
来实现这一点,它允许(潜在有状态的)方法定义此元素的延迟时间(无需对元素重新排序),例如:

import akka.stream.scaladsl.{ DelayStrategy, DelayOverflowStrategy }
import scala.concurrent.duration.FiniteDuration

def decliningDelay(): DelayStrategy[Any] =
  new DelayStrategy {
    var nextDelaySeconds: Option[Int] = None

    def nextDelay(elem: Any): FiniteDuration =
      nextDelaySeconds match {
        case None =>
          nextDelaySeconds = 2
          0.seconds
        case Some(delay) if delay > 0 =>
          nextDelaySeconds = delay - 1
          delay.seconds
        case _ => 0.seconds
      }
  }

def getThrottler(ac: ActorRef): ActorRef =
  Source.actorRef(bufferSize = 1000, OverflowStrategy.dropNew)
    .delayWith(() => decliningDelay(), DelayOverflowStrategy.backpressure)
    .to(Sink.actorRef(ac, NotUsed))
    .run()

您可以将
delayWith
throttle
组合使用;我可能会把
delayWith
放在
throttle

之前,你的节流器实际上每3秒最多只发送1条消息,而不是每秒最多发送3条消息。