Use BGT to update UI for restart/reload (#6727)

Fixes #6711

Author: stevemessick

flutter-idea/src/io/flutter/actions/FlutterRetargetAppAction.java

@@ -5,9 +5,7 @@
  */
 package io.flutter.actions;
 
-import com.intellij.openapi.actionSystem.AnAction;
-import com.intellij.openapi.actionSystem.AnActionEvent;
-import com.intellij.openapi.actionSystem.Presentation;
+import com.intellij.openapi.actionSystem.*;
 import com.intellij.openapi.project.DumbAwareAction;
 import com.intellij.openapi.project.Project;
 import io.flutter.utils.FlutterModuleUtils;
@@ -40,6 +38,10 @@ public abstract class FlutterRetargetAppAction extends DumbAwareAction {
     myPlaces.addAll(Arrays.asList(places));
   }
 
+  public @NotNull ActionUpdateThread getActionUpdateThread() {
+    return ActionUpdateThread.BGT;
+  }
+
   @Override
   public void actionPerformed(AnActionEvent e) {
     final AnAction action = getAction(e.getProject());

tool/plugin/lib/edit.dart

@@ -87,7 +87,17 @@ List<EditCommand> editCommands = [
   ),
   Subst(
     path:
-        'flutter-idea/src/io/flutter/actions/DeviceSelectorRefresherAction.java',
+    'flutter-idea/src/io/flutter/actions/DeviceSelectorRefresherAction.java',
+    initial: '''
+  public @NotNull ActionUpdateThread getActionUpdateThread() {
+    return ActionUpdateThread.BGT;
+  }
+''',
+    replacement: '',
+    version: '2022.1',
+  ),
+  Subst(
+    path: 'flutter-idea/src/io/flutter/actions/FlutterRetargetAppAction.java',
     initial: '''
   public @NotNull ActionUpdateThread getActionUpdateThread() {
     return ActionUpdateThread.BGT;