重新启动连接到资源的Observable

时间:2017-01-08 21:50:52

标签: scala rx-scala

在下面的代码中,我将TCP套接字转换为Observable[Array[Byte]]

import rx.lang.scala.Observable
import rx.lang.scala.schedulers.IOScheduler

val sock = new Socket
type Bytes = Array[Byte]

lazy val s: Observable[Bytes] = Obs.using[Bytes, Socket] {
  sock.connect(new InetSocketAddress("10.0.2.2", 9002), 1000)
  sock
}(
  socket => Observable.from[Bytes] {

    val incoming = socket.getInputStream
    val buffer = new Bytes(1024)

    Stream.continually {
      val read = incoming.read(buffer, 0, 1024)
      buffer.take(read)
    }.takeWhile(_.nonEmpty)

  },

  socket => {
    println("Socket disposed")
    socket.close
    s.retry // Does not work
  })
  .subscribeOn(IOScheduler.apply)

s.subscribe(bytes => println(new String(bytes, "UTF-8")), println)

与远程服务器的连接可能随时中断,在这种情况下,我希望Observable尝试自动重新连接,但s.retry不执行任何操作。我怎样才能做到这一点?也可以这样做"内部"当前Observable没有创建新的并重新订阅?

1 个答案:

答案 0 :(得分:1)

您希望在每个新订阅上设置新的套接字连接。对于(A)SyncOnSubscribe,这是最简单的,从版本0.26.5移植到RxScala。您可以使用这种可观察的正常错误控制方法,例如.retry

这样的事情:

val socketObservable: Observable[Byte] = Observable.create(SyncOnSubscribe.singleState(
  generator = () =>
    sock
      .connect(new InetSocketAddress("10.0.2.2", 9002), 1000)
      .getInputStream
)(next = is => Try(is.read()) match {
    case Success(-1) => Notification.OnCompleted()
    case Success(byte) => Notification.OnNext(byte)
    case Failure(e) => Notification.OnError(e)
  },
  onUnsubscribe = is => Try(is.close)
)

注意:这一次读取一个字节并且效率不高。您可以使用ASyncOnSubscribe或使您的observable的每个事件都是一个字节数组来改进这一点。

注意:这是一个cold observable,它将为每个订阅者创建一个新的套接字。例如,这将打开2个插座:

socketObservable.foreach(b => System.out.print(b))
socketObservable.buffer(1024).foreach(kiloByte => System.out.println(kiloByte))

如果这不是您想要的,可以将其变为热门.share