azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

FirTypeIntersectionScope: add intersection overrides caching

Browse Source
This commit is contained in:
Mikhail Glukhikh
2020-07-14 12:22:32 +03:00
parent c1d223dbde
commit 10a1d5c03b
1 changed files with 26 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/fir/resolve/src/org/jetbrains/kotlin/fir/scopes/impl/FirTypeIntersectionScope.kt
+7 -2
View File
@@ -37,6 +37,8 @@ class FirTypeIntersectionScope private constructor(
private val overriddenSymbols: MutableMap<FirCallableSymbol<*>, Collection<FirCallableSymbol<*>>> = mutableMapOf() private val overriddenSymbols: MutableMap<FirCallableSymbol<*>, Collection<FirCallableSymbol<*>>> = mutableMapOf()
private val intersectionOverrides: MutableMap<FirNamedFunctionSymbol, FirNamedFunctionSymbol> = mutableMapOf()
override fun processFunctionsByName(name: Name, processor: (FirFunctionSymbol<*>) -> Unit) { override fun processFunctionsByName(name: Name, processor: (FirFunctionSymbol<*>) -> Unit) {
if (!processCallablesByName(name, processor, absentFunctions, FirScope::processFunctionsByName)) { if (!processCallablesByName(name, processor, absentFunctions, FirScope::processFunctionsByName)) {
super.processFunctionsByName(name, processor) super.processFunctionsByName(name, processor)
@@ -93,6 +95,7 @@ class FirTypeIntersectionScope private constructor(
val mostSpecific = selectMostSpecificMember(extractedOverrides) val mostSpecific = selectMostSpecificMember(extractedOverrides)
if (extractedOverrides.size > 1 && mostSpecific is FirNamedFunctionSymbol) { if (extractedOverrides.size > 1 && mostSpecific is FirNamedFunctionSymbol) {
// TODO: same code for properties // TODO: same code for properties
val intersectionOverride = intersectionOverrides.getOrPut(mostSpecific) {
val newSymbol = FirNamedFunctionSymbol(mostSpecific.callableId, mostSpecific.isFakeOverride, mostSpecific) val newSymbol = FirNamedFunctionSymbol(mostSpecific.callableId, mostSpecific.isFakeOverride, mostSpecific)
val mostSpecificFunction = mostSpecific.fir val mostSpecificFunction = mostSpecific.fir
createFunctionCopy(mostSpecific.fir, newSymbol).apply { createFunctionCopy(mostSpecific.fir, newSymbol).apply {
@@ -113,9 +116,11 @@ class FirTypeIntersectionScope private constructor(
} }
} }
}.build() }.build()
overriddenSymbols[newSymbol] = extractedOverrides newSymbol
}
overriddenSymbols[intersectionOverride] = extractedOverrides
@Suppress("UNCHECKED_CAST") @Suppress("UNCHECKED_CAST")
processor(newSymbol as D) processor(intersectionOverride as D)
} else { } else {
overriddenSymbols[mostSpecific] = extractedOverrides overriddenSymbols[mostSpecific] = extractedOverrides
processor(mostSpecific) processor(mostSpecific)