#137 | Deprecated API - Impex Folding traverser

epam · Jan 8, 2023 · 74bacc7 · 74bacc7
1 parent 31b78aa
commit 74bacc7
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/src/com/intellij/idea/plugin/hybris/lang/folding/ImpexFoldingBuilder.java b/src/com/intellij/idea/plugin/hybris/lang/folding/ImpexFoldingBuilder.java
@@ -29,8 +29,7 @@
 import com.intellij.openapi.editor.Document;
 import com.intellij.openapi.editor.FoldingGroup;
 import com.intellij.psi.PsiElement;
-import com.intellij.psi.search.PsiElementProcessor.CollectFilteredElements;
-import com.intellij.psi.util.PsiTreeUtil;
+import com.intellij.psi.SyntaxTraverser;
 import org.apache.commons.lang3.Validate;
 import org.jetbrains.annotations.Contract;
 import org.jetbrains.annotations.NotNull;
@@ -106,12 +105,10 @@ protected Collection<PsiElement> findFoldingBlocksAndLineBreaks(@Nullable final
             return Collections.emptyList();
         }
 
-        final List<PsiElement> foldingBlocks = new ArrayList<>();
-        PsiTreeUtil.processElements(root, new CollectFilteredElements<>(
-            PsiElementFilterFactory.getPsiElementFilter(), foldingBlocks
-        ));
-
-        return foldingBlocks;
+        final var filter = PsiElementFilterFactory.getPsiElementFilter();
+        return SyntaxTraverser.psiTraverser(root)
+                              .filter(filter::isAccepted)
+                              .toList();
     }
 
     @Nullable