org.jetbrains.kotlin.cli.common.profiling.ProfilingCompilerPerformanceManager.kt Maven / Gradle / Ivy
/*
* Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
*/
package org.jetbrains.kotlin.cli.common.profiling
import org.jetbrains.kotlin.cli.common.CommonCompilerPerformanceManager
import java.io.File
import java.text.SimpleDateFormat
import java.util.*
class ProfilingCompilerPerformanceManager(
profilerPath: String,
val command: String,
val outputDir: File
) : CommonCompilerPerformanceManager("Profiling") {
private val profiler = AsyncProfilerHelper.getInstance(profilerPath)
private val runDate = Date()
private val formatter = SimpleDateFormat("yyyy-MM-dd__HH-mm")
private var active = false
init {
startProfiling()
}
private fun startProfiling() {
profiler.execute(command)
active = true
}
private fun stopProfiling() {
if (active) profiler.stop()
active = false
}
private fun restartProfiling() {
stopProfiling()
startProfiling()
}
private fun dumpProfile(postfix: String) {
outputDir.mkdirs()
val outputFile = outputDir.resolve("snapshot-${formatter.format(runDate)}-$postfix.collapsed")
outputFile.writeText(profiler.execute("collapsed"))
active = false
}
override fun notifyRepeat(total: Int, number: Int) {
dumpProfile("repeat$number")
restartProfiling()
}
override fun notifyCompilationFinished() {
dumpProfile("final")
stopProfiling()
}
companion object {
fun create(profileCompilerArgument: String): ProfilingCompilerPerformanceManager {
val (path, command, outputDir) = profileCompilerArgument.split(":", limit = 3)
return ProfilingCompilerPerformanceManager(path, command, File(outputDir))
}
}
}