From 556814e9b62493b10e05e6ad1f3ec67bfacecd13 Mon Sep 17 00:00:00 2001
From: Dmitry Gridin <dmitry.gridin@jetbrains.com>
Date: Thu, 23 Jan 2020 18:00:13 +0700
Subject: [PATCH] TrailingCommaVisitor: add `checkCanceled` for the recursive
 case

#KT-34744
---
 .../kotlin/idea/formatter/TrailingCommaVisitor.kt          | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/idea/src/org/jetbrains/kotlin/idea/formatter/TrailingCommaVisitor.kt b/idea/src/org/jetbrains/kotlin/idea/formatter/TrailingCommaVisitor.kt
index 904724c63fe..34cb42e2e04 100644
--- a/idea/src/org/jetbrains/kotlin/idea/formatter/TrailingCommaVisitor.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/formatter/TrailingCommaVisitor.kt
@@ -51,10 +51,9 @@ abstract class TrailingCommaVisitor : KtTreeVisitorVoid() {
     }
 
     override fun visitElement(element: PsiElement) {
-        if (recursively)
-            super.visitElement(element)
-        else
-            ProgressIndicatorProvider.checkCanceled()
+        ProgressIndicatorProvider.checkCanceled()
+
+        if (recursively) super.visitElement(element)
     }
 
     protected abstract fun process(commaOwner: KtElement)