[BTA] Gather build metrics from the daemon

^KT-78198 In Progress
Relates to KT-78199

Author: ALikhachev

compiler/build-tools/kotlin-build-tools-impl/src/main/kotlin/org/jetbrains/kotlin/buildtools/internal/CompilationServiceImpl.kt

@@ -306,7 +306,8 @@ internal object CompilationServiceImpl : CompilationService {
             BasicCompilerServicesWithResultsFacadeServer(loggerAdapter),
             DaemonCompilationResults(
                 loggerAdapter.kotlinLogger,
-                compilationConfiguration.aggregatedIcConfiguration?.options?.rootProjectDir
+                compilationConfiguration.aggregatedIcConfiguration?.options?.rootProjectDir,
+                DoNothingBuildMetricsReporter,
             )
         ).get()
 

compiler/build-tools/kotlin-build-tools-impl/src/main/kotlin/org/jetbrains/kotlin/buildtools/internal/daemonAdapters.kt

@@ -6,6 +6,9 @@
 package org.jetbrains.kotlin.buildtools.internal
 
 import org.jetbrains.kotlin.build.report.metrics.BuildMetrics
+import org.jetbrains.kotlin.build.report.metrics.BuildMetricsReporter
+import org.jetbrains.kotlin.build.report.metrics.GradleBuildPerformanceMetric
+import org.jetbrains.kotlin.build.report.metrics.GradleBuildTime
 import org.jetbrains.kotlin.buildtools.api.KotlinLogger
 import org.jetbrains.kotlin.buildtools.api.jvm.ClasspathSnapshotBasedIncrementalCompilationApproachParameters
 import org.jetbrains.kotlin.buildtools.api.jvm.ClasspathSnapshotBasedIncrementalJvmCompilationConfiguration
@@ -70,7 +73,11 @@ internal val JvmCompilationConfigurationImpl.asDaemonCompilationOptions: Compila
         }
     }
 
-internal class DaemonCompilationResults(private val kotlinLogger: KotlinLogger, private val rootProjectDir: File?) : CompilationResults,
+internal class DaemonCompilationResults(
+    private val kotlinLogger: KotlinLogger,
+    private val rootProjectDir: File?,
+    private val buildMetricsReporter: BuildMetricsReporter<GradleBuildTime, GradleBuildPerformanceMetric>
+) : CompilationResults,
     UnicastRemoteObject(
         SOCKET_ANY_FREE_PORT,
         LoopbackNetworkInterface.clientLoopbackSocketFactory,
@@ -87,6 +94,9 @@ internal class DaemonCompilationResults(private val kotlinLogger: KotlinLogger,
         // TODO propagate the values to the caller via callbacks, requires to make metrics a part of the API
         when (compilationResultCategory) {
             CompilationResultCategory.IC_COMPILE_ITERATION.code -> kotlinLogger.debug(value as? CompileIterationResult, rootProjectDir)
+            CompilationResultCategory.BUILD_METRICS.code -> @Suppress("UNCHECKED_CAST") (value as? BuildMetrics<GradleBuildTime, GradleBuildPerformanceMetric>)?.let {
+                buildMetricsReporter.addMetrics(it)
+            }
             else -> kotlinLogger.debug("Result category=$compilationResultCategory value=$value")
         }
     }

compiler/build-tools/kotlin-build-tools-impl/src/main/kotlin/org/jetbrains/kotlin/buildtools/internal/jvm/operations/JvmCompilationOperationImpl.kt

@@ -126,9 +126,10 @@ internal class JvmCompilationOperationImpl(
                 val aggregatedIcConfigurationOptions =
                     aggregatedIcConfiguration.options as JvmSnapshotBasedIncrementalCompilationOptionsImpl
                 val sourcesChanges = aggregatedIcConfiguration.sourcesChanges
-                val requestedCompilationResults = arrayOf(
+                val requestedCompilationResults = listOfNotNull(
                     CompilationResultCategory.IC_COMPILE_ITERATION.code,
-                )
+                    CompilationResultCategory.BUILD_METRICS.code.takeIf { this[XX_KGP_METRICS_COLLECTOR] },
+                ).toTypedArray()
                 val classpathChanges = aggregatedIcConfiguration.classpathChanges
                 IncrementalCompilationOptions(
                     sourcesChanges,
@@ -215,13 +216,14 @@ internal class JvmCompilationOperationImpl(
         val aggregatedIcConfigurationOptions = aggregatedIcConfiguration?.options as? JvmSnapshotBasedIncrementalCompilationOptionsImpl
         val rootProjectDir = aggregatedIcConfigurationOptions?.get(ROOT_PROJECT_DIR)
         logCompilerArguments(loggerAdapter, arguments, get(COMPILER_ARGUMENTS_LOG_LEVEL))
+        val metricsReporter = getMetricsReporter()
         val exitCode = daemon.compile(
             sessionId,
             arguments.toArgumentStrings().toTypedArray(),
             daemonCompileOptions,
             BasicCompilerServicesWithResultsFacadeServer(loggerAdapter),
             DaemonCompilationResults(
-                loggerAdapter.kotlinLogger, rootProjectDir?.toFile()
+                loggerAdapter.kotlinLogger, rootProjectDir?.toFile(), metricsReporter
             )
         ).get()
 
@@ -235,8 +237,13 @@ internal class JvmCompilationOperationImpl(
             ExitCode.OK
         } else {
             ExitCode.COMPILATION_ERROR
-        }).asCompilationResult
-
+        }).asCompilationResult.also {
+            if (this@JvmCompilationOperationImpl[XX_KGP_METRICS_COLLECTOR] && metricsReporter is BuildMetricsReporterImpl) {
+                this@JvmCompilationOperationImpl[XX_KGP_METRICS_COLLECTOR_OUT] = ByteArrayOutputStream().apply {
+                    ObjectOutputStream(this).writeObject(metricsReporter)
+                }.toByteArray()
+            }
+        }
     }
 
     private fun getCurrentClasspath() =