FIR IDE: collect diagnostics under read lock

2021-03-05 13:23:27 +01:00
parent 770468c78e
commit 81baf4fae5
1 changed files with 5 additions and 2 deletions
@@ -59,10 +59,13 @@ internal class FileStructure(
    @OptIn(ExperimentalStdlibApi::class)
    fun getAllDiagnosticsForFile(diagnosticCheckerFilter: DiagnosticCheckerFilter): Collection<FirPsiDiagnostic<*>> {
        val structureElements = getAllStructureElements()
-        return buildSet {
+
+        return moduleFileCache.firFileLockProvider.withReadLock(firFile) {
+            buildSet {
                collectDiagnosticsFromStructureElements(structureElements, diagnosticCheckerFilter)
            }
        }
+    }

    private fun MutableSet<FirPsiDiagnostic<*>>.collectDiagnosticsFromStructureElements(
        structureElements: Collection<FileStructureElement>,