• Home
• io.github.keirlawson
• fs2-progress_3
• 0.2.0
• source code
• ProgressBar.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

fs2.progress.ProgressBar.scala Maven / Gradle / Ivy

package example

import cats.effect.IO
import me.tongfei.progressbar.{ProgressBar => pBar}
import fs2.{Stream, Pipe}
import cats.implicits._
import cats.effect.kernel.Resource
import cats.effect.kernel.Sync

trait ProgressBar[F[_]] {

  def stepTo(n: Long): F[Unit]

  def pipe[A]: Pipe[F, A, A]
  
}

object ProgressBar {
  def resource[F[_] : Sync](name: String, initialMax: Long): Resource[F, ProgressBar[F]] = Resource.fromAutoCloseable(Sync[F].delay(new pBar(name, initialMax))).map { pb =>
    new ProgressBar[F] {

      override def pipe[A]: Pipe[F,A,A] = { input =>
        input.zipWithIndex.evalTap { case (_, idx) =>
          stepTo(idx)
        }.map(_._1)
      }


      override def stepTo(n: Long): F[Unit] = Sync[F].delay(pb.stepTo(n))

    }
  }

  def stream[F[_] : Sync](name: String, initialMax: Long): Stream[F, ProgressBar[F]] = {
    Stream.resource(resource[F](name, initialMax))
  }

}