kyo.bench.Bench.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 kyo._
import kyo.concurrent.fibers._
import kyo.ios.IOs
import zio.UIO
import org.openjdk.jmh.annotations._
import cats.effect.IO
import cats.effect.unsafe.implicits.global

@State(Scope.Benchmark)
@Fork(
    value = 1,
    jvmArgs = Array(
        // "-XX:+UnlockDiagnosticVMOptions",
        // "-Dgraal.PrintCompilation=true",
        // "-Dgraal.Log",
        // "-Dgraal.MethodFilter=kyo.scheduler.IOFiber.*",
        // "-XX:+TraceDeoptimization",
        // "-XX:+LogCompilation",
        // "-Dgraal.Dump=", //, "-Dgraal.MethodFilter=kyo2.*",
        // "-XX:+PrintCompilation", "-XX:+PrintInlining",
        // "-XX:+TraceTypeProfile",
        "-Dcats.effect.tracing.mode=DISABLED"
    )
    // jvmArgsPrepend = Array(
    //     "--enable-preview",
    //     "--add-modules=jdk.incubator.concurrent"
    // )
    // jvm = "/Users/flavio.brasil/Downloads/graalvm-ce-java17-22.3.0/Contents/Home/bin/java"
)
@BenchmarkMode(Array(Mode.Throughput))
sealed abstract class Bench[T] {
  def zioBench(): UIO[T]
  def kyoBenchFiber(): T > (IOs with Fibers) = kyoBench()
  def kyoBench(): T > IOs
  def catsBench(): IO[T]
}

object Bench {

  abstract class ForkOnly[T] extends Bench[T] {

    @Benchmark
    def forkKyo(): T = IOs.run(Fibers.forkFiber(kyoBenchFiber()).flatMap(_.block))

    @Benchmark
    def forkCats(): T = IO.cede.flatMap(_ => catsBench()).unsafeRunSync()

    @Benchmark
    def forkZio(): T = zio.Unsafe.unsafe(implicit u =>
      zio.Runtime.default.unsafe.run(zio.ZIO.yieldNow.flatMap(_ => zioBench())).getOrThrow()
    )
  }

  abstract class SyncAndFork[T] extends ForkOnly[T] {

    @Benchmark
    def syncKyo(): T = IOs.run(kyoBench())

    @Benchmark
    def syncCats(): T = catsBench().unsafeRunSync()

    @Benchmark
    def syncZio(): T = zio.Unsafe.unsafe(implicit u =>
      zio.Runtime.default.unsafe.run(zioBench()).getOrThrow()
    )
  }

}