Add different permutations

Author: clementetb

benchmarks/androidApp/src/androidTest/kotlin/io/realm/kotlin/benchmarks/android/OpenRealmTestsGenerated.kt

@@ -17,10 +17,9 @@ package io.realm.kotlin.benchmarks.android
 
 import androidx.benchmark.junit4.BenchmarkRule
 import androidx.benchmark.junit4.measureRepeated
-import io.realm.generated.tenStringObjectClasses
+import io.realm.generated.openCloseRealmClassesMap
 import io.realm.kotlin.Realm
 import io.realm.kotlin.RealmConfiguration
-import io.realm.kotlin.benchmarks.SchemaSize
 import io.realm.kotlin.types.RealmObject
 import org.junit.After
 import org.junit.Before
@@ -31,13 +30,22 @@ import org.junit.runners.Parameterized
 import kotlin.reflect.KClass
 
 @RunWith(Parameterized::class)
-class OpenRealmTestsGenerated(private val schemaSize: Int) {
+class OpenRealmTestsGenerated(
+    private val className: String,
+    private val schemaSize: Int,
+) {
 
     companion object {
         @JvmStatic
-        @Parameterized.Parameters(name = "schema-size-{0}")
-        fun initParameters(): Collection<Int> {
-            return setOf(1, 10, 100)
+        @Parameterized.Parameters(name = "{0}-schema-size-{1}")
+        fun initParameters(): Collection<Array<*>> {
+            val schemaSizes = listOf(1, 10, 100)
+
+            return openCloseRealmClassesMap.keys
+                .flatMap {className ->
+                    schemaSizes.map { schemaSize -> arrayOf(className, schemaSize) }
+                }
+                .toList()
         }
     }
 
@@ -49,7 +57,11 @@ class OpenRealmTestsGenerated(private val schemaSize: Int) {
 
     @Before
     fun setUp() {
-        val schema: Set<KClass<out RealmObject>> = tenStringObjectClasses.subList(0, schemaSize).toSet()
+        val schema: Set<KClass<out RealmObject>> =
+            openCloseRealmClassesMap[className]!!
+                .subList(0, schemaSize)
+                .toSet()
+
         config = RealmConfiguration.Builder(schema)
             .directory("./build/benchmark-realms")
             .build()

benchmarks/shared/build.gradle.kts

@@ -1,5 +1,6 @@
+import com.android.build.gradle.internal.tasks.factory.dependsOn
 import io.realm.ClassGeneratorSpec
-import io.realm.generate
+import io.realm.BenchmarkClassSuite
 
 plugins {
     kotlin("multiplatform")
@@ -27,7 +28,7 @@ kotlin {
 //            baseName = "shared"
 //        }
 //    }
-    
+
     sourceSets {
         val commonMain by getting {
             dependencies {
@@ -60,25 +61,54 @@ kotlin {
     }
 }
 
-abstract class ClassGenerator : DefaultTask() {
-    @TaskAction
-    fun generate() {
-        val output = project.file("./src/commonMain/kotlin/")
+// Create a task using the task type
+val genClassesTask = tasks.create("classGen") {
+    val output = project.file("./src/commonMain/kotlin/")
 
-        // Clear out any previous contents
-        project.file("./src/commonMain/kotlin/io/realm/generated").deleteRecursively()
+    // Clear out any previous contents
+    project.file("./src/commonMain/kotlin/io/realm/generated").deleteRecursively()
 
-        ClassGeneratorSpec(
+    BenchmarkClassSuite(
+        name = "openCloseRealm",
+        packageName = "io.realm.generated",
+        output = output,
+    ) {
+        addClassGeneratorSpec(
+            classCount = 100,
+            className = "OneString",
+            stringFieldCount = 1,
+        )
+        addClassGeneratorSpec(
+            classCount = 100,
+            className = "OneStringRealmList",
+            stringRealmListCount = 1,
+        )
+        addClassGeneratorSpec(
+            classCount = 100,
+            className = "TenStrings",
+            stringFieldCount = 10,
+        )
+        addClassGeneratorSpec(
             classCount = 100,
-            packageName = "io.realm.generated",
-            className = "TenStringObject",
-            stringFieldCount = 10
-        ).generate(output)
+            className = "TenStringRealmLists",
+            stringRealmListCount = 10,
+        )
+        addClassGeneratorSpec(
+            classCount = 100,
+            className = "HundredStrings",
+            stringFieldCount = 100,
+        )
+        addClassGeneratorSpec(
+            classCount = 100,
+            className = "HundredStringRealmLists",
+            stringRealmListCount = 100,
+        )
     }
 }
 
-// Create a task using the task type
-tasks.register<ClassGenerator>("classGen")
+afterEvaluate {
+    tasks.named("assemble").dependsOn(genClassesTask.name)
+}
 
 android {
     compileSdk = Versions.Android.compileSdkVersion