Fix some slow operations on EDT (#2750)

Author: rli

.changes/next-release/bugfix-cbbd8be9-0cab-4b04-ba8e-a42cc67eaf09.json

@@ -0,0 +1,4 @@
+{
+  "type" : "bugfix",
+  "description" : "Fix some warnings due to slow operations on EDT (#2735)"
+}

jetbrains-core/src/software/aws/toolkits/jetbrains/services/lambda/execution/sam/SamInvokeRunner.kt

@@ -3,6 +3,7 @@
 
 package software.aws.toolkits.jetbrains.services.lambda.execution.sam
 
+import com.intellij.execution.ExecutionResult
 import com.intellij.execution.configurations.RunProfile
 import com.intellij.execution.configurations.RunProfileState
 import com.intellij.execution.configurations.RunnerSettings
@@ -12,6 +13,9 @@ import com.intellij.execution.runners.AsyncProgramRunner
 import com.intellij.execution.runners.ExecutionEnvironment
 import com.intellij.execution.runners.RunContentBuilder
 import com.intellij.execution.ui.RunContentDescriptor
+import com.intellij.openapi.application.ModalityState
+import com.intellij.openapi.application.ReadAction
+import com.intellij.util.concurrency.AppExecutorUtil
 import org.jetbrains.concurrency.AsyncPromise
 import org.jetbrains.concurrency.Promise
 import org.slf4j.event.Level
@@ -68,11 +72,14 @@ class SamInvokeRunner : AsyncProgramRunner<RunnerSettings>() {
 
     override fun execute(environment: ExecutionEnvironment, state: RunProfileState): Promise<RunContentDescriptor?> {
         val runPromise = AsyncPromise<RunContentDescriptor?>()
-        val runContentDescriptor = state.execute(environment.executor, this)?.let {
-            RunContentBuilder(it, environment).showRunContent(environment.contentToReuse)
-        }
-
-        runPromise.setResult(runContentDescriptor)
+        ReadAction.nonBlocking<ExecutionResult> {
+            state.execute(environment.executor, this)
+        }.finishOnUiThread(ModalityState.any()) { executionResult ->
+            executionResult?.let {
+                val runContent = RunContentBuilder(it, environment).showRunContent(environment.contentToReuse)
+                runPromise.setResult(runContent)
+            }
+        }.submit(AppExecutorUtil.getAppExecutorService())
 
         return runPromise
     }

jetbrains-core/src/software/aws/toolkits/jetbrains/services/lambda/upload/CreateLambdaFunctionAction.kt

@@ -6,6 +6,7 @@ package software.aws.toolkits.jetbrains.services.lambda.upload
 import com.intellij.openapi.actionSystem.AnAction
 import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.actionSystem.LangDataKeys
+import com.intellij.openapi.actionSystem.UpdateInBackground
 import com.intellij.psi.PsiElement
 import com.intellij.psi.SmartPsiElementPointer
 import icons.AwsIcons
@@ -20,7 +21,7 @@ class CreateLambdaFunctionAction(
     private val handlerName: String?,
     private val elementPointer: SmartPsiElementPointer<PsiElement>?,
     private val lambdaHandlerResolver: LambdaHandlerResolver?
-) : AnAction(message("lambda.create_new"), null, AwsIcons.Actions.LAMBDA_FUNCTION_NEW) {
+) : AnAction(message("lambda.create_new"), null, AwsIcons.Actions.LAMBDA_FUNCTION_NEW), UpdateInBackground {
 
     @Suppress("unused") // Used by ActionManager in plugin.xml
     constructor() : this(null, null, null)