change benchmarks

Author: de-shyt

benchmarks/src/jmh/kotlin/benchmarks/ChannelSinkBenchmark.kt

@@ -13,6 +13,9 @@ import kotlin.coroutines.*
 @State(Scope.Benchmark)
 @Fork(1)
 open class ChannelSinkBenchmark {
+    @Param("${Channel.RENDEZVOUS}", "${Channel.BUFFERED}")
+    var capacity: Int = 0
+
     private val tl = ThreadLocal.withInitial({ 42 })
     private val tl2 = ThreadLocal.withInitial({ 239 })
 
@@ -42,15 +45,15 @@ open class ChannelSinkBenchmark {
             .fold(0) { a, b -> a + b }
     }
 
-    private fun Channel.Factory.range(start: Int, count: Int, context: CoroutineContext) = GlobalScope.produce(context) {
+    private fun Channel.Factory.range(start: Int, count: Int, context: CoroutineContext) = GlobalScope.produce(context, capacity) {
         for (i in start until (start + count))
             send(i)
     }
 
     // Migrated from deprecated operators, are good only for stressing channels
 
     private fun <E> ReceiveChannel<E>.filter(context: CoroutineContext = Dispatchers.Unconfined, predicate: suspend (E) -> Boolean): ReceiveChannel<E> =
-        GlobalScope.produce(context, onCompletion = { cancel() }) {
+        GlobalScope.produce(context, capacity, onCompletion = { cancel() }) {
             for (e in this@filter) {
                 if (predicate(e)) send(e)
             }

benchmarks/src/jmh/kotlin/benchmarks/ChannelSinkDepthBenchmark.kt

@@ -13,6 +13,9 @@ import kotlin.coroutines.*
 @State(Scope.Benchmark)
 @Fork(2)
 open class ChannelSinkDepthBenchmark {
+    @Param("${Channel.RENDEZVOUS}", "${Channel.BUFFERED}")
+    var capacity: Int = 0
+
     private val tl = ThreadLocal.withInitial({ 42 })
 
     private val unconfinedOneElement = Dispatchers.Unconfined + tl.asContextElement()
@@ -45,7 +48,7 @@ open class ChannelSinkDepthBenchmark {
     }
 
     private fun Channel.Factory.range(start: Int, count: Int, context: CoroutineContext) =
-        GlobalScope.produce(context) {
+        GlobalScope.produce(context, capacity) {
             for (i in start until (start + count))
                 send(i)
         }
@@ -57,7 +60,7 @@ open class ChannelSinkDepthBenchmark {
         context: CoroutineContext = Dispatchers.Unconfined,
         predicate: suspend (Int) -> Boolean
     ): ReceiveChannel<Int> =
-        GlobalScope.produce(context, onCompletion = { cancel() }) {
+        GlobalScope.produce(context, capacity, onCompletion = { cancel() }) {
             deeplyNestedFilter(this, callTraceDepth, predicate)
         }
 

benchmarks/src/jmh/kotlin/benchmarks/ChannelSinkNoAllocationsBenchmark.kt

@@ -13,6 +13,9 @@ import kotlin.coroutines.*
 @State(Scope.Benchmark)
 @Fork(1)
 open class ChannelSinkNoAllocationsBenchmark {
+    @Param("${Channel.RENDEZVOUS}", "${Channel.BUFFERED}")
+    var capacity: Int = 0
+
     private val unconfined = Dispatchers.Unconfined
 
     @Benchmark
@@ -26,7 +29,7 @@ open class ChannelSinkNoAllocationsBenchmark {
         return size
     }
 
-    private fun Channel.Factory.range(context: CoroutineContext) = GlobalScope.produce(context) {
+    private fun Channel.Factory.range(context: CoroutineContext) = GlobalScope.produce(context, capacity) {
         for (i in 0 until 100_000)
             send(Unit) // no allocations
     }