org.jetbrains.kotlin.backend.konan.PerformanceManagerHelpers.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlin-native-compiler-embeddable Show documentation

Embeddable JAR of Kotlin/Native compiler

There is a newer version: 2.1.0-RC2

Show newest version

/*
 * Copyright 2010-2024 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.backend.konan

import org.jetbrains.kotlin.cli.common.CLIConfigurationKeys
import org.jetbrains.kotlin.cli.common.CommonCompilerPerformanceManager
import org.jetbrains.kotlin.config.CompilerConfiguration

class K2NativeCompilerPerformanceManager : CommonCompilerPerformanceManager("Kotlin to Native Compiler") {

    private val children = mutableListOf()

    /**
     * Creates a separate performance manager. Use case: create a child, use in a separate thread, collect back the measurements.
     * Measurements may be collected back by [collectChildMeasurements].
     *
     * WARNING: not thread-safe!
     */
    internal fun createChild(): CommonCompilerPerformanceManager {
        return ChildCompilerPerformanceManager().also {
            children += it
        }
    }

    /**
     * Collects the measurements from the "child" performance managers.
     *
     * WARNING: not thread-safe!
     */
    fun collectChildMeasurements() {
        children.forEach {
            measurements += it.getMeasurementResults()
            it.clearMeasurements()
        }
    }

    private inner class ChildCompilerPerformanceManager : CommonCompilerPerformanceManager("Kotlin to Native Compiler per single thread") {
        fun clearMeasurements() {
            measurements.clear()
        }
    }
}

var CompilerConfiguration.performanceManager: CommonCompilerPerformanceManager?
    get() = this[CLIConfigurationKeys.PERF_MANAGER]
    set(v) {
        this.putIfNotNull(CLIConfigurationKeys.PERF_MANAGER, v)
    }

internal inline fun  CommonCompilerPerformanceManager?.trackAnalysis(fn: () -> T): T {
    this?.notifyAnalysisStarted()
    try {
        return fn()
    } finally {
        this?.notifyAnalysisFinished()
    }
}

internal inline fun  CommonCompilerPerformanceManager?.trackIRTranslation(fn: () -> T): T {
    this?.notifyIRTranslationStarted()
    try {
        return fn()
    } finally {
        this?.notifyIRTranslationFinished()
    }
}

internal inline fun  CommonCompilerPerformanceManager?.trackGeneration(fn: () -> T): T {
    this?.notifyGenerationStarted()
    try {
        return fn()
    } finally {
        this?.notifyGenerationFinished()
    }
}

internal inline fun  CommonCompilerPerformanceManager?.trackIRLowering(fn: () -> T): T {
    this?.notifyIRLoweringStarted()
    try {
        return fn()
    } finally {
        this?.notifyIRLoweringFinished()
    }
}