使用星尘时，优化基础类型的注入，例如IConfigProvider

NewLifeX · Oct 31, 2024 · 985d464 · 985d464
1 parent 848cbb9
commit 985d464
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/Stardust.Extensions/StarFactoryExtensions.cs b/Stardust.Extensions/StarFactoryExtensions.cs
@@ -49,7 +49,10 @@ public static StarFactory AddStardust(this IServiceCollection services, String?
         //services.Replace(new ServiceDescriptor(typeof(IConfigProvider), p => star.Config, ServiceLifetime.Singleton));
         //var old = services.LastOrDefault(e => e.ServiceType == typeof(IConfigProvider))?.ImplementationInstance as IConfigProvider;
         //old ??= JsonConfigProvider.LoadAppSettings();
-        services.Replace(new ServiceDescriptor(typeof(IConfigProvider), p => star.GetConfig()!, ServiceLifetime.Singleton));
+        if (services.Any(e => e.ServiceType == typeof(IConfigProvider)))
+            services.Replace(new ServiceDescriptor(typeof(IConfigProvider), p => star.GetConfig()!, ServiceLifetime.Singleton));
+        else
+            services.TryAddSingleton(p => star.GetConfig()!);
 
         // 分布式缓存
         //services.Replace(new ServiceDescriptor(typeof(CacheService), p => new RedisCacheService(p), ServiceLifetime.Singleton));

diff --git a/Stardust/StarHelper.cs b/Stardust/StarHelper.cs
@@ -26,10 +26,7 @@ public static StarFactory AddStardust(this IObjectContainer services, String? se
         var old = JsonConfigProvider.LoadAppSettings();
         star.SetLocalConfig(old);
 
-        if (services != ObjectContainer.Current && services is ObjectContainer container)
-        {
-            star.Register(services);
-        }
+        star.Register(services);
 
         return star;
     }