FIR IDE: add check canceled between diagnostics

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

@@ -62,9 +62,12 @@ abstract class AbstractDiagnosticCollector(
         componentsInitialized = true
     }
 
+    protected open fun beforeCollecting() {}
+
     private inner class Visitor : FirDefaultVisitor<Unit, Nothing?>() {
         private fun <T : FirElement> T.runComponents() {
             components.forEach {
+                beforeCollecting()
                 this.accept(it, context)
             }
         }

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

@@ -48,6 +48,10 @@ internal abstract class AbstractFirIdeDiagnosticsCollector(
         reporter = Reporter()
     }
 
+    override fun beforeCollecting() {
+        checkCanceled()
+    }
+
     override fun getCollectedDiagnostics(): Iterable<FirDiagnostic<*>> {
         // Not necessary in IDE
         return emptyList()