fix: run more UI refreshes async & in read action

snyk · Mar 20, 2024 · 0ec30a6 · 0ec30a6
1 parent 09fe662
commit 0ec30a6
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 7 deletions.
diff --git a/src/main/kotlin/io/snyk/plugin/services/SnykTaskQueueService.kt b/src/main/kotlin/io/snyk/plugin/services/SnykTaskQueueService.kt
@@ -1,6 +1,8 @@
 package io.snyk.plugin.services
 
 import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.application.ReadAction
+import com.intellij.openapi.application.invokeLater
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.fileEditor.FileDocumentManager
@@ -153,7 +155,7 @@ class SnykTaskQueueService(val project: Project) {
                     }
                 }
                 logger.debug("Container scan completed")
-                refreshAnnotationsForOpenFiles(project)
+                invokeLater { refreshAnnotationsForOpenFiles(project) }
             }
         })
     }
@@ -228,7 +230,7 @@ class SnykTaskQueueService(val project: Project) {
                         ossResult.getFirstError()?.let { scanPublisher?.scanningOssError(it) }
                     }
                 }
-                refreshAnnotationsForOpenFiles(project)
+                invokeLater { refreshAnnotationsForOpenFiles(project) }
             }
         })
     }
@@ -271,7 +273,7 @@ class SnykTaskQueueService(val project: Project) {
                     }
                 }
                 logger.debug("IaC scan completed")
-                refreshAnnotationsForOpenFiles(project)
+                invokeLater { refreshAnnotationsForOpenFiles(project) }
             }
         })
     }

diff --git a/src/main/kotlin/io/snyk/plugin/ui/toolwindow/SnykToolWindowPanel.kt b/src/main/kotlin/io/snyk/plugin/ui/toolwindow/SnykToolWindowPanel.kt
@@ -444,10 +444,10 @@ class SnykToolWindowPanel(val project: Project) : JPanel(), Disposable {
             }
         }
 
-        refreshAnnotationsForOpenFiles(project)
 
         ApplicationManager.getApplication().invokeLater {
             doCleanUi(true)
+            refreshAnnotationsForOpenFiles(project)
         }
     }
 

diff --git a/src/main/kotlin/snyk/common/AnnotatorCommon.kt b/src/main/kotlin/snyk/common/AnnotatorCommon.kt
@@ -1,5 +1,6 @@
 package snyk.common
 
+import com.intellij.openapi.application.invokeLater
 import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiFile
@@ -33,7 +34,7 @@ object AnnotatorCommon {
                 SnykProductsOrSeverityListener.SNYK_ENABLEMENT_TOPIC,
                 object : SnykProductsOrSeverityListener {
                     override fun enablementChanged() {
-                        refreshAnnotationsForOpenFiles(project)
+                        invokeLater { refreshAnnotationsForOpenFiles(project) }
                     }
                 }
             )
@@ -42,7 +43,7 @@ object AnnotatorCommon {
                 SnykSettingsListener.SNYK_SETTINGS_TOPIC,
                 object : SnykSettingsListener {
                     override fun settingsChanged() {
-                        refreshAnnotationsForOpenFiles(project)
+                        invokeLater { refreshAnnotationsForOpenFiles(project) }
                     }
                 }
             )

diff --git a/src/main/kotlin/snyk/container/ContainerBulkFileListener.kt b/src/main/kotlin/snyk/container/ContainerBulkFileListener.kt
@@ -74,8 +74,8 @@ class ContainerBulkFileListener : SnykBulkFileListener() {
         snykCachedResults.currentContainerResult = newContainerCache
         ApplicationManager.getApplication().invokeLater {
             getSnykToolWindowPanel(project)?.displayContainerResults(newContainerCache)
+            refreshAnnotationsForOpenFiles(project)
         }
-        refreshAnnotationsForOpenFiles(project)
     }
 
     private fun makeObsolete(containerIssuesForImage: ContainerIssuesForImage): ContainerIssuesForImage =