kyo.bench.EnqueueDequeueBench.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kyo-bench_3 Show documentation

kyo-bench

There is a newer version: 3.0.7

package kyo.bench

import org.openjdk.jmh.annotations._
import cats.effect.IO
import kyo._
import kyo.ios._
import zio.{ZIO, UIO}
import java.util.concurrent.Executors
import kyo.concurrent.fibers._
import kyo.concurrent.channels._
import kyo.concurrent.Access

import kyo.bench.Bench
import java.util.concurrent.atomic.AtomicInteger

class EnqueueDequeueBench extends Bench.ForkOnly[Unit] {

  val depth = 10000

  def catsBench(): IO[Unit] = {
    import cats.effect.std.Queue

    def loop(q: Queue[IO, Unit], i: Int): IO[Unit] =
      if (i >= depth)
        IO.unit
      else
        q.offer(()).flatMap(_ => q.take.flatMap(_ => loop(q, i + 1)))

    Queue.bounded[IO, Unit](1).flatMap(loop(_, 0))
  }

  def kyoBench() = Fibers.runBlocking(Fibers.fork(kyoBenchFiber()))

  override def kyoBenchFiber(): Unit > (IOs with Fibers) = {

    def loop(c: Channel[Unit], i: Int): Unit > (IOs with Fibers) =
      if (i >= depth)
        IOs.unit
      else
        c.put(()).flatMap(_ => c.take.flatMap(_ => loop(c, i + 1)))

    Channels.init[Unit](1, Access.Spsc).flatMap(loop(_, 0))
  }

  def zioBench(): UIO[Unit] = {
    import zio.Queue

    def loop(q: Queue[Unit], i: Int): ZIO[Any, Nothing, Unit] =
      if (i >= depth)
        ZIO.unit
      else
        q.offer(()).flatMap(_ => q.take.flatMap(_ => loop(q, i + 1)))

    Queue.bounded[Unit](1).flatMap(loop(_, 0))
  }
}