作为handleMessagesWithSinkSource

时间:2016-04-23 20:04:01

标签: scala websocket stream akka

我是AKKA Streams的新手。 (使用Akka v 2.4.4) 我正在尝试创建一个可以将新通知推送到订阅客户端的Websocket。我的策略是实现一个ActorPublisher,我稍后可以发送消息,然后将其推送到客户端。

开始时我复制了一个ActorPublisher的例子:

case class Tick()

class TickActor extends ActorPublisher[Int] {
  import scala.concurrent.duration._

  implicit val ec = context.dispatcher

  val tick = context.system.scheduler.schedule(1 second, 1 second, self, `Tick())`

  var cnt = 0
  var buffer = Vector.empty[Int]

  override def receive: Receive = {
    case Tick() => {
      cnt = cnt + 1
      if (buffer.isEmpty && totalDemand > 0) {
        onNext(cnt)
      }
      else {
        buffer :+= cnt
        if (totalDemand > 0) {
          val (use,keep) = buffer.splitAt(totalDemand.toInt)
          buffer = keep
          use foreach onNext
        }
      }
    }
  }

  override def postStop() = tick.cancel()
}

我的问题是我不知道如何使用它作为来源。

我尝试了以下内容:

val source: Source[Strict, ActorRef] = Source.actorPublisher(Props[TickActor]).map(i => TextMessage(i.toString))
  optionalHeaderValueByType[akka.http.scaladsl.model.ws.UpgradeToWebSocket]() {
    case Some(upgrade) =>
      complete(
        upgrade.handleMessagesWithSinkSource(Sink.ignore,source))
    case None =>
      reject(akka.http.scaladsl.server.ExpectedWebSocketRequestRejection)
  }

但是当我与客户端连接时,我得到以下ClassCastException:java.lang.ClassCastException:java.lang.Integer无法强制转换为scala.runtime.Nothing $

如果我将Source更改为:

val src: Source[Strict, NotUsed] = Source.fromIterator(() => Iterator.continually(ThreadLocalRandom.current.nextInt()))
      .filter(i => i > 0 && i % 2 == 0).map(i => TextMessage(i.toString))

运行得很好。

我努力连接点,所以希望你能带领我走向正确的方向。

1 个答案:

答案 0 :(得分:0)

我尝试了你的例子并且能够重现这个问题。我只做了一次修改就解决了这个问题。这是添加类型参数,它现在有意义,因为在akka流中的某个地方,有一个类似的代码 elem.asInstanceOf[T]。因此,当actorPublisher缺少类型时,该类型将被推断为Nothing

val source = Source.actorPublisher[Int](Props[TickActor]).map(i => TextMessage(i.toString))