Memoize absent function/property names in FirSuperTypeScope

compiler/fir/java/src/org/jetbrains/kotlin/fir/java/scopes/JavaClassUseSiteScope.kt

@@ -17,6 +17,7 @@ import org.jetbrains.kotlin.fir.scopes.FirScope
 import org.jetbrains.kotlin.fir.scopes.ProcessorAction
 import org.jetbrains.kotlin.fir.scopes.ProcessorAction.*
 import org.jetbrains.kotlin.fir.scopes.impl.FirAbstractProviderBasedScope
+import org.jetbrains.kotlin.fir.scopes.impl.FirSuperTypeScope
 import org.jetbrains.kotlin.fir.symbols.*
 import org.jetbrains.kotlin.fir.symbols.impl.*
 import org.jetbrains.kotlin.fir.typeContext
@@ -27,7 +28,7 @@ import org.jetbrains.kotlin.name.Name
 class JavaClassUseSiteScope(
     klass: FirRegularClass,
     session: FirSession,
-    private val superTypesScope: FirScope,
+    private val superTypesScope: FirSuperTypeScope,
     private val declaredMemberScope: FirScope
 ) : FirAbstractProviderBasedScope(session, lookupInFir = true) {
     internal val symbol = klass.symbol

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/scopes/impl/FirClassUseSiteScope.kt

@@ -18,7 +18,7 @@ import org.jetbrains.kotlin.name.Name
 
 class FirClassUseSiteScope(
     session: FirSession,
-    private val superTypesScope: FirScope,
+    private val superTypesScope: FirSuperTypeScope,
     private val declaredMemberScope: FirScope
 ) : AbstractFirOverrideScope(session) {
 

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/scopes/impl/FirSuperTypeScope.kt

@@ -17,15 +17,23 @@ class FirSuperTypeScope(
     val scopes: List<FirScope>
 ) : AbstractFirOverrideScope(session) {
 
+    private val absentFunctions = mutableSetOf<Name>()
+
+    private val absentProperties = mutableSetOf<Name>()
+
     override fun processFunctionsByName(name: Name, processor: (FirFunctionSymbol<*>) -> ProcessorAction): ProcessorAction {
+        if (name in absentFunctions) {
+            return ProcessorAction.NEXT
+        }
         val accepted = HashSet<FirFunctionSymbol<*>>()
         val pending = mutableListOf<FirFunctionSymbol<*>>()
+        var empty = true
         for (scope in scopes) {
-            if (scope.processFunctionsByName(name) {
+            if (scope.processFunctionsByName(name) { functionSymbol ->
-
+                    empty = false
-                    if (it !in accepted && it.isOverridden(accepted) == null) {
+                    if (functionSymbol !in accepted && functionSymbol.isOverridden(accepted) == null) {
-                        pending += it
+                        pending += functionSymbol
-                        processor(it)
+                        processor(functionSymbol)
                     } else {
                         ProcessorAction.NEXT
                     }
@@ -36,14 +44,22 @@ class FirSuperTypeScope(
             accepted += pending
             pending.clear()
         }
+        if (empty) {
+            absentFunctions += name
+        }
         return super.processFunctionsByName(name, processor)
     }
 
     override fun processPropertiesByName(name: Name, processor: (FirCallableSymbol<*>) -> ProcessorAction): ProcessorAction {
+        if (name in absentProperties) {
+            return ProcessorAction.NEXT
+        }
         val accepted = HashSet<FirCallableSymbol<*>>()
         val pending = mutableListOf<FirCallableSymbol<*>>()
+        var empty = true
         for (scope in scopes) {
             if (scope.processPropertiesByName(name) {
+                    empty = false
                     if (it !in accepted && it.isOverridden(accepted) == null) {
                         pending += it
                         processor(it)
@@ -57,6 +73,9 @@ class FirSuperTypeScope(
             accepted += pending
             pending.clear()
         }
+        if (empty) {
+            absentProperties += name
+        }
         return super.processPropertiesByName(name, processor)
     }
 }