org.jetbrains.kotlin.fir.resolve.providers.SymbolProviderCache.kt Maven / Gradle / Ivy

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

Kotlin Compiler

There is a newer version: 2.0.0

/*
 * 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.fir.resolve.providers

import org.jetbrains.kotlin.fir.PrivateForInline

@Suppress("EXPERIMENTAL_FEATURE_WARNING")
inline class SymbolProviderCache @PrivateForInline constructor(@PrivateForInline val cache: HashMap) {
    @OptIn(PrivateForInline::class)
    constructor() : this(HashMap())

    @PrivateForInline
    object NullValue

    @OptIn(PrivateForInline::class)
    inline fun lookupCacheOrCalculate(key: K, crossinline l: (K) -> V?): V? {
        @Suppress("UNCHECKED_CAST")
        return when (val value = cache[key]) {
            null -> {
                val calculated = l(key)
                cache[key] = calculated ?: NullValue
                calculated
            }
            NullValue -> null
            else -> value as V
        }
    }

    @OptIn(PrivateForInline::class)
    inline fun  lookupCacheOrCalculateWithPostCompute(
        key: K, crossinline l: (K) -> Pair, postCompute: (V, T) -> Unit
    ): V? {
        @Suppress("UNCHECKED_CAST")
        return when (val value = cache[key]) {
            null -> {
                val calculated = l(key)
                cache[key] = calculated.first ?: NullValue
                calculated.first?.let { postCompute(it, calculated.second) }
                calculated.first
            }
            NullValue -> null
            else -> value as V
        }
    }

    @OptIn(PrivateForInline::class)
    operator fun contains(key: K): Boolean = key in cache

    @Suppress("UNCHECKED_CAST")
    @OptIn(PrivateForInline::class)
    operator fun get(key: K): V? = cache[key].takeIf { it !== NullValue } as V?

    @OptIn(PrivateForInline::class)
    operator fun set(key: K, value: V) {
        cache[key] = value
    }

    @OptIn(PrivateForInline::class)
    fun remove(key: K) {
        cache.remove(key)
    }
}