Question

我想创建自定义StatefulStage，它应该像groupBy方法一样工作并发出Source[A, Unit]元素，但我不了解如何创建Source[A, Unit]的实例和将传入元素推送到它。这是存根：

class GroupBy[A, Mat]() extends StatefulStage[A, Source[A, Unit]] {
  override def initial: StageState[A, Source[A, Unit]] = new StageState[A, Source[A, Unit]] {
    override def onPush(elem: A, ctx: Context[Source[A, Unit]]): SyncDirective = {
      val source: Source[A, Unit] = ... // Need to create source here

      // and push `elem` to `source` here

      emit(List(source).iterator, ctx)
    }
  }
}

您可以使用以下代码段来测试GroupBy流（它应该从创建的流中打印事件）：

case class Tick()
case class Event(timestamp: Long, sessionUid: String, traffic: Int)

implicit val system = ActorSystem()
import system.dispatcher

implicit val materializer = ActorMaterializer()

var rnd = Random
rnd.setSeed(1)

val eventsSource = Source
  .tick(FiniteDuration(0, SECONDS), FiniteDuration(1, SECONDS), () => Tick)
  .map {
    case _ => Event(System.currentTimeMillis / 1000, s"session-${rnd.nextInt(5)}", rnd.nextInt(10) * 10)
  }

val flow = Flow[Event]
  .transform(() => new GroupByUntil)
  .map {
    (source) => source.runForeach(println)
  }

eventsSource
  .via(flow)
  .runWith(Sink.ignore)
  .onComplete(_ => system.shutdown())

有人能解释我怎么做吗？

更新：

我根据this回答编写了以下onPush方法，但它没有打印事件。据我所知，只有当它作为流程的一部分运行时才能将元素推送到源，但我想在测试片段中运行GroupBy之外的流。如果我在此GroupBy中运行流程，则会处理事件并将其发送到Sink.ignore。我认为这是我的测试片段没有打印事件的原因。

override def onPush(elem: A, ctx: Context[Source[A, Unit]]): SyncDirective = {
  val source: Source[A, ActorRef] = Source.actorRef[A](1000, OverflowStrategy.fail)
  val flow = Flow[A].to(Sink.ignore).runWith(source)

  flow ! elem

  emit(List(source.asInstanceOf[Source[A, Unit]]).iterator, ctx)
}

那么，如何修复呢？

如何手动创建Source并将元素推送到它？

0 个答案: