Add Gradle project to run regression benchmarks

This project will contain all required Gradle infrastructure to run regression benchmark scripts. ^KT-49921 In Progress
2021-12-20 14:41:09 +01:00
parent 7418d3c898
commit d1f6faf236
3 changed files with 91 additions and 0 deletions
@@ -0,0 +1,70 @@
+plugins {
+    `java-base`
+}
+
+val compilerClasspath = configurations.create("compilerClasspath") {
+    isCanBeResolved = true
+    isCanBeConsumed = false
+}
+val scriptsClasspath = configurations.create("scriptsClasspath") {
+    isCanBeResolved = true
+    isCanBeConsumed = false
+}
+
+dependencies {
+    compilerClasspath.name(project(":kotlin-compiler-embeddable"))
+    compilerClasspath.name(project(":kotlin-scripting-compiler-embeddable"))
+
+    scriptsClasspath.name(project(":gradle:regression-benchmark-templates"))
+    scriptsClasspath.name(project(":kotlin-script-runtime"))
+    scriptsClasspath.name(kotlinStdlib())
+}
+
+val service = project.extensions.getByType<JavaToolchainService>()
+
+abstract class ScriptArgumentProvider @Inject constructor(
+    layout: ProjectLayout
+) : CommandLineArgumentProvider {
+    @get:Classpath
+    abstract val scriptClasspath: ConfigurableFileCollection
+
+    @get:Input
+    abstract val script: Property<String>
+
+    @get:OutputDirectory
+    val scriptOutputDirectories: Provider<Directory> = layout.buildDirectory.dir("benchmark-script")
+
+    override fun asArguments(): Iterable<String> {
+        return listOf(
+            "-no-reflect",
+            "-no-stdlib",
+            "-classpath", scriptClasspath.asFileTree.files.joinToString(separator = File.pathSeparator),
+            "-script", "benchmarkScripts/${script.get()}",
+            scriptOutputDirectories.get().asFile.absolutePath
+        )
+    }
+}
+
+fun addBenchmarkTask(
+    taskName: String,
+    script: String,
+    jdkVersion: JavaLanguageVersion = JavaLanguageVersion.of(8)
+): TaskProvider<JavaExec> {
+    return tasks.register<JavaExec>(taskName) {
+        group = "Gradle Regression Benchmark"
+        description = "Runs regression benchmark from $script"
+
+        dependsOn(":kotlin-gradle-plugin:install")
+
+        javaLauncher.set(service.launcherFor {
+            languageVersion.set(jdkVersion)
+        })
+        classpath = compilerClasspath
+        mainClass.set("org.jetbrains.kotlin.cli.jvm.K2JVMCompiler")
+
+        val scriptArgs = objects.newInstance<ScriptArgumentProvider>()
+        scriptArgs.script.set(script)
+        scriptArgs.scriptClasspath.from(scriptsClasspath)
+        argumentProviders.add(scriptArgs)
+    }
+}