Handle process cancellation during find usages

cashapp · Jul 6, 2023 · 08bca18 · 08bca18
1 parent 10106a6
commit 08bca18
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/...plugin/src/main/kotlin/app/cash/sqldelight/intellij/SqlDelightFindUsagesHandlerFactory.kt b/...plugin/src/main/kotlin/app/cash/sqldelight/intellij/SqlDelightFindUsagesHandlerFactory.kt
@@ -15,6 +15,7 @@ import com.intellij.find.findUsages.FindUsagesOptions
 import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.application.ReadAction
 import com.intellij.openapi.module.ModuleUtil
+import com.intellij.openapi.progress.ProcessCanceledException
 import com.intellij.openapi.roots.ProjectRootManager
 import com.intellij.openapi.util.Computable
 import com.intellij.openapi.vfs.VirtualFile
@@ -41,7 +42,13 @@ internal class SqlDelightFindUsagesHandlerFactory : FindUsagesHandlerFactory() {
   override fun createFindUsagesHandler(
     element: PsiElement,
     forHighlightUsages: Boolean,
-  ): FindUsagesHandler = SqlDelightIdentifierHandler(element)
+  ): FindUsagesHandler? {
+    return try {
+      SqlDelightIdentifierHandler(element)
+    } catch (e: ProcessCanceledException) {
+      null
+    }
+  }
 }
 
 internal class SqlDelightIdentifierHandler(