FS2流一直运行到InputStream结束

时间:2018-05-24 18:07:47

标签: scala functional-programming scala-cats fs2

我对FS2很新,需要一些关于设计的帮助。我正在尝试设计一个流,它将从底层的InputStream中拉出块直到它结束。这是我试过的:

import java.io.{File, FileInputStream, InputStream}

import cats.effect.IO
import cats.effect.IO._

object Fs2 {

  def main(args: Array[String]): Unit = {
    val is = new FileInputStream(new File("/tmp/my-file.mf"))
    val stream = fs2.Stream.eval(read(is))
    stream.compile.drain.unsafeRunSync()
  }

  def read(is: InputStream): IO[Array[Byte]] = IO {
    val buf = new Array[Byte](4096)
    is.read(buf)
    println(new String(buf))
    buf
  }
}

程序打印出唯一的第一个块。这是合理的。但我想找到一种方法来“发出信号”,在哪里停止阅读以及在哪里不停止。我的意思是继续打电话给read(is)直到它结束。有没有办法实现这个目标?

我也试过repeatEval(read(is)),但它一直在阅读......我需要介于两者之间。

1 个答案:

答案 0 :(得分:4)

使用fs2.io.readInputStreamfs2.io.readInputStreamAsync。前者阻止当前线程;后者阻止ExecutionContext中的一个线程。例如:

val is: InputStream = new FileInputStream(new File("/tmp/my-file.mf"))
val stream = fs2.io.readInputStreamAsync(IO(is), 128)