[Blazor] Resolve collection for registered services once (#61391)

* Avoid checking the shared cache on a per-request basis to resolve the type for existing persistent service registrations
* Avoid resolving registrations on a per-request basis

Author: javiercn

src/Components/Components/src/PersistentState/IPersistentServiceRegistration.cs

@@ -10,4 +10,6 @@ internal interface IPersistentServiceRegistration
     public string FullTypeName { get; }
 
     public IComponentRenderMode? GetRenderModeOrDefault();
+
+    public Type? GetResolvedTypeOrNull() => null;
 }

src/Components/Components/src/PersistentState/PersistentServiceRegistration.cs

@@ -13,5 +13,7 @@ internal sealed class PersistentServiceRegistration<TService>(IComponentRenderMo
 
     public IComponentRenderMode? GetRenderModeOrDefault() => componentRenderMode;
 
+    public Type? GetResolvedTypeOrNull() => typeof(TService);
+
     private string GetDebuggerDisplay() => $"{Assembly}::{FullTypeName}";
 }

src/Components/Components/src/PersistentState/PersistentServicesRegistry.cs

@@ -26,9 +26,8 @@ internal sealed class PersistentServicesRegistry
 
     public PersistentServicesRegistry(IServiceProvider serviceProvider)
     {
-        var registrations = serviceProvider.GetRequiredService<IEnumerable<IPersistentServiceRegistration>>();
         _serviceProvider = serviceProvider;
-        _registrations = ResolveRegistrations(registrations);
+        _registrations = ResolveRegistrations(serviceProvider.GetService<RegisteredPersistentServiceRegistrationCollection>()?.Registrations ?? []);
     }
 
     internal IComponentRenderMode? RenderMode { get; set; }
@@ -50,7 +49,7 @@ internal void RegisterForPersistence(PersistentComponentState state)
         for (var i = 0; i < _registrations.Length; i++)
         {
             var registration = _registrations[i];
-            var type = ResolveType(registration.Assembly, registration.FullTypeName);
+            var type = ResolveType(registration);
             if (type == null)
             {
                 continue;
@@ -112,7 +111,7 @@ private void RestoreRegistrationsIfAvailable(PersistentComponentState state)
     {
         foreach (var registration in _registrations)
         {
-            var type = ResolveType(registration.Assembly, registration.FullTypeName);
+            var type = ResolveType(registration);
             if (type == null)
             {
                 continue;
@@ -140,10 +139,18 @@ private static void RestoreInstanceState(object instance, Type type, PersistentC
         }
     }
 
-    private static IPersistentServiceRegistration[] ResolveRegistrations(IEnumerable<IPersistentServiceRegistration> registrations) => [.. registrations.DistinctBy(r => (r.Assembly, r.FullTypeName)).OrderBy(r => r.Assembly).ThenBy(r => r.FullTypeName)];
+    internal static IPersistentServiceRegistration[] ResolveRegistrations(IEnumerable<IPersistentServiceRegistration> registrations) => [.. registrations.DistinctBy(r => (r.Assembly, r.FullTypeName)).OrderBy(r => r.Assembly).ThenBy(r => r.FullTypeName)];
 
-    private static Type? ResolveType(string assembly, string fullTypeName) =>
-        _persistentServiceTypeCache.GetRootType(assembly, fullTypeName);
+    private static Type? ResolveType(IPersistentServiceRegistration registration)
+    {
+        if (registration.GetResolvedTypeOrNull() is Type type)
+        {
+            return type;
+        }
+        var assembly = registration.Assembly;
+        var fullTypeName = registration.FullTypeName;
+        return _persistentServiceTypeCache.GetRootType(assembly, fullTypeName);
+    }
 
     private sealed class PropertiesAccessor
     {

src/Components/Components/src/PersistentState/RegisteredPersistentServiceRegistrationCollection.cs

@@ -0,0 +1,12 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+namespace Microsoft.AspNetCore.Components.Infrastructure;
+
+internal class RegisteredPersistentServiceRegistrationCollection(IEnumerable<IPersistentServiceRegistration> registrations)
+{
+    private readonly IEnumerable<IPersistentServiceRegistration> _registrations =
+        PersistentServicesRegistry.ResolveRegistrations(registrations);
+
+    public IEnumerable<IPersistentServiceRegistration> Registrations => _registrations;
+}

src/Components/Components/src/RegisterPersistentComponentStateServiceCollectionExtensions.cs

@@ -36,6 +36,7 @@ public static class RegisterPersistentComponentStateServiceCollectionExtensions
         // We resolve the service from the DI container.
         // We loop through the properties in the type and try to restore the properties that have SupplyParameterFromPersistentComponentState on them.
         services.TryAddEnumerable(ServiceDescriptor.Singleton<IPersistentServiceRegistration>(new PersistentServiceRegistration<TService>(componentRenderMode)));
+        services.TryAddSingleton<RegisteredPersistentServiceRegistrationCollection>();
 
         return services;
     }

src/Components/Components/test/PersistentState/PersistentServicesRegistryTest.cs

@@ -260,6 +260,7 @@ public void ResolveRegistrations_RemovesDuplicateRegistrations()
     {
         // Arrange
         var serviceProvider = new ServiceCollection()
+            .AddSingleton<RegisteredPersistentServiceRegistrationCollection>()
             .AddSingleton<IPersistentServiceRegistration>(new TestPersistentRegistration { Assembly = "Assembly1", FullTypeName = "Type1" })
             .AddSingleton<IPersistentServiceRegistration>(new TestPersistentRegistration { Assembly = "Assembly1", FullTypeName = "Type1" }) // Duplicate
             .AddSingleton<IPersistentServiceRegistration>(new TestPersistentRegistration { Assembly = "Assembly2", FullTypeName = "Type2" })