ensure active

Author: shalom938

jvm-common/src/main/kotlin/org/digma/intellij/plugin/idea/discovery/AbstractCodeObjectDiscovery.kt

@@ -3,6 +3,7 @@ package org.digma.intellij.plugin.idea.discovery
 import com.intellij.lang.Language
 import com.intellij.openapi.application.readAction
 import com.intellij.openapi.diagnostic.Logger
+import com.intellij.openapi.project.DumbService
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiFile
 import kotlinx.coroutines.ensureActive
@@ -86,7 +87,7 @@ abstract class AbstractCodeObjectDiscovery(private val spanDiscovery: AbstractSp
             //don't call this method on EDT or in read access, read access is taken when needed
             EDT.assertNonDispatchThread()
             ReadActions.assertNotInReadAccess()
-
+            DumbService.getInstance(project).waitForSmartMode()
             //maybe uFile is null,there is nothing to do without a UFile.
             val fileData = readAction {
                 FileData.buildFileData(psiFile)

jvm-common/src/main/kotlin/org/digma/intellij/plugin/idea/discovery/AbstractJvmFileDiscoveryProvider.kt

@@ -1,6 +1,7 @@
 package org.digma.intellij.plugin.idea.discovery
 
 import com.intellij.openapi.diagnostic.thisLogger
+import com.intellij.openapi.project.DumbService
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.vfs.VirtualFile
 import org.digma.intellij.plugin.discovery.FileDiscoveryProvider
@@ -27,7 +28,7 @@ abstract class AbstractJvmFileDiscoveryProvider: FileDiscoveryProvider {
         project: Project,
         file: VirtualFile
     ): FileDiscoveryInfo {
-
+        DumbService.getInstance(project).waitForSmartMode()
         Log.trace(logger, project, "starting discovery for {}", file)
 
         val languageService = LanguageServiceProvider.getInstance(project).getLanguageService(file)