From fd0919a89eb421829db76d4dafa46b31d3fccadb Mon Sep 17 00:00:00 2001
From: Daniil Ovchinnikov <daniil.ovchinnikov@jetbrains.com>
Date: Thu, 31 Jan 2019 16:22:43 +0300
Subject: [PATCH] [groovy] ProgressManager.checkCanceled() inside loops

---
 .../groovy/lang/resolve/NonCodeMembersContributor.java       | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/resolve/NonCodeMembersContributor.java b/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/resolve/NonCodeMembersContributor.java
index aa210a31301f..5b458c70806e 100644
--- a/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/resolve/NonCodeMembersContributor.java
+++ b/plugins/groovy/groovy-psi/src/org/jetbrains/plugins/groovy/lang/resolve/NonCodeMembersContributor.java
@@ -79,13 +79,16 @@ public abstract class NonCodeMembersContributor {
 
     if (aClass != null) {
       for (String superClassName : ClassUtil.getSuperClassesWithCache(aClass).keySet()) {
+        ProgressManager.checkCanceled();
         for (NonCodeMembersContributor enhancer : ourClassSpecifiedContributors.get(superClassName)) {
+          ProgressManager.checkCanceled();
           if (!invokeContributor(qualifierType, place, state, aClass, allDelegates, enhancer)) return false;
         }
       }
     }
 
     for (NonCodeMembersContributor contributor : ourAllTypeContributors) {
+      ProgressManager.checkCanceled();
       if (!invokeContributor(qualifierType, place, state, aClass, allDelegates, contributor)) return false;
     }
 
@@ -102,8 +105,8 @@ public abstract class NonCodeMembersContributor {
                                            PsiClass aClass,
                                            List<MyDelegatingScopeProcessor> allDelegates,
                                            NonCodeMembersContributor enhancer) {
-    ProgressManager.checkCanceled();
     for (MyDelegatingScopeProcessor delegatingProcessor : allDelegates) {
+      ProgressManager.checkCanceled();
       enhancer.processDynamicElements(qualifierType, aClass, delegatingProcessor, place, state);
       if (!delegatingProcessor.wantMore) {
         return false;