scala - 无法在 Akka Stream 中使用 GraphStage 类运行 SourceShape

Question

我正在尝试使用 GraphStage 构造创建 Redis Akka 流源。这个想法是，每当我从 subscribe 方法获得更新时，我都会将其推送到下一个组件。此外，如果没有拉信号，该组件应该背压。这是代码:

class SubscriberSourceShape(channel: String, subscriber: Subscriber) 
    extends GraphStage[SourceShape[String]] {

  private val outlet: Outlet[String] = Outlet("SubscriberSource.Out")

  override def shape: SourceShape[String] = SourceShape(outlet)

  override def createLogic(inheritedAttributes: Attributes): GraphStageLogic = {
    new GraphStageLogic(shape) {
      val callback = getAsyncCallback((msg: String) => push(outlet, msg)
      val handler = (msg: String) => callback.invoke(msg)

      override def preStart(): Unit = subscriber.subscribe(channel)(handler)
    }
  }
}

但是，当我使用简单的接收器运行此程序时，我收到此错误:

阶段 [akka.http.impl.engine.server.HttpServerBluePrint$ProtocolSwitchStage@58344854] 中出现错误:阶段 [...SubscriberSourceShape@6a9193dd] 中没有为输出端口 [RedisSubscriberSource.Out(1414886352) 定义处理程序。所有入口和导出必须在图形阶段逻辑的构造函数中使用 setHandler 分配一个处理程序。

出了什么问题？

Answer 1

您收到此错误是因为您尚未为 Source 设置任何输出处理程序，因此当下游组件(Flow 或 Sink)向此 Source 发送拉取信号时，没有处理程序来处理拉取信号。

您可以添加一个 OutputHandler 来消除错误。将 onPull 方法留空，因为您是在 asyncCallback 上生成元素。只需将其添加到 GraphStageLogic 主体中即可:

      setHandler(outlet, new OutHandler {
        override def onPull(): Unit = {}
      })