InvokeAsync for UiMessageAlert

framework/src/Volo.Abp.BlazoriseUI/Components/EntityAction.razor.cs

@@ -53,10 +53,13 @@ namespace Volo.Abp.BlazoriseUI.Components
         {
             if (ConfirmationMessage != null)
             {
-                if (await UiMessageService.Confirm(ConfirmationMessage()))
+                await InvokeAsync(async () =>
                 {
-                    await Clicked.InvokeAsync();
-                }
+                    if (await UiMessageService.Confirm(ConfirmationMessage()))
+                    {
+                        await Clicked.InvokeAsync();
+                    }
+                });
             }
             else
             {

framework/src/Volo.Abp.BlazoriseUI/Components/UiMessageAlert.razor.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using System.Text;
 using System.Threading.Tasks;
 using Blazorise;
@@ -87,7 +88,7 @@ namespace Volo.Abp.BlazoriseUI.Components
             UiMessageService.MessageReceived += OnMessageReceived;
         }
 
-        private void OnMessageReceived(object sender, UiMessageEventArgs e)
+        private async void OnMessageReceived(object sender, UiMessageEventArgs e)
         {
             MessageType = e.MessageType;
             Message = e.Message;
@@ -95,7 +96,7 @@ namespace Volo.Abp.BlazoriseUI.Components
             Options = e.Options;
             Callback = e.Callback;
 
-            ModalRef.Show();
+            await InvokeAsync(ModalRef.Show);
         }
 
         public void Dispose()
@@ -106,35 +107,44 @@ namespace Volo.Abp.BlazoriseUI.Components
             }
         }
 
-        protected Task OnOkClicked()
+        protected async Task OnOkClicked()
         {
-            ModalRef.Hide();
+            await InvokeAsync(async () =>
+            {
+                await Okayed.InvokeAsync(null);
 
-            return Okayed.InvokeAsync(null);
+                ModalRef.Hide();
+            });
         }
 
-        protected Task OnConfirmClicked()
+        protected async Task OnConfirmClicked()
         {
-            ModalRef.Hide();
-
-            if (IsConfirmation && Callback != null)
+            await InvokeAsync(async () =>
             {
-                Callback.SetResult(true);
-            }
+                ModalRef.Hide();
+
+                if (IsConfirmation && Callback != null)
+                {
+                    await InvokeAsync(() => Callback.SetResult(true));
+                }
 
-            return Confirmed.InvokeAsync(null);
+                await Confirmed.InvokeAsync(null);
+            });
         }
 
-        protected Task OnCancelClicked()
+        protected async Task OnCancelClicked()
         {
-            ModalRef.Hide();
-
-            if (IsConfirmation && Callback != null)
+            await InvokeAsync(async () =>
             {
-                Callback.SetResult(false);
-            }
+                ModalRef.Hide();
+
+                if (IsConfirmation && Callback != null)
+                {
+                    await InvokeAsync(() => Callback.SetResult(false));
+                }
 
-            return Canceled.InvokeAsync(null);
+                await Canceled.InvokeAsync(null);
+            });
         }
 
         protected virtual void OnModalClosing(ModalClosingEventArgs eventArgs)