refactor(diff): wrap document update in runWriteAction

Ensure thread safety by wrapping document updates in `runWriteAction` to prevent potential UI thread issues.
unit-mesh · Jan 19, 2025 · 777480b · 777480b
1 parent 2411fb7
commit 777480b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/diff/DiffStreamHandler.kt b/core/src/main/kotlin/cc/unitmesh/devti/diff/DiffStreamHandler.kt
@@ -32,6 +32,7 @@ import cc.unitmesh.devti.diff.model.streamDiff
 import cc.unitmesh.devti.llms.LlmFactory
 import cc.unitmesh.devti.util.AutoDevCoroutineScope
 import cc.unitmesh.devti.util.parser.CodeFence
+import com.intellij.openapi.application.runWriteAction
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.cancellable
 import kotlinx.coroutines.flow.flowOf
@@ -99,7 +100,9 @@ class DiffStreamHandler(
 
     fun acceptAll() {
         resetState()
-        editor.document.setText(newCode)
+        runWriteAction {
+            editor.document.setText(newCode)
+        }
     }
 
     fun rejectAll() {