scala - FS2 流运行直到 InputStream 末尾

Question

我对 FS2 很陌生，需要一些有关设计的帮助。我正在尝试设计一个流，它将从底层的 InputStream 中提取 block ，直到结束。这是我尝试过的:

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
  }
}

程序打印唯一的第一个 block 。这是合理的。但我想找到一种方法来“指示”哪里停止阅读以及哪里不停止。我的意思是继续调用 read(is) 直到结束。有办法实现吗？

我也尝试过repeatEval(read(is))，但它一直在读取......我需要介于两者之间的东西。

Answer 1

使用fs2.io.readInputStream或fs2.io.readInputStreamAsync 。前者阻塞当前线程；后者会阻塞 ExecutionContext 中的线程。例如:

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