Подтвердить что ты не робот

Как создать источник, который может получать элементы позже через вызов метода?

Я хотел бы создать Source и более поздние push-элементы на нем, например:

val src = ... // create the Source here
// and then, do something like this
pushElement(x1, src)
pushElement(x2, src)

Каков рекомендуемый способ сделать это?

Спасибо!

4b9b3361

Ответ 1

Этого можно добиться тремя способами:

1. Сообщение материализации с SourceQueue

Вы можете использовать Source.queue, который материализует поток в SourceQueue:

case class Weather(zipCode : String, temperature : Double, raining : Boolean)

val bufferSize = 100

//if the buffer fills up then this strategy drops the oldest elements
//upon the arrival of a new element.
val overflowStrategy = akka.stream.OverflowStrategy.dropHead

val queue = Source.queue(bufferSize, overflowStrategy)
                  .filter(!_.raining)
                  .to(Sink foreach println)
                  .run() // in order to "keep" the queue Materialized value instead of the Sink's

queue offer Weather("02139", 32.0, true)

2. Сообщение материализации с актером

Есть аналогичный вопрос и ответ здесь, суть в том, что вы материализуете поток как ActorRef и отправляете сообщения на этот номер ссылки:

val ref = Source.actorRef[Weather](Int.MaxValue, fail)
                .filter(!_.raining)
                .to(Sink foreach println )
                .run() // in order to "keep" the ref Materialized value instead of the Sink's

ref ! Weather("02139", 32.0, true)

3. Предварительная материализация с актером

Аналогично, вы можете явно создать Актера, который содержит буфер сообщений, использовать этого Актера для создания Источника, а затем отправить эти сообщения Актера, как описано в ответе здесь:

object WeatherForwarder {
  def props : Props = Props[WeatherForwarder]
}

//see provided link for example definition
class WeatherForwarder extends Actor {...}

val actorRef = actorSystem actorOf WeatherForwarder.props 

//note the stream has not been instatiated yet
actorRef ! Weather("02139", 32.0, true) 

//stream already has 1 Weather value to process which is sitting in the 
//ActorRef internal buffer
val stream = Source(ActorPublisher[Weather](actorRef)).runWith{...}

Ответ 2

После игры и поиска хорошего решения, я наткнулся на это решение, которое является чистым, простым и работает как до, так и после материализации. fooobar.com/questions/161323/...

  val (ref: ActorRef, publisher: Publisher[Int]) =
    Source.actorRef[Int](bufferSize = 1000, OverflowStrategy.fail)
      .toMat(Sink.asPublisher(true))(Keep.both).run()

  ref ! 1 //before

  val source = Source.fromPublisher(publisher)

  ref ! 2 //before
  Thread.sleep(1000)
  ref ! 3 //before

  source.runForeach(println)

  ref ! 4 //after
  Thread.sleep(1000)
  ref ! 5 //after

Выход:

1
2
3
4
5

Ответ 3

Поскольку Akka 2.5 Source имеет метод preMaterialize.

Согласно документации, это похоже на указанный способ сделать то, что вы просите:

Существуют ситуации, в которых вам требуется материализованное значение Source, прежде чем Source подключится к остальной части графика. Это особенно полезно в случае "материализованных источников стоимости", таких как Source.queue, Source.actorRef или Source.maybe.

Ниже приведен пример того, как это будет с SourceQueue. Элементы помещаются в очередь до и после материализации, а также из Flow:

import akka.actor.ActorSystem
import akka.stream.scaladsl._
import akka.stream.{ActorMaterializer, OverflowStrategy}

implicit val system = ActorSystem("QuickStart")
implicit val materializer = ActorMaterializer()


val sourceDecl = Source.queue[String](bufferSize = 2, OverflowStrategy.backpressure)
val (sourceMat, source) = sourceDecl.preMaterialize()

// Adding element before actual materialization
sourceMat.offer("pre materialization element")

val flow = Flow[String].map { e =>
  if(!e.contains("new")) {
    // Adding elements from within the flow
    sourceMat.offer("new element generated inside the flow")
  }
  s"Processing $e"
}

// Actually materializing with 'run'
source.via(flow).to(Sink.foreach(println)).run()

// Adding element after materialization
sourceMat.offer("post materialization element")

Выход:

Processing pre materialization element
Processing post materialization element
Processing new element generated inside the flow
Processing new element generated inside the flow