Handle process cancellation during find usages (#4340)

Alec Kazakova · web-flow · commit dd9be0e43c55 · 2023-07-12T09:11:16.000-04:00
diff --git a/sqldelight-idea-plugin/src/main/kotlin/app/cash/sqldelight/intellij/SqlDelightColumnRenameProcessor.kt b/sqldelight-idea-plugin/src/main/kotlin/app/cash/sqldelight/intellij/SqlDelightColumnRenameProcessor.kt
@@ -29,7 +29,7 @@ class SqlDelightColumnRenameProcessor : RenamePsiElementProcessor() {
     val collectProcessor = CommonProcessors.CollectProcessor<UsageInfo>()
     val readActionProcessor = ReadActionProcessor.wrapInReadAction(collectProcessor)
     val findUsagesHandler = findUsagesHandlerFactory.createFindUsagesHandler(element, false)
-    findUsagesHandler.processElementUsages(
+    findUsagesHandler?.processElementUsages(
       /* element = */
       element,
       /* processor = */
diff --git a/sqldelight-idea-plugin/src/main/kotlin/app/cash/sqldelight/intellij/SqlDelightFindUsagesHandlerFactory.kt b/sqldelight-idea-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(
diff --git a/sqldelight-idea-plugin/src/main/kotlin/app/cash/sqldelight/intellij/inspections/UnusedQueryInspection.kt b/sqldelight-idea-plugin/src/main/kotlin/app/cash/sqldelight/intellij/inspections/UnusedQueryInspection.kt
@@ -60,7 +60,7 @@ internal class UnusedQueryInspection : LocalInspectionTool() {
       isSearchForTextOccurrences = false
     }
     val findUsagesHandler = findUsagesFactory.createFindUsagesHandler(stmtIdentifier, true)
-    findUsagesHandler.processElementUsages(stmtIdentifier, readActionProcessor, findUsagesOptions)
+    findUsagesHandler?.processElementUsages(stmtIdentifier, readActionProcessor, findUsagesOptions)
     return findFirstProcessor.isFound
   }
 

Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ internal class UnusedQueryInspection : LocalInspectionTool() {`
`60`	`60`	`isSearchForTextOccurrences = false`
`61`	`61`	`}`
`62`	`62`	`val findUsagesHandler = findUsagesFactory.createFindUsagesHandler(stmtIdentifier, true)`
`63`		`- findUsagesHandler.processElementUsages(stmtIdentifier, readActionProcessor, findUsagesOptions)`
	`63`	`+ findUsagesHandler?.processElementUsages(stmtIdentifier, readActionProcessor, findUsagesOptions)`
`64`	`64`	`return findFirstProcessor.isFound`
`65`	`65`	`}`
`66`	`66`