隊列中的 fs2 Stream 不消耗元素

Question

我想從將所有元素打印到控制台的隊列中創建 stream。 當前的一個片段不打印任何內容：

object TestApp extends App {

  implicit val contextShift: ContextShift[IO] = IO.contextShift(ExecutionContext.global)

  private val value: IO[(fs2.Stream[IO, Unit], String => IO[Unit], () => IO[Unit])] = for {
    queue <- Queue.noneTerminated[IO, String]
  } yield {
    val stream: fs2.Stream[IO, Unit] = queue.dequeue.map(println)

    def send(msg: String): IO[Unit] = queue.enqueue1(Some(msg))

    def close(): IO[Unit] = queue.enqueue1(None)

    (stream, send _, close _)
  }

  val (stream, send, close) = value.unsafeRunSync()

  send("msg1").unsafeRunSync()
  send("msg2").unsafeRunSync()

}

stream 創建有什么問題？

Answer 1

在您的示例中，您只是創建了一個隊列並創建了 stream 的描述。 為了運行 stream，您需要調用compile這將公開幾個方法，這將允許運行和使用 steam 的值，例如toList 、 fold或drain

在您的情況下，您對值並不真正感興趣，因為您只想打印它們，所以您應該使用drain ：

implicit val contextShift: ContextShift[IO] = IO.contextShift(ExecutionContext.global)
implicit val timer: Timer[IO] = IO.timer(ExecutionContext.global)

private val value: IO[(fs2.Stream[IO, Unit], String => IO[Unit], () => IO[Unit])] = for {
  queue <- Queue.noneTerminated[IO, String]
} yield {
  //I changed map to evalMap, since printing is effect and should be wrapped in IO
  val stream: fs2.Stream[IO, Unit] = queue.dequeue.evalMap(v => IO(println(v)))

  def send(msg: String): IO[Unit] = queue.enqueue1(Some(msg))

  def close(): IO[Unit] = queue.enqueue1(None)

  (stream, send _, close _)
}

val (stream, send, close) = value.unsafeRunSync()

send("msg1").unsafeRunSync()
send("msg2").unsafeRunSync()
//Closing of stream will be delayed by 5s and run in separate fiber
close().delayBy(5.seconds).start.unsafeRunSync()

//steam would block here until it's closed
stream.compile.drain.unsafeRunSync()