From 7161c8bf6b44df424fd172a13160f1015edf550c Mon Sep 17 00:00:00 2001
From: Nicolay Mitropolsky <nicolay.mitropolsky@jetbrains.com>
Date: Tue, 1 Dec 2020 21:13:31 +0300
Subject: [PATCH] Adding some `CancellationCheck.runWithCancellationCheck`

GitOrigin-RevId: 5c2ec8ecc78122d115c721ac25e2559a63a1671e
---
 uast/uast-common/intellij.platform.uast.iml                    | 1 +
 .../src/com/intellij/psi/UastReferenceProviderAdapter.kt       | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/uast/uast-common/intellij.platform.uast.iml b/uast/uast-common/intellij.platform.uast.iml
index bf8a3dddfa6a..cdcceb9935d1 100644
--- a/uast/uast-common/intellij.platform.uast.iml
+++ b/uast/uast-common/intellij.platform.uast.iml
@@ -14,5 +14,6 @@
     <orderEntry type="module" module-name="intellij.platform.lang" />
     <orderEntry type="library" name="Trove4j" level="project" />
     <orderEntry type="module" module-name="intellij.platform.indexing.impl" />
+    <orderEntry type="module" module-name="intellij.platform.ide.impl" />
   </component>
 </module>
\ No newline at end of file
diff --git a/uast/uast-common/src/com/intellij/psi/UastReferenceProviderAdapter.kt b/uast/uast-common/src/com/intellij/psi/UastReferenceProviderAdapter.kt
index 9b25a217b923..b424d0471e30 100644
--- a/uast/uast-common/src/com/intellij/psi/UastReferenceProviderAdapter.kt
+++ b/uast/uast-common/src/com/intellij/psi/UastReferenceProviderAdapter.kt
@@ -1,6 +1,7 @@
 // Copyright 2000-2020 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
 package com.intellij.psi
 
+import com.intellij.openapi.progress.impl.CancellationCheck
 import com.intellij.util.ProcessingContext
 import org.jetbrains.uast.UElement
 
@@ -9,7 +10,7 @@ internal class UastReferenceProviderAdapter(private val supportedUElementTypes:
 
   override fun getReferencesByElement(element: PsiElement, context: ProcessingContext): Array<PsiReference> {
     val uElement = UastPatternAdapter.getOrCreateCachedElement(element, context, supportedUElementTypes) ?: return PsiReference.EMPTY_ARRAY
-    return provider.getReferencesByElement(uElement, context)
+    return CancellationCheck.runWithCancellationCheck { provider.getReferencesByElement (uElement, context) }
   }
 
   override fun acceptsTarget(target: PsiElement): Boolean {