Do not render components until state is restored

sample/BitzArt.Blazor.MVVM.SampleApp/BitzArt.Blazor.MVVM.SampleApp.Client/ViewModels/TestPageViewModel.cs

@@ -1,14 +1,20 @@
-namespace BitzArt.Blazor.MVVM.SampleApp;
+
+namespace BitzArt.Blazor.MVVM.SampleApp;
 
 public class TestPageViewModel : ViewModel<TestPageState>
 {
-    protected override void OnDependenciesInjected()
+    public override async Task InitializeStateAsync()
     {
-        OnStateInitializedAsync += async (_) =>
-        {
-            State.Text = "Test Page State initialized";
-            await ComponentStateChangedAsync();
-        };
+        State.Text = "Test Page State was initialized";
+        await ComponentStateChangedAsync();
+    }
+
+    public override async Task OnStateRestoredAsync()
+    {
+        await Task.Delay(3000);
+
+        State.Text = "Test Page State was restored";
+        await ComponentStateChangedAsync();
     }
 }
 

sample/BitzArt.Blazor.MVVM.SampleApp/BitzArt.Blazor.MVVM.SampleApp/Components/Pages/Weather.razor

@@ -42,7 +42,7 @@ else
     protected override async Task OnInitializedAsync()
     {
         // Simulate asynchronous loading to demonstrate streaming rendering
-        await Task.Delay(500);
+        await Task.Delay(5000);
 
         var startDate = DateOnly.FromDateTime(DateTime.Now);
         var summaries = new[] { "Freezing", "Bracing", "Chilly", "Cool", "Mild", "Warm", "Balmy", "Hot", "Sweltering", "Scorching" };

src/BitzArt.Blazor.MVVM/Components/ComponentBase{TViewModel}.cs

@@ -1,5 +1,6 @@
 using Microsoft.AspNetCore.Components;
 using Microsoft.JSInterop;
+using System.Reflection;
 
 namespace BitzArt.Blazor.MVVM;
 
@@ -30,6 +31,20 @@ public abstract class ComponentBase<TViewModel> : ComponentBase, IStateComponent
     [Inject]
     protected NavigationManager NavigationManager { get; set; } = null!;
 
+    private readonly FieldInfo _hasPendingQueuedRenderField;
+
+    protected ComponentBase()
+    {
+        _hasPendingQueuedRenderField = typeof(ComponentBase)
+            .GetField("_hasPendingQueuedRender", BindingFlags.Instance | BindingFlags.NonPublic)!;
+    }
+
+    protected override void OnInitialized()
+    {
+        _hasPendingQueuedRenderField.SetValue(this, true);
+        base.OnInitialized();
+    }
+
     /// <summary>
     /// Method invoked when the component is ready to start, having received its initial
     /// parameters from its parent in the render tree. Override this method if you will
@@ -47,6 +62,8 @@ protected override async Task OnInitializedAsync()
             await InvokeAsync(StateHasChanged);
         };
         await InvokeAsync(StateHasChanged);
+
+        _hasPendingQueuedRenderField.SetValue(this, false);
     }
 
     private async Task SetupStateAsync()