Refactor map benchmark generators

Author: Abduqodiri Qurbonzoda

benchmarks-libraries/src/main/kotlin/generators/benchmarkSourceGenerator.kt

@@ -116,21 +116,21 @@ private val listBuilderImpls = listOf(
 )
 
 private val mapImpls = listOf(
-        MapKotlinBenchmark(),
-        MapKotlinOrderedBenchmark(),
-        MapCapsuleBenchmark(),
-        MapPaguroBenchmark(),
-        MapPaguroSortedBenchmark(),
-        MapCyclopsBenchmark(),
-        MapCyclopsOrderedBenchmark(),
-        MapCyclopsTrieBenchmark(),
-        MapClojureBenchmark(),
-        MapClojureSortedBenchmark(),
-        MapScalaBenchmark(),
-        MapScalaSortedBenchmark(),
-        MapVavrBenchmark(),
-        MapVavrSortedBenchmark(),
-        MapVavrOrderedBenchmark()
+        KotlinMapImplementation,
+        KotlinOrderedMapImplementation,
+        CapsuleMapImplementation,
+        PaguroMapImplementation,
+        PaguroSortedMapImplementation,
+        CyclopsMapImplementation,
+        CyclopsOrderedMapImplementation,
+        CyclopsTrieMapImplementation,
+        ClojureMapImplementation,
+        ClojureSortedMapImplementation,
+        ScalaMapImplementation,
+        ScalaSortedMapImplementation,
+        VavrMapImplementation,
+        VavrSortedMapImplementation,
+        VavrOrderedMapImplementation
 )
 private val mapBuilderImpls = listOf(
         MapBuilderKotlinBenchmark(),
@@ -195,12 +195,19 @@ fun generateBenchmarks() {
             listBuilderImpls.filterIsInstance<ListBuilderSetBenchmark>().map { ListBuilderSetBenchmarkGenerator(it) }
     )*/
 
-    val mapBenchmarks = listOf(
+    val mapBenchmarks = mapImpls.map { listOf(
+            MapGetBenchmarkGenerator(it),
+            MapIterateBenchmarkGenerator(it),
+            MapPutBenchmarkGenerator(it),
+            MapRemoveBenchmarkGenerator(it),
+            MapUtilsGenerator(it)
+    ) }
+    /*listOf(
             mapImpls.filterIsInstance<MapGetBenchmark>().map { MapGetBenchmarkGenerator(it) },
             mapImpls.filterIsInstance<MapIterateBenchmark>().map { MapIterateBenchmarkGenerator(it) },
             mapImpls.filterIsInstance<MapPutBenchmark>().map { MapPutBenchmarkGenerator(it) },
             mapImpls.filterIsInstance<MapRemoveBenchmark>().map { MapRemoveBenchmarkGenerator(it) }
-    )
+    )*/
     val mapBuilderBenchmarks = listOf(
             mapBuilderImpls.filterIsInstance<MapBuilderGetBenchmark>().map { MapBuilderGetBenchmarkGenerator(it) },
             mapBuilderImpls.filterIsInstance<MapBuilderIterateBenchmark>().map { MapBuilderIterateBenchmarkGenerator(it) },
@@ -235,7 +242,6 @@ fun generateBenchmarks() {
 }
 
 fun generateUtils() {
-    val mapUtils = mapImpls.filterIsInstance<MapBenchmarkUtils>().map { MapUtilsGenerator(it) }
     val mapBuilderUtils = mapBuilderImpls.filterIsInstance<MapBuilderBenchmarkUtils>().map { MapBuilderUtilsGenerator(it) }
 
     val setUtils = setImpls.filterIsInstance<SetBenchmarkUtils>().map { SetUtilsGenerator(it) }
@@ -246,7 +252,7 @@ fun generateUtils() {
             CommonUtilsGenerator()
     )
 
-    val utils = mapUtils + mapBuilderUtils + setUtils + setBuilderUtils + commonUtils
+    val utils = mapBuilderUtils + setUtils + setBuilderUtils + commonUtils
 
     utils.forEach { util ->
         val path = util.getPackage().replace('.', '/') + "/" + util.outputFileName + ".kt"

benchmarks-libraries/src/main/kotlin/generators/immutableMap/getGenerator.kt

@@ -19,13 +19,7 @@ package generators.immutableMap
 import generators.BenchmarkSourceGenerator
 import java.io.PrintWriter
 
-interface MapGetBenchmark {
-    val packageName: String
-    fun emptyOf(K: String, V: String): String
-    val getOperation: String
-}
-
-class MapGetBenchmarkGenerator(private val impl: MapGetBenchmark) : BenchmarkSourceGenerator() {
+class MapGetBenchmarkGenerator(private val impl: MapImplementation) : BenchmarkSourceGenerator() {
     override val outputFileName: String = "Get"
 
     override fun getPackage(): String {
@@ -43,8 +37,8 @@ open class Get {
     @Param(ASCENDING_HASH_CODE, RANDOM_HASH_CODE, COLLISION_HASH_CODE, NON_EXISTING_HASH_CODE)
     var hashCodeType = ""
 
-    private var keys = listOf<IntWrapper>()
-    private var persistentMap = ${impl.emptyOf("IntWrapper", "String")}
+    private var keys = listOf<$mapKeyType>()
+    private var persistentMap = ${impl.empty()}
 
     @Setup(Level.Trial)
     fun prepare() {
@@ -59,7 +53,7 @@ open class Get {
     @Benchmark
     fun get(bh: Blackhole) {
         repeat(times = size) { index ->
-            bh.consume(persistentMap.${impl.getOperation}(keys[index]))
+            bh.consume(${impl.getOperation("persistentMap", "keys[index]")})
         }
     }
 }

benchmarks-libraries/src/main/kotlin/generators/immutableMap/impl/capsule.kt

@@ -18,22 +18,24 @@ package generators.immutableMap.impl
 
 import generators.immutableMap.*
 
-class MapCapsuleBenchmark:
-        MapGetBenchmark,
-        MapIterateBenchmark,
-        MapPutBenchmark,
-        MapRemoveBenchmark,
-        MapBenchmarkUtils
-{
-    override val packageName: String = "capsule"
+object CapsuleMapImplementation: MapImplementation {
+    override val packageName: String
+            = "capsule"
 
-    override fun mapType(K: String, V: String): String = "io.usethesource.capsule.Map.Immutable<$K, $V>"
-    override fun emptyOf(K: String, V: String): String = "io.usethesource.capsule.core.PersistentTrieMap.of<$K, $V>()"
+    override fun type(): String
+            = "io.usethesource.capsule.Map.Immutable<$mapKeyType, $mapValueType>"
+    override fun empty(): String
+            = "io.usethesource.capsule.core.PersistentTrieMap.of<$mapKeyType, $mapValueType>()"
 
-    override val getOperation: String = "get"
-    override fun putOperation(map: String, key: String, value: String): String = "$map.__put($key, $value)"
-    override fun removeOperation(map: String, key: String): String = "$map.__remove($key)"
+    override fun keysOperation(map: String): String
+            = "$map.keys"
+    override fun valuesOperation(map: String): String
+            = "$map.values"
 
-    override val keys: String = "keys"
-    override val values: String = "values"
+    override fun getOperation(map: String, key: String): String
+            = "$map.get($key)"
+    override fun putOperation(map: String, key: String, value: String): String
+            = "$map.__put($key, $value)"
+    override fun removeOperation(map: String, key: String): String
+            = "$map.__remove($key)"
 }

benchmarks-libraries/src/main/kotlin/generators/immutableMap/impl/clojure.kt

@@ -18,22 +18,24 @@ package generators.immutableMap.impl
 
 import generators.immutableMap.*
 
-class MapClojureBenchmark:
-        MapGetBenchmark,
-        MapIterateBenchmark,
-        MapPutBenchmark,
-        MapRemoveBenchmark,
-        MapBenchmarkUtils
-{
-    override val packageName: String = "clojure"
+object ClojureMapImplementation: MapImplementation {
+    override val packageName: String
+            = "clojure"
 
-    override fun mapType(K: String, V: String): String = "clojure.lang.PersistentHashMap"
-    override fun emptyOf(K: String, V: String): String = "clojure.lang.PersistentHashMap.EMPTY"
+    override fun type(): String
+            = "clojure.lang.PersistentHashMap"
+    override fun empty(): String
+            = "clojure.lang.PersistentHashMap.EMPTY"
 
-    override val getOperation: String = "valAt"
-    override fun putOperation(map: String, key: String, value: String): String = "$map.assoc($key, $value) as clojure.lang.PersistentHashMap"
-    override fun removeOperation(map: String, key: String): String = "$map.without($key) as clojure.lang.PersistentHashMap"
+    override fun keysOperation(map: String): String
+            = "$map.keys"
+    override fun valuesOperation(map: String): String
+            = "$map.values"
 
-    override val keys: String = "keys"
-    override val values: String = "values"
+    override fun getOperation(map: String, key: String): String
+            = "$map.valAt($key)"
+    override fun putOperation(map: String, key: String, value: String): String
+            = "$map.assoc($key, $value) as clojure.lang.PersistentHashMap"
+    override fun removeOperation(map: String, key: String): String
+            = "$map.without($key) as clojure.lang.PersistentHashMap"
 }

benchmarks-libraries/src/main/kotlin/generators/immutableMap/impl/clojureSorted.kt

@@ -18,22 +18,24 @@ package generators.immutableMap.impl
 
 import generators.immutableMap.*
 
-class MapClojureSortedBenchmark:
-        MapGetBenchmark,
-        MapIterateBenchmark,
-        MapPutBenchmark,
-        MapRemoveBenchmark,
-        MapBenchmarkUtils
-{
-    override val packageName: String = "clojureSorted"
+object ClojureSortedMapImplementation: MapImplementation {
+    override val packageName: String
+            = "clojureSorted"
 
-    override fun mapType(K: String, V: String): String = "clojure.lang.PersistentTreeMap"
-    override fun emptyOf(K: String, V: String): String = "clojure.lang.PersistentTreeMap.EMPTY"
+    override fun type(): String
+            = "clojure.lang.PersistentTreeMap"
+    override fun empty(): String
+            = "clojure.lang.PersistentTreeMap.EMPTY"
 
-    override val getOperation: String = "valAt"
-    override fun putOperation(map: String, key: String, value: String): String = "$map.assoc($key, $value)"
-    override fun removeOperation(map: String, key: String): String = "$map.without($key)"
+    override fun keysOperation(map: String): String
+            = "$map.keys"
+    override fun valuesOperation(map: String): String
+            = "$map.values"
 
-    override val keys: String = "keys"
-    override val values: String = "values"
+    override fun getOperation(map: String, key: String): String
+            = "$map.valAt($key)"
+    override fun putOperation(map: String, key: String, value: String): String
+            = "$map.assoc($key, $value)"
+    override fun removeOperation(map: String, key: String): String
+            = "$map.without($key)"
 }

benchmarks-libraries/src/main/kotlin/generators/immutableMap/impl/cyclops.kt

@@ -18,22 +18,24 @@ package generators.immutableMap.impl
 
 import generators.immutableMap.*
 
-class MapCyclopsBenchmark:
-        MapGetBenchmark,
-        MapIterateBenchmark,
-        MapPutBenchmark,
-        MapRemoveBenchmark,
-        MapBenchmarkUtils
-{
-    override val packageName: String = "cyclops"
+object CyclopsMapImplementation: MapImplementation {
+    override val packageName: String
+            = "cyclops"
 
-    override fun mapType(K: String, V: String): String = "cyclops.data.HashMap<$K, $V>"
-    override fun emptyOf(K: String, V: String): String = "cyclops.data.HashMap.empty<$K, $V>()"
+    override fun type(): String
+            = "cyclops.data.HashMap<$mapKeyType, $mapValueType>"
+    override fun empty(): String
+            = "cyclops.data.HashMap.empty<$mapKeyType, $mapValueType>()"
 
-    override val getOperation: String = "get"
-    override fun putOperation(map: String, key: String, value: String): String = "$map.put($key, $value)"
-    override fun removeOperation(map: String, key: String): String = "$map.remove($key)"
+    override fun keysOperation(map: String): String
+            = "$map.keys()"
+    override fun valuesOperation(map: String): String
+            = "$map.values()"
 
-    override val keys: String = "keys()"
-    override val values: String = "values()"
+    override fun getOperation(map: String, key: String): String
+            = "$map.get($key)"
+    override fun putOperation(map: String, key: String, value: String): String
+            = "$map.put($key, $value)"
+    override fun removeOperation(map: String, key: String): String
+            = "$map.remove($key)"
 }

benchmarks-libraries/src/main/kotlin/generators/immutableMap/impl/cyclopsOrdered.kt

@@ -18,22 +18,24 @@ package generators.immutableMap.impl
 
 import generators.immutableMap.*
 
-class MapCyclopsOrderedBenchmark:
-        MapGetBenchmark,
-        MapIterateBenchmark,
-        MapPutBenchmark,
-        MapRemoveBenchmark,
-        MapBenchmarkUtils
-{
-    override val packageName: String = "cyclopsOrdered"
+object CyclopsOrderedMapImplementation: MapImplementation {
+    override val packageName: String
+            = "cyclopsOrdered"
 
-    override fun mapType(K: String, V: String): String = "cyclops.data.LinkedMap<$K, $V>"
-    override fun emptyOf(K: String, V: String): String = "cyclops.data.LinkedMap.empty<$K, $V>()"
+    override fun type(): String
+            = "cyclops.data.LinkedMap<$mapKeyType, $mapValueType>"
+    override fun empty(): String
+            = "cyclops.data.LinkedMap.empty<$mapKeyType, $mapValueType>()"
 
-    override val getOperation: String = "get"
-    override fun putOperation(map: String, key: String, value: String): String = "$map.put($key, $value)"
-    override fun removeOperation(map: String, key: String): String = "$map.remove($key)"
+    override fun keysOperation(map: String): String
+            = "$map.keys()"
+    override fun valuesOperation(map: String): String
+            = "$map.values()"
 
-    override val keys: String = "keys()"
-    override val values: String = "values()"
+    override fun getOperation(map: String, key: String): String
+            = "$map.get($key)"
+    override fun putOperation(map: String, key: String, value: String): String
+            = "$map.put($key, $value)"
+    override fun removeOperation(map: String, key: String): String
+            = "$map.remove($key)"
 }

benchmarks-libraries/src/main/kotlin/generators/immutableMap/impl/cyclopsTrie.kt

@@ -18,22 +18,24 @@ package generators.immutableMap.impl
 
 import generators.immutableMap.*
 
-class MapCyclopsTrieBenchmark:
-        MapGetBenchmark,
-        MapIterateBenchmark,
-        MapPutBenchmark,
-        MapRemoveBenchmark,
-        MapBenchmarkUtils
-{
-    override val packageName: String = "cyclopsTrie"
+object CyclopsTrieMapImplementation: MapImplementation {
+    override val packageName: String
+            = "cyclopsTrie"
 
-    override fun mapType(K: String, V: String): String = "cyclops.data.TrieMap<$K, $V>"
-    override fun emptyOf(K: String, V: String): String = "cyclops.data.TrieMap.empty<$K, $V>()"
+    override fun type(): String
+            = "cyclops.data.TrieMap<$mapKeyType, $mapValueType>"
+    override fun empty(): String
+            = "cyclops.data.TrieMap.empty<$mapKeyType, $mapValueType>()"
 
-    override val getOperation: String = "get"
-    override fun putOperation(map: String, key: String, value: String): String = "$map.put($key, $value)"
-    override fun removeOperation(map: String, key: String): String = "$map.remove($key)"
+    override fun keysOperation(map: String): String
+            = "$map.keys()"
+    override fun valuesOperation(map: String): String
+            = "$map.values()"
 
-    override val keys: String = "keys()"
-    override val values: String = "values()"
+    override fun getOperation(map: String, key: String): String
+            = "$map.get($key)"
+    override fun putOperation(map: String, key: String, value: String): String
+            = "$map.put($key, $value)"
+    override fun removeOperation(map: String, key: String): String
+            = "$map.remove($key)"
 }

benchmarks-libraries/src/main/kotlin/generators/immutableMap/impl/kotlin.kt

@@ -18,22 +18,24 @@ package generators.immutableMap.impl
 
 import generators.immutableMap.*
 
-class MapKotlinBenchmark:
-        MapGetBenchmark,
-        MapIterateBenchmark,
-        MapPutBenchmark,
-        MapRemoveBenchmark,
-        MapBenchmarkUtils
-{
-    override val packageName: String = "kotlin"
+object KotlinMapImplementation: MapImplementation {
+    override val packageName: String
+            = "kotlin"
 
-    override fun mapType(K: String, V: String): String = "kotlinx.collections.immutable.PersistentMap<$K, $V>"
-    override fun emptyOf(K: String, V: String): String = "kotlinx.collections.immutable.persistentHashMapOf<$K, $V>()"
+    override fun type(): String
+            = "kotlinx.collections.immutable.PersistentMap<$mapKeyType, $mapValueType>"
+    override fun empty(): String
+            = "kotlinx.collections.immutable.persistentHashMapOf<$mapKeyType, $mapValueType>()"
 
-    override val getOperation: String = "get"
-    override fun putOperation(map: String, key: String, value: String): String = "$map.put($key, $value)"
-    override fun removeOperation(map: String, key: String): String = "$map.remove($key)"
+    override fun keysOperation(map: String): String
+            = "$map.keys"
+    override fun valuesOperation(map: String): String
+            = "$map.values"
 
-    override val keys: String = "keys"
-    override val values: String = "values"
+    override fun getOperation(map: String, key: String): String
+            = "$map.get($key)"
+    override fun putOperation(map: String, key: String, value: String): String
+            = "$map.put($key, $value)"
+    override fun removeOperation(map: String, key: String): String
+            = "$map.remove($key)"
 }