From 445edf8e2827a694980530ade5909fe86b657fc3 Mon Sep 17 00:00:00 2001 From: Roman Golyshev Date: Thu, 24 Dec 2020 15:16:00 +0300 Subject: [PATCH] FIR IDE: Mark `FirTowerDataContextCollector` as `ThreadSafeMutableState` Also, use HashMap instead of LinkedHashMap in `FirTowerDataContextCollector` --- .../level/api/element/builder/FirTowerDataContextCollector.kt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/element/builder/FirTowerDataContextCollector.kt b/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/element/builder/FirTowerDataContextCollector.kt index 351aa2dd053..77a6db41950 100644 --- a/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/element/builder/FirTowerDataContextCollector.kt +++ b/idea/idea-frontend-fir/idea-fir-low-level-api/src/org/jetbrains/kotlin/idea/fir/low/level/api/element/builder/FirTowerDataContextCollector.kt @@ -5,14 +5,16 @@ package org.jetbrains.kotlin.idea.fir.low.level.api.element.builder +import org.jetbrains.kotlin.fir.ThreadSafeMutableState import org.jetbrains.kotlin.fir.declarations.FirDeclaration import org.jetbrains.kotlin.fir.expressions.FirStatement import org.jetbrains.kotlin.fir.psi import org.jetbrains.kotlin.fir.resolve.FirTowerDataContext import org.jetbrains.kotlin.psi.KtElement +@ThreadSafeMutableState class FirTowerDataContextCollector { - private val state: MutableMap = mutableMapOf() + private val state: MutableMap = hashMapOf() fun addStatementContext(statement: FirStatement, context: FirTowerDataContext) { (statement.psi as? KtElement)?.let { state[it] = context }