org.jetbrains.kotlin.analysis.utils.collections.ConcurrentNullableMap.kt Maven / Gradle / Ivy

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

Annotation Processor for Kotlin

There is a newer version: 2.1.0-Beta1

Show newest version

/*
 * Copyright 2010-2021 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.analysis.utils.collections

import java.util.concurrent.ConcurrentMap

public class ConcurrentMapBasedCache(public val map: ConcurrentMap) {
    public inline fun getOrPut(
        key: K,
        crossinline compute: (K) -> V?
    ): V {
        val value = map.getOrPut(key) { compute(key) ?: NullValue }
        return value.nullValueToNull()
    }
}

public object NullValue

@Suppress("UNCHECKED_CAST")
public fun  Any.nullValueToNull(): V = when (this) {
    NullValue -> null
    else -> this
} as V