Merge branch 'main' into workspace-test

aws · Jan 16, 2025 · 62f896d · 62f896d
2 parents 4662cca + 5e5ba0e
commit 62f896d
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/...rc/software/aws/toolkits/jetbrains/services/amazonqCodeTest/storage/ChatSessionStorage.kt b/...rc/software/aws/toolkits/jetbrains/services/amazonqCodeTest/storage/ChatSessionStorage.kt
@@ -8,13 +8,8 @@ import software.aws.toolkits.jetbrains.services.amazonqCodeTest.session.Session
 class ChatSessionStorage {
     private val sessions = mutableMapOf<String, Session>()
 
-    private fun createSession(tabId: String): Session {
-        val session = Session(tabId)
-        sessions[tabId] = session
-        return session
-    }
-
-    fun getSession(tabId: String): Session = sessions[tabId] ?: createSession(tabId)
+    @Synchronized
+    fun getSession(tabId: String): Session = sessions.getOrPut(tabId) { Session(tabId) }
 
     fun deleteSession(tabId: String) {
         sessions.remove(tabId)