From b1064c1edda2da593636e58b0bd91e92a0b40ec6 Mon Sep 17 00:00:00 2001
From: nik <Nikolay.Chashnikov@jetbrains.com>
Date: Wed, 28 Mar 2012 14:13:45 +0400
Subject: [PATCH] fixed state for recursive visitor interrupted by
 ProcessCanceledException (IDEA-55374)

---
 platform/util/src/com/intellij/util/WalkingState.java | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/platform/util/src/com/intellij/util/WalkingState.java b/platform/util/src/com/intellij/util/WalkingState.java
index 9106b72fa6d2..c4bcb1db6081 100644
--- a/platform/util/src/com/intellij/util/WalkingState.java
+++ b/platform/util/src/com/intellij/util/WalkingState.java
@@ -48,8 +48,12 @@ public class WalkingState<T> {
     if (!startedWalking) {
       stopped = false;
       startedWalking = true;
-      walkChildren(element);
-      startedWalking = false;
+      try {
+        walkChildren(element);
+      }
+      finally {
+        startedWalking = false;
+      }
     }
   }