From 65e52142be22d89b3adae239af2894274c2c94f9 Mon Sep 17 00:00:00 2001
From: peter <peter@jetbrains.com>
Date: Tue, 6 May 2014 15:44:58 +0200
Subject: [PATCH] merge EditorNotifications requests, update automatically on
 dumb mode changes

---
 .../com/intellij/ui/EditorNotifications.java  |  8 +----
 .../intellij/ui/EditorNotificationsImpl.java  | 31 ++++++++++++++++++-
 2 files changed, 31 insertions(+), 8 deletions(-)

diff --git a/platform/platform-api/src/com/intellij/ui/EditorNotifications.java b/platform/platform-api/src/com/intellij/ui/EditorNotifications.java
index ff8f096c6aff..4db32d94dd30 100644
--- a/platform/platform-api/src/com/intellij/ui/EditorNotifications.java
+++ b/platform/platform-api/src/com/intellij/ui/EditorNotifications.java
@@ -17,7 +17,6 @@ package com.intellij.ui;
 
 import com.intellij.openapi.components.AbstractProjectComponent;
 import com.intellij.openapi.fileEditor.FileEditor;
-import com.intellij.openapi.fileEditor.FileEditorManager;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.project.ProjectManager;
 import com.intellij.openapi.util.Key;
@@ -48,12 +47,7 @@ public abstract class EditorNotifications extends AbstractProjectComponent {
 
   public abstract void updateNotifications(final VirtualFile file);
 
-  public void updateAllNotifications() {
-    VirtualFile[] files = FileEditorManager.getInstance(myProject).getOpenFiles();
-    for (VirtualFile file : files) {
-      updateNotifications(file);
-    }
-  }
+  public abstract void updateAllNotifications();
 
   public static void updateAll() {
     Project[] projects = ProjectManager.getInstance().getOpenProjects();
diff --git a/platform/platform-impl/src/com/intellij/ui/EditorNotificationsImpl.java b/platform/platform-impl/src/com/intellij/ui/EditorNotificationsImpl.java
index eb727d9d6200..c5525e5685b4 100644
--- a/platform/platform-impl/src/com/intellij/ui/EditorNotificationsImpl.java
+++ b/platform/platform-impl/src/com/intellij/ui/EditorNotificationsImpl.java
@@ -26,12 +26,16 @@ import com.intellij.openapi.progress.ProgressIndicator;
 import com.intellij.openapi.progress.util.ProgressIndicatorBase;
 import com.intellij.openapi.progress.util.ProgressIndicatorUtils;
 import com.intellij.openapi.progress.util.ReadTask;
+import com.intellij.openapi.project.DumbService;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.util.Key;
 import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.util.containers.ConcurrentWeakHashMap;
 import com.intellij.util.containers.ContainerUtil;
+import com.intellij.util.messages.MessageBusConnection;
 import com.intellij.util.ui.UIUtil;
+import com.intellij.util.ui.update.MergingUpdateQueue;
+import com.intellij.util.ui.update.Update;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
@@ -45,15 +49,30 @@ import java.util.Map;
 public class EditorNotificationsImpl extends EditorNotifications {
   private static final ExtensionPointName<Provider> EXTENSION_POINT_NAME = ExtensionPointName.create("com.intellij.editorNotificationProvider");
   private final Map<VirtualFile, ProgressIndicator> myCurrentUpdates = new ConcurrentWeakHashMap<VirtualFile, ProgressIndicator>();
+  private final MergingUpdateQueue myUpdateMerger;
 
   public EditorNotificationsImpl(Project project) {
     super(project);
-    project.getMessageBus().connect(project).subscribe(FileEditorManagerListener.FILE_EDITOR_MANAGER, new FileEditorManagerAdapter() {
+    myUpdateMerger = new MergingUpdateQueue("EditorNotifications update merger", 100, true, null);
+    MessageBusConnection connection = project.getMessageBus().connect(project);
+    connection.subscribe(FileEditorManagerListener.FILE_EDITOR_MANAGER, new FileEditorManagerAdapter() {
       @Override
       public void fileOpened(@NotNull FileEditorManager source, @NotNull VirtualFile file) {
         updateNotifications(file);
       }
     });
+    connection.subscribe(DumbService.DUMB_MODE, new DumbService.DumbModeListener() {
+      @Override
+      public void enteredDumbMode() {
+        updateAllNotifications();
+      }
+
+      @Override
+      public void exitDumbMode() {
+        updateAllNotifications();
+      }
+    });
+
   }
 
   public void updateNotifications(final VirtualFile file) {
@@ -137,4 +156,14 @@ public class EditorNotificationsImpl extends EditorNotifications {
     }
   }
 
+  public void updateAllNotifications() {
+    myUpdateMerger.queue(new Update("update") {
+      @Override
+      public void run() {
+        for (VirtualFile file : FileEditorManager.getInstance(myProject).getOpenFiles()) {
+          updateNotifications(file);
+        }
+      }
+    });
+  }
 }