Merge pull request #2294 from digma-ai/fix-slow-operation-on-eap

allow slow operations on EAP

Author: shalom938

ide-common/src/main/kotlin/org/digma/intellij/plugin/codelens/CodeLensServiceDocumentChangeListener.kt

@@ -13,14 +13,15 @@ import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.psi.PsiDocumentManager
 import com.intellij.psi.PsiManager
 import com.intellij.util.Alarm
-import com.intellij.util.AlarmFactory
+import org.digma.intellij.plugin.common.allowSlowOperation
 import org.digma.intellij.plugin.common.isProjectValid
 import org.digma.intellij.plugin.common.isValidVirtualFile
 import org.digma.intellij.plugin.common.runInReadAccessWithResult
 import org.digma.intellij.plugin.errorreporting.ErrorReporter
 import org.digma.intellij.plugin.log.Log
 import org.digma.intellij.plugin.psi.LanguageService
 import org.digma.intellij.plugin.psi.PsiUtils
+import java.util.function.Supplier
 
 //any change in the current editor needs a code lens refresh.
 //there is a listener for CodeLensChanged events, but its not enough. the file in the editor may change
@@ -46,7 +47,10 @@ class CodeLensServiceDocumentChangeListener(private val project: Project) : File
             }
 
 
-            val languageService = LanguageService.findLanguageServiceByFile(project, file)
+            val languageService = allowSlowOperation(Supplier {
+                LanguageService.findLanguageServiceByFile(project, file)
+            })
+
 
             //only some languages need code vision support
             if (!languageService.isCodeVisionSupported) {
@@ -64,9 +68,9 @@ class CodeLensServiceDocumentChangeListener(private val project: Project) : File
 
     private fun installDocumentListener(file: VirtualFile, languageService: LanguageService) {
 
-        val psiFile = PsiManager.getInstance(project).findFile(file)
-        if (PsiUtils.isValidPsiFile(psiFile) && languageService.isRelevant(file)) {
-            val document = PsiDocumentManager.getInstance(project).getDocument(psiFile!!)
+        val psiFile = allowSlowOperation(Supplier { PsiManager.getInstance(project).findFile(file) })
+        if (psiFile != null && PsiUtils.isValidPsiFile(psiFile) && languageService.isRelevant(file)) {
+            val document = PsiDocumentManager.getInstance(project).getDocument(psiFile)
             if (document != null) {
 
                 val parentDisposable = Disposer.newDisposable()
@@ -75,7 +79,7 @@ class CodeLensServiceDocumentChangeListener(private val project: Project) : File
 
                 document.addDocumentListener(object : DocumentListener {
 
-                    private val documentChangeAlarm = AlarmFactory.getInstance().create(
+                    private val documentChangeAlarm = Alarm(
                         Alarm.ThreadToUse.POOLED_THREAD, parentDisposable
                     )
 

jvm-common/src/main/kotlin/org/digma/intellij/plugin/idea/navigation/DocumentsChangeListenerForJvmNavigationDiscovery.kt

@@ -12,14 +12,15 @@ import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.psi.PsiDocumentManager
 import com.intellij.psi.PsiManager
 import com.intellij.util.Alarm
-import com.intellij.util.AlarmFactory
+import org.digma.intellij.plugin.common.allowSlowOperation
 import org.digma.intellij.plugin.common.isProjectValid
 import org.digma.intellij.plugin.common.isValidVirtualFile
 import org.digma.intellij.plugin.errorreporting.ErrorReporter
 import org.digma.intellij.plugin.idea.psi.JvmLanguageService
 import org.digma.intellij.plugin.log.Log
 import org.digma.intellij.plugin.psi.LanguageService
 import org.digma.intellij.plugin.psi.PsiUtils
+import java.util.function.Supplier
 
 /**
  * a listener for document change that updates navigation discovery on document changed events.
@@ -44,7 +45,9 @@ class DocumentsChangeListenerForJvmNavigationDiscovery(private val project: Proj
                 return
             }
 
-            val languageService = LanguageService.findLanguageServiceByFile(project, file)
+            val languageService = allowSlowOperation(Supplier {
+                LanguageService.findLanguageServiceByFile(project, file)
+            })
 
             //only jvm languages are supported here
             if (!JvmLanguageService::class.java.isAssignableFrom(languageService.javaClass)) {
@@ -62,9 +65,9 @@ class DocumentsChangeListenerForJvmNavigationDiscovery(private val project: Proj
 
     private fun installDocumentListener(file: VirtualFile, languageService: LanguageService) {
 
-        val psiFile = PsiManager.getInstance(project).findFile(file)
-        if (PsiUtils.isValidPsiFile(psiFile) && languageService.isRelevant(file)) {
-            val document = PsiDocumentManager.getInstance(project).getDocument(psiFile!!)
+        val psiFile = allowSlowOperation(Supplier { PsiManager.getInstance(project).findFile(file) })
+        if (psiFile != null && PsiUtils.isValidPsiFile(psiFile) && languageService.isRelevant(file)) {
+            val document = PsiDocumentManager.getInstance(project).getDocument(psiFile)
             if (document != null) {
 
                 val parentDisposable = Disposer.newDisposable()
@@ -73,7 +76,7 @@ class DocumentsChangeListenerForJvmNavigationDiscovery(private val project: Proj
 
                 document.addDocumentListener(object : DocumentListener {
 
-                    private val documentChangeAlarm = AlarmFactory.getInstance().create(
+                    private val documentChangeAlarm = Alarm(
                         Alarm.ThreadToUse.POOLED_THREAD, parentDisposable
                     )