fix read-lock asserts in PsiViewerTreeModel

https://jb-web.exa.aws.intellij.net/issue/994763
JetBrains · Dec 22, 2023 · f2e3a0d · f2e3a0d
1 parent 5638775
commit f2e3a0d
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/src/main/java/idea/plugin/psiviewer/model/PsiViewerTreeModel.java b/src/main/java/idea/plugin/psiviewer/model/PsiViewerTreeModel.java
@@ -22,6 +22,7 @@
 
 package idea.plugin.psiviewer.model;
 
+import com.intellij.openapi.application.ReadAction;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiWhiteSpace;
 import idea.plugin.psiviewer.controller.project.PsiViewerProjectService;
@@ -69,14 +70,16 @@ public boolean isLeaf(Object node) {
     }
 
     private List<PsiElement> getFilteredChildren(PsiElement psi) {
-        final List<PsiElement> filteredChildren = new ArrayList<>();
+        return ReadAction.compute(() -> {
+            final List<PsiElement> filteredChildren = new ArrayList<>();
 
-        for (PsiElement e = psi.getFirstChild(); e != null; e = e.getNextSibling())
-            if (isValid(e)) {
-                filteredChildren.add(e);
-            }
+            for (PsiElement e = psi.getFirstChild(); e != null; e = e.getNextSibling())
+                if (isValid(e)) {
+                    filteredChildren.add(e);
+                }
 
-        return filteredChildren;
+            return filteredChildren;
+        });
     }
 
     private boolean isValid(PsiElement psiElement) {