FIR: remove beforeRunningAllComponentsOnElement/beforeRunningSingleComponentOnElement from AbstractDiagnosticCollectorVisitor

compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/collectors/AbstractDiagnosticCollectorVisitor.kt

@@ -29,9 +29,6 @@ abstract class AbstractDiagnosticCollectorVisitor(
     protected val components: List<AbstractDiagnosticCollectorComponent>,
 ) : FirDefaultVisitor<Unit, Nothing?>() {
 
-    protected open fun beforeRunningAllComponentsOnElement(element: FirElement) {}
-    protected open fun beforeRunningSingleComponentOnElement(element: FirElement) {}
-
     protected open fun shouldVisitDeclaration(declaration: FirDeclaration) = true
     protected open fun onDeclarationExit(declaration: FirDeclaration) {}
 

compiler/fir/checkers/src/org/jetbrains/kotlin/fir/analysis/collectors/CheckerRunningDiagnosticCollectorVisitor.kt

@@ -19,9 +19,7 @@ open class CheckerRunningDiagnosticCollectorVisitor(
     }
 
     override fun runComponents(element: FirElement) {
-        beforeRunningAllComponentsOnElement(element)
         components.forEach {
-            beforeRunningSingleComponentOnElement(element)
             element.accept(it, context)
         }
     }

idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/diagnostics/FirIdeDiagnosticVisitor.kt

@@ -18,10 +18,6 @@ internal open class FirIdeDiagnosticVisitor(
 ) : CheckerRunningDiagnosticCollectorVisitor(context, components) {
     private val beforeElementDiagnosticCollectionHandler = context.session.beforeElementDiagnosticCollectionHandler
 
-    override fun beforeRunningSingleComponentOnElement(element: FirElement) {
-        checkCanceled()
-    }
-
     override fun goToNestedDeclarations(element: FirElement) {
         if (element is FirDeclaration) {
             beforeElementDiagnosticCollectionHandler?.beforeGoingNestedDeclaration(element, context)
@@ -29,7 +25,11 @@ internal open class FirIdeDiagnosticVisitor(
         super.goToNestedDeclarations(element)
     }
 
-    override fun beforeRunningAllComponentsOnElement(element: FirElement) {
+    override fun runComponents(element: FirElement) {
         beforeElementDiagnosticCollectionHandler?.beforeCollectingForElement(element)
+        components.forEach {
+            checkCanceled()
+            element.accept(it, context)
+        }
     }
 }