// Copyright (c) 2025-2026 GeWuYou // SPDX-License-Identifier: Apache-2.0 using System.Reflection; using System.Threading; using GFramework.Core.Abstractions.Bases; using GFramework.Core.Abstractions.Ioc; using GFramework.Core.Abstractions.Logging; using GFramework.Core.Abstractions.Systems; using GFramework.Core.Rule; using GFramework.Cqrs; using GFramework.Cqrs.Abstractions.Cqrs; namespace GFramework.Core.Ioc; /// /// Microsoft.Extensions.DependencyInjection 适配器 /// 将 Microsoft DI 包装为 IIocContainer 接口实现 /// 提供线程安全的依赖注入容器功能 /// /// /// 该适配器负责维护服务注册表、冻结后的根 以及并发访问控制。 /// 容器释放后会阻止任何进一步访问,并统一抛出 , /// 以避免 benchmark、测试宿主或短生命周期架构误用失效的 DI 状态。 /// /// 可选的IServiceCollection实例,默认创建新的ServiceCollection public class MicrosoftDiContainer(IServiceCollection? serviceCollection = null) : ContextAwareBase, IIocContainer { #region Helper Methods /// /// 抛出统一的容器释放异常,避免并发路径泄露底层锁类型的实现细节。 /// /// 始终抛出,且对象名固定为当前容器类型。 private void ThrowDisposedException() { const string objectName = nameof(MicrosoftDiContainer); _logger.Warn("Attempted to use a disposed MicrosoftDiContainer."); throw new ObjectDisposedException(objectName); } /// /// 检查容器是否已释放,避免访问已经失效的服务提供者与同步原语。 /// /// 当容器已释放时抛出。 private void ThrowIfDisposed() { if (!_disposed) return; ThrowDisposedException(); } /// /// 进入读锁,并在获取锁前后都复核释放状态,确保等待中的线程也能稳定得到容器级异常。 /// /// 当容器已释放,或等待期间被其他线程释放时抛出。 private void EnterReadLockOrThrowDisposed() { var lockTaken = false; try { _lock.EnterReadLock(); lockTaken = true; } catch (ObjectDisposedException) when (_disposed) { ThrowDisposedException(); } if (!_disposed) { return; } if (lockTaken) { _lock.ExitReadLock(); } ThrowDisposedException(); } /// /// 进入写锁,并在获取锁前后都复核释放状态,确保等待中的线程不会泄露底层锁异常。 /// /// 当容器已释放,或等待期间被其他线程释放时抛出。 private void EnterWriteLockOrThrowDisposed() { var lockTaken = false; try { _lock.EnterWriteLock(); lockTaken = true; } catch (ObjectDisposedException) when (_disposed) { ThrowDisposedException(); } if (!_disposed) { return; } if (lockTaken) { _lock.ExitWriteLock(); } ThrowDisposedException(); } /// /// 在释放标志已经对外可见后,等待遗留 waiter 退场,再尝试释放底层锁。 /// /// /// 容器会先把 置为 并退出写锁, /// 这样所有已在等待队列中的线程都能醒来并通过统一路径抛出容器级 /// 。只有当这些线程退场后,底层锁才可安全释放。 /// 该步骤只允许一个释放调用者执行,避免并发 重复销毁同一个 /// 并破坏幂等契约。 /// private void DisposeLockWhenQuiescent() { if (Interlocked.CompareExchange(ref _lockDisposalStarted, 1, 0) != 0) { return; } const int maxDisposeSpinAttempts = 512; var spinWait = new SpinWait(); for (var attempt = 0; attempt < maxDisposeSpinAttempts; attempt++) { if (_lock.CurrentReadCount == 0 && _lock.WaitingReadCount == 0 && _lock.WaitingWriteCount == 0 && _lock.WaitingUpgradeCount == 0) { try { _lock.Dispose(); return; } catch (SynchronizationLockException) { // 等待中的线程刚好在本轮检查后切换状态;继续自旋直到锁真正静默。 } } spinWait.SpinOnce(); } _logger.Warn("MicrosoftDiContainer lock disposal was skipped because waiters did not quiesce in time."); } /// /// 检查容器是否已冻结,如果已冻结则抛出异常 /// 用于保护注册操作的安全性 /// /// 当容器已冻结时抛出 private void ThrowIfFrozen() { if (!_frozen) return; const string errorMsg = "MicrosoftDiContainer is frozen"; _logger.Error(errorMsg); throw new InvalidOperationException(errorMsg); } #endregion /// /// 记录某个实例在未冻结查询中可见的服务类型分组信息。 /// /// 当前分组对应的服务类型。 /// 该服务类型下的描述符数量。 /// 该服务类型首次出现的位置,用于稳定打破并列。 private sealed record VisibleServiceTypeGroup(Type ServiceType, int Count, int FirstIndex); #region Fields /// /// 服务提供者,在容器冻结后构建,用于解析服务实例 /// private IServiceProvider? _provider; /// /// 冻结后可复用的服务类型可见性索引。 /// 容器冻结后注册集合不再变化,因此 可以安全复用该索引。 /// private FrozenServiceTypeIndex? _frozenServiceTypeIndex; /// /// 容器冻结状态标志,true表示容器已冻结不可修改 /// private volatile bool _frozen; /// /// 容器释放状态标志,true 表示容器已释放,不允许继续访问。 /// private volatile bool _disposed; /// /// 标记底层读写锁的销毁流程是否已经启动,确保并发释放时最多只有一个线程尝试销毁锁实例。 /// private int _lockDisposalStarted; /// /// 读写锁,确保多线程环境下的线程安全操作 /// private readonly ReaderWriterLockSlim _lock = new(LockRecursionPolicy.NoRecursion); /// /// 已注册实例的集合,用于快速检查实例是否存在 /// private readonly HashSet _registeredInstances = []; /// /// 日志记录器,用于记录容器操作日志 /// private readonly ILogger _logger = LoggerFactoryResolver.Provider.CreateLogger(nameof(MicrosoftDiContainer)); #endregion #region Register /// /// 注册单例服务实例 /// 确保同一类型只能注册一次,避免重复注册 /// /// 服务类型 /// 要注册的实例对象 /// 当容器已冻结或类型已被注册时抛出 public void RegisterSingleton(T instance) { ThrowIfDisposed(); var type = typeof(T); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); // 检查是否已注册该类型,防止重复注册 if (GetServicesUnsafe.Any(s => s.ServiceType == type)) { var errorMsg = $"Singleton already registered for type: {type.Name}"; _logger.Error(errorMsg); throw new InvalidOperationException(errorMsg); } GetServicesUnsafe.AddSingleton(type, instance!); _registeredInstances.Add(instance!); _logger.Debug($"Singleton registered: {type.Name}"); } finally { _lock.ExitWriteLock(); } } /// /// 注册单例服务,指定服务类型和实现类型 /// 直接使用底层DI容器注册类型映射关系 /// /// 服务接口或基类类型 /// 具体的实现类型 public void RegisterSingleton() where TImpl : class, TService where TService : class { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); GetServicesUnsafe.AddSingleton(); _logger.Debug($"Singleton registered: {typeof(TService).Name}"); } finally { _lock.ExitWriteLock(); } } /// /// 注册瞬态服务,指定服务类型和实现类型 /// 每次解析时都会创建新的实例 /// /// 服务接口或基类类型 /// 具体的实现类型 public void RegisterTransient() where TImpl : class, TService where TService : class { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); GetServicesUnsafe.AddTransient(); _logger.Debug($"Transient registered: {typeof(TService).Name}"); } finally { _lock.ExitWriteLock(); } } /// /// 注册作用域服务,指定服务类型和实现类型 /// 在同一作用域内共享实例,不同作用域使用不同实例 /// /// 服务接口或基类类型 /// 具体的实现类型 public void RegisterScoped() where TImpl : class, TService where TService : class { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); GetServicesUnsafe.AddScoped(); _logger.Debug($"Scoped registered: {typeof(TService).Name}"); } finally { _lock.ExitWriteLock(); } } /// /// 注册多个实例到其所有接口和具体类型 /// 实现一个实例支持多种接口类型的解析 /// /// 要注册的对象实例 /// 当容器已冻结时抛出 public void RegisterPlurality(object instance) { ThrowIfDisposed(); var concreteType = instance.GetType(); var interfaces = concreteType.GetInterfaces(); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); // 注册具体类型映射 GetServicesUnsafe.AddSingleton(concreteType, instance); // 注册所有接口类型映射(指向同一实例) foreach (var interfaceType in interfaces) { GetServicesUnsafe.AddSingleton(interfaceType, _ => instance); } _registeredInstances.Add(instance); _logger.Debug($"Plurality registered: {concreteType.Name} with {interfaces.Length} interfaces"); } finally { _lock.ExitWriteLock(); } } /// /// 注册多个实例到其所有接口 /// 实现一个实例支持多种接口类型的解析 /// public void RegisterPlurality() where T : class { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); var concreteType = typeof(T); var interfaces = concreteType.GetInterfaces(); // 注册具体类型 GetServicesUnsafe.AddSingleton(); // 注册所有接口(指向同一个实例) foreach (var interfaceType in interfaces) { GetServicesUnsafe.AddSingleton(interfaceType, sp => sp.GetRequiredService()); } _logger.Debug($"Type registered: {concreteType.Name} with {interfaces.Length} interfaces"); } finally { _lock.ExitWriteLock(); } } /// /// 注册系统实例 /// 通过RegisterPlurality方法注册ISystem类型实例 /// /// 要注册的系统实例 public void RegisterSystem(ISystem system) { RegisterPlurality(system); } /// /// 注册指定泛型类型的服务实例 /// /// 服务类型 /// 要注册的实例对象 /// 当容器已冻结时抛出 public void Register(T instance) { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); GetServicesUnsafe.AddSingleton(typeof(T), instance!); _registeredInstances.Add(instance!); _logger.Debug($"Registered: {typeof(T).Name}"); } finally { _lock.ExitWriteLock(); } } /// /// 注册指定类型的服务实例 /// /// 服务类型 /// 要注册的实例对象 /// 当容器已冻结时抛出 public void Register(Type type, object instance) { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); GetServicesUnsafe.AddSingleton(type, instance); _registeredInstances.Add(instance); _logger.Debug($"Registered: {type.Name}"); } finally { _lock.ExitWriteLock(); } } /// /// 注册工厂方法来创建服务实例 /// 通过委托函数动态创建服务实例,支持依赖注入 /// /// 服务类型 /// 创建服务实例的工厂委托函数,接收IServiceProvider参数 public void RegisterFactory( Func factory) where TService : class { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); GetServicesUnsafe.AddSingleton(factory); } finally { _lock.ExitWriteLock(); } } /// /// 注册 CQRS 请求管道行为。 /// 同时支持开放泛型行为类型和已闭合的具体行为类型, /// 以兼容通用行为和针对单一请求的专用行为两种注册方式。 /// /// 行为类型,必须是引用类型 public void RegisterCqrsPipelineBehavior() where TBehavior : class { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); RegisterCqrsPipelineBehaviorCore( typeof(TBehavior), typeof(IPipelineBehavior<,>), "IPipelineBehavior<,>", "pipeline behavior"); } finally { _lock.ExitWriteLock(); } } /// /// 注册 CQRS 流式请求管道行为。 /// 同时支持开放泛型行为类型和已闭合的具体行为类型, /// 以兼容通用行为和针对单一流式请求的专用行为两种注册方式。 /// /// 行为类型,必须是引用类型 public void RegisterCqrsStreamPipelineBehavior() where TBehavior : class { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); RegisterCqrsPipelineBehaviorCore( typeof(TBehavior), typeof(IStreamPipelineBehavior<,>), "IStreamPipelineBehavior<,>", "stream pipeline behavior"); } finally { _lock.ExitWriteLock(); } } /// /// 复用 CQRS 行为注册的开放泛型/封闭接口校验逻辑, /// 让 request 与 stream 两条入口保持一致的容器注册语义。 /// /// 待注册的行为运行时类型。 /// 行为必须实现的开放泛型接口类型。 /// 用于日志与异常的接口显示名称。 /// 用于日志的注册类别名称。 /// 未实现目标行为接口。 private void RegisterCqrsPipelineBehaviorCore( Type behaviorType, Type openGenericInterfaceType, string interfaceTypeDisplayName, string registrationLabel) { if (behaviorType.IsGenericTypeDefinition) { GetServicesUnsafe.AddSingleton(openGenericInterfaceType, behaviorType); } else { var pipelineInterfaces = behaviorType .GetInterfaces() .Where(type => type.IsGenericType && type.GetGenericTypeDefinition() == openGenericInterfaceType) .ToList(); if (pipelineInterfaces.Count == 0) { var errorMessage = $"{behaviorType.Name} does not implement {interfaceTypeDisplayName}"; _logger.Error(errorMessage); throw new InvalidOperationException(errorMessage); } // 为每个已闭合的行为接口建立显式映射,支持针对特定请求/响应对的专用行为。 foreach (var pipelineInterface in pipelineInterfaces) { GetServicesUnsafe.AddSingleton(pipelineInterface, behaviorType); } } _logger.Debug($"CQRS {registrationLabel} registered: {behaviorType.Name}"); } /// /// 从指定程序集显式注册 CQRS 处理器。 /// /// 包含 CQRS 处理器或生成注册器的程序集。 /// /// 容器已冻结,无法继续注册 CQRS 处理器。 public void RegisterCqrsHandlersFromAssembly(Assembly assembly) { ArgumentNullException.ThrowIfNull(assembly); RegisterCqrsHandlersFromAssemblies([assembly]); } /// /// 从多个程序集显式注册 CQRS 处理器。 /// 同一程序集只会被接入一次,避免默认启动路径与扩展模块重复注册相同 handlers。 /// /// 要接入的程序集集合。 /// /// 中存在 元素。 /// 容器已冻结,无法继续注册 CQRS 处理器。 public void RegisterCqrsHandlersFromAssemblies(IEnumerable assemblies) { ArgumentNullException.ThrowIfNull(assemblies); ThrowIfDisposed(); var assemblyArray = assemblies.ToArray(); foreach (var assembly in assemblyArray) { if (assembly is null) { throw new ArgumentNullException(nameof(assemblies), "Assemblies collection cannot contain null items."); } } EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); ResolveCqrsRegistrationService().RegisterHandlers(assemblyArray); } finally { _lock.ExitWriteLock(); } } /// /// 配置服务 /// /// 服务配置委托 public void ExecuteServicesHook(Action? configurator = null) { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { ThrowIfFrozen(); configurator?.Invoke(GetServicesUnsafe); } finally { _lock.ExitWriteLock(); } } #endregion #region Get /// /// 获取当前容器中已注册的 CQRS 程序集注册协调器。 /// 该方法仅供容器内部在注册阶段使用,因此直接读取服务描述符中的实例绑定, /// 避免在容器未冻结前依赖完整的服务提供者构建流程。 /// /// 已注册的 CQRS 程序集注册协调器实例。 /// 未找到可用的 CQRS 程序集注册协调器实例时抛出。 private ICqrsRegistrationService ResolveCqrsRegistrationService() { var registrationService = CollectRegisteredImplementationInstances(typeof(ICqrsRegistrationService)) .OfType() .LastOrDefault(); if (registrationService != null) return registrationService; const string errorMessage = "ICqrsRegistrationService is not visible during the registration stage. Ensure the CQRS runtime module " + "has been installed and that the registration service is pre-materialized as an instance binding before " + "registering handlers."; _logger.Error(errorMessage); throw new InvalidOperationException(errorMessage); } /// /// 获取指定泛型类型的服务实例 /// 返回第一个匹配的注册实例,如果不存在则返回null /// /// 服务类型 /// 服务实例或null public T? Get() where T : class { ThrowIfDisposed(); EnterReadLockOrThrowDisposed(); try { if (_provider == null) { return CollectRegisteredImplementationInstances(typeof(T)).OfType().FirstOrDefault(); } var result = _provider!.GetService(); _logger.Debug(result != null ? $"Retrieved instance: {typeof(T).Name}" : $"No instance found for type: {typeof(T).Name}"); return result; } finally { _lock.ExitReadLock(); } } /// /// 获取指定类型的服务实例 /// 返回第一个匹配的注册实例,如果不存在则返回null /// /// 服务类型 /// 服务实例或null public object? Get(Type type) { ArgumentNullException.ThrowIfNull(type); ThrowIfDisposed(); EnterReadLockOrThrowDisposed(); try { if (_provider == null) { return CollectRegisteredImplementationInstances(type).FirstOrDefault(); } var result = _provider!.GetService(type); if (_logger.IsDebugEnabled()) { _logger.Debug(result != null ? $"Retrieved instance: {type.Name}" : $"No instance found for type: {type.Name}"); } return result; } finally { _lock.ExitReadLock(); } } /// /// 获取指定泛型类型的必需服务实例 /// 必须存在且唯一,否则抛出异常 /// /// 服务类型 /// 唯一的服务实例 /// 当实例不存在或多于一个时抛出 public T GetRequired() where T : class { var list = GetAll(); switch (list.Count) { case 0: var notFoundMsg = $"No instance registered for {typeof(T).Name}"; _logger.Error(notFoundMsg); throw new InvalidOperationException(notFoundMsg); case 1: _logger.Debug($"Retrieved required instance: {typeof(T).Name}"); return list[0]; default: var multipleMsg = $"Multiple instances registered for {typeof(T).Name}"; _logger.Error(multipleMsg); throw new InvalidOperationException(multipleMsg); } } /// /// 获取指定类型的必需服务实例 /// 必须存在且唯一,否则抛出异常 /// /// 服务类型 /// 唯一的服务实例 /// 当实例不存在或多于一个时抛出 public object GetRequired(Type type) { var list = GetAll(type); switch (list.Count) { case 0: var notFoundMsg = $"No instance registered for {type.Name}"; _logger.Error(notFoundMsg); throw new InvalidOperationException(notFoundMsg); case 1: _logger.Debug($"Retrieved required instance: {type.Name}"); return list[0]; default: var multipleMsg = $"Multiple instances registered for {type.Name}"; _logger.Error(multipleMsg); throw new InvalidOperationException(multipleMsg); } } /// /// 获取指定泛型类型的所有服务实例 /// /// 服务类型 /// 只读的服务实例列表 public IReadOnlyList GetAll() where T : class { ThrowIfDisposed(); EnterReadLockOrThrowDisposed(); try { if (_provider == null) { return CollectRegisteredImplementationInstances(typeof(T)).Cast().ToList(); } var services = _provider!.GetServices().ToList(); if (_logger.IsDebugEnabled()) { _logger.Debug($"Retrieved {services.Count} instances of {typeof(T).Name}"); } return services; } finally { _lock.ExitReadLock(); } } /// /// 获取指定类型的所有服务实例 /// /// 服务类型 /// 只读的服务实例列表 /// 时抛出 public IReadOnlyList GetAll(Type type) { ArgumentNullException.ThrowIfNull(type); ThrowIfDisposed(); EnterReadLockOrThrowDisposed(); try { if (_provider == null) { return CollectRegisteredImplementationInstances(type); } var services = _provider!.GetServices(type).ToList(); if (_logger.IsDebugEnabled()) { _logger.Debug($"Retrieved {services.Count} instances of {type.Name}"); } return services.Where(o => o != null).Cast().ToList(); } finally { _lock.ExitReadLock(); } } /// /// 在容器未冻结时,从服务描述符中收集当前可直接观察到的实例绑定。 /// /// 调用方请求的服务类型。 /// 按当前未冻结语义可见的实例列表。 /// /// 该方法只读取 ,因为未冻结路径不会主动执行工厂方法, /// 也不会提前构造 。 /// 若同一实例同时经由多个可赋值的 暴露, /// 这里会把它视为兼容别名并只保留一个规范服务类型对应的结果; /// 但同一 的重复显式注册仍会完整保留,以维持注册顺序和多次注册语义。 /// private List CollectRegisteredImplementationInstances(Type requestedServiceType) { ArgumentNullException.ThrowIfNull(requestedServiceType); var matchingDescriptors = GetServicesUnsafe .Where(descriptor => descriptor.ServiceType == requestedServiceType || requestedServiceType.IsAssignableFrom(descriptor.ServiceType)) .ToList(); if (matchingDescriptors.Count == 0) return []; var preferredServiceTypes = BuildPreferredVisibleServiceTypes(matchingDescriptors, requestedServiceType); var result = new List(); foreach (var descriptor in matchingDescriptors) { if (descriptor.ImplementationInstance is { } instance) { if (preferredServiceTypes.TryGetValue(instance, out var preferredServiceType) && preferredServiceType == descriptor.ServiceType) { result.Add(instance); } } else if (descriptor.ImplementationFactory != null) { // 在未冻结状态下无法调用工厂方法,跳过。 } else if (descriptor.ImplementationType != null) { // 在未冻结状态下无法创建实例,跳过。 } } return result; } /// /// 为每个可见实例选择一个规范服务类型,避免同一实例因兼容别名重复出现在未冻结查询结果中。 /// /// 已按请求类型过滤过的服务描述符集合。 /// 调用方请求的服务类型。 /// 实例到其规范服务类型的映射。 private static Dictionary BuildPreferredVisibleServiceTypes( IReadOnlyList matchingDescriptors, Type requestedServiceType) { var preferredServiceTypes = new Dictionary(ReferenceEqualityComparer.Instance); foreach (var instanceGroup in matchingDescriptors .Where(static descriptor => descriptor.ImplementationInstance is not null) .GroupBy(static descriptor => descriptor.ImplementationInstance!, ReferenceEqualityComparer.Instance)) { preferredServiceTypes.Add( instanceGroup.Key, SelectPreferredVisibleServiceType(instanceGroup, requestedServiceType)); } return preferredServiceTypes; } /// /// 在“同一实例被多个服务类型暴露”的场景下,选择未冻结查询结果应保留的规范服务类型。 /// /// 引用同一实例的服务描述符。 /// 调用方请求的服务类型。 /// 应在结果中保留的服务类型。 private static Type SelectPreferredVisibleServiceType( IEnumerable descriptorsForInstance, Type requestedServiceType) { var serviceTypeGroups = descriptorsForInstance .GroupBy(static descriptor => descriptor.ServiceType) .Select((group, index) => new VisibleServiceTypeGroup(group.Key, group.Count(), index)) .ToList(); // 若调用方请求的正是其中一个服务类型,优先保留它,使未冻结行为尽量贴近冻结后的精确服务解析口径。 var requestedGroup = serviceTypeGroups.FirstOrDefault(group => group.ServiceType == requestedServiceType); if (requestedGroup is not null) return requestedGroup.ServiceType; // 否则优先保留“同一服务类型下注册次数最多”的那组,避免显式多次注册被较宽泛的别名折叠掉。 return serviceTypeGroups .OrderByDescending(static group => group.Count) .ThenBy(static group => group.FirstIndex) .First() .ServiceType; } /// /// 获取并排序指定泛型类型的所有服务实例 /// 主要用于系统调度场景 /// /// 服务类型 /// 比较委托,用于定义排序规则 /// 排序后的只读服务实例列表 public IReadOnlyList GetAllSorted(Comparison comparison) where T : class { ThrowIfDisposed(); var list = GetAll().ToList(); list.Sort(comparison); return list; } /// /// 获取指定类型的所有实例,并按优先级排序 /// 实现 IPrioritized 接口的服务将按优先级排序(数值越小优先级越高) /// 未实现 IPrioritized 的服务将使用默认优先级 0 /// /// 期望获取的实例类型 /// 按优先级排序后的实例列表 public IReadOnlyList GetAllByPriority() where T : class { var services = GetAll(); return SortByPriority(services); } /// /// 获取指定类型的所有实例,并按优先级排序 /// 实现 IPrioritized 接口的服务将按优先级排序(数值越小优先级越高) /// 未实现 IPrioritized 的服务将使用默认优先级 0 /// /// 期望获取的实例类型 /// 按优先级排序后的实例列表 public IReadOnlyList GetAllByPriority(Type type) { var services = GetAll(type); return SortByPriority(services); } /// /// 按优先级排序服务列表 /// 实现 IPrioritized 接口的服务按 Priority 属性排序(升序) /// 未实现接口的服务使用默认优先级 0 /// 相同优先级保持原有注册顺序(稳定排序) /// private static IReadOnlyList SortByPriority(IReadOnlyList services) where T : class { if (services.Count <= 1) return services; // 使用 OrderBy 确保稳定排序(相同优先级保持原有顺序) return services .Select((service, index) => new { Service = service, Index = index }) .OrderBy(x => { var priority = x.Service is IPrioritized p ? p.Priority : 0; return (priority, x.Index); // 先按优先级,再按索引 }) .Select(x => x.Service) .ToList(); } #endregion #region Utility /// /// 检查容器中是否包含指定泛型类型的实例 /// 根据容器状态选择不同的检查策略 /// /// 要检查的类型 /// true表示包含该类型实例,false表示不包含 public bool Contains() where T : class { ThrowIfDisposed(); EnterReadLockOrThrowDisposed(); try { if (_provider == null) return GetServicesUnsafe.Any(s => s.ServiceType == typeof(T)); return _provider.GetService() != null; } finally { _lock.ExitReadLock(); } } /// /// 检查容器中是否存在可赋值给指定服务类型的注册项,而不要求先解析实例。 /// /// 要检查的服务类型。 /// 若存在显式注册或开放泛型映射可满足该服务类型,则返回 ;否则返回 public bool HasRegistration(Type type) { ArgumentNullException.ThrowIfNull(type); ThrowIfDisposed(); EnterReadLockOrThrowDisposed(); try { if (_frozenServiceTypeIndex is not null) { return _frozenServiceTypeIndex.Contains(type); } return HasRegistrationCore(type); } finally { _lock.ExitReadLock(); } } /// /// 判断容器中是否包含某个具体的实例对象 /// 通过已注册实例集合进行快速查找 /// /// 要检查的实例对象 /// true表示包含该实例,false表示不包含 public bool ContainsInstance(object instance) { ThrowIfDisposed(); EnterReadLockOrThrowDisposed(); try { return _registeredInstances.Contains(instance); } finally { _lock.ExitReadLock(); } } /// /// 在当前容器状态下检查指定服务类型是否存在可见注册。 /// /// 要检查的服务类型。 /// 存在可满足该类型的注册时返回 ;否则返回 /// /// 该检查只回答“是否可能解析到服务”,不会为了判断结果而激活实例。 /// 预冻结阶段只基于当前服务描述符推断;冻结后则同样只观察描述符, /// 避免把瞬态/多实例解析成本混入热路径中的存在性判断。 /// private bool HasRegistrationCore(Type requestedType) { foreach (var descriptor in GetServicesUnsafe) { if (CanSatisfyServiceType(descriptor.ServiceType, requestedType)) { return true; } } return false; } /// /// 判断某个服务描述符声明的服务类型是否能满足当前请求类型。 /// /// 注册时声明的服务类型。 /// 调用方请求的服务类型。 /// 若当前注册可用于解析 ,则返回 private static bool CanSatisfyServiceType(Type registeredServiceType, Type requestedType) { // 这里刻意与 Get/GetAll 的“按服务键解析”语义保持一致: // 只有注册时声明的服务类型本身命中,或开放泛型服务键能闭合到请求类型时,才视为存在可见注册。 if (registeredServiceType == requestedType) { return true; } if (requestedType.IsConstructedGenericType && registeredServiceType.IsGenericTypeDefinition) { return requestedType.GetGenericTypeDefinition() == registeredServiceType; } return false; } /// /// 清空容器中的所有实例和服务注册 /// 只有在容器未冻结状态下才能执行清空操作 /// public void Clear() { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { // 冻结的容器不允许清空操作 if (_frozen) { _logger.Warn("Cannot clear frozen container"); return; } // 未冻结的容器不会构建根 ServiceProvider,因此这里仅重置注册状态即可。 GetServicesUnsafe.Clear(); _registeredInstances.Clear(); _provider = null; _frozenServiceTypeIndex = null; _frozen = false; _logger.Info("Container cleared"); } finally { _lock.ExitWriteLock(); } } /// /// 冻结容器并构建ServiceProvider /// 冻结后容器变为只读状态,不能再注册新服务 /// public void Freeze() { ThrowIfDisposed(); EnterWriteLockOrThrowDisposed(); try { // 防止重复冻结 if (_frozen) { _logger.Warn("Container already frozen"); return; } _provider = GetServicesUnsafe.BuildServiceProvider(); _frozenServiceTypeIndex = FrozenServiceTypeIndex.Create(GetServicesUnsafe); _frozen = true; _logger.Info("IOC Container frozen - ServiceProvider built"); } finally { _lock.ExitWriteLock(); } } /// /// 保存冻结后按服务键可见的精确服务类型与开放泛型定义集合。 /// /// /// 该索引只回答“按当前服务键语义是否可见”,因此与 / /// 一样不会退化为更宽松的可赋值匹配。 /// private sealed class FrozenServiceTypeIndex(HashSet exactServiceTypes, HashSet openGenericServiceTypes) { private readonly HashSet _exactServiceTypes = exactServiceTypes; private readonly HashSet _openGenericServiceTypes = openGenericServiceTypes; /// /// 基于冻结时最终确定的服务描述符集合创建索引。 /// /// 冻结时的服务描述符序列。 /// 供存在性判断热路径复用的服务键索引。 public static FrozenServiceTypeIndex Create(IEnumerable descriptors) { ArgumentNullException.ThrowIfNull(descriptors); var exactServiceTypes = new HashSet(); var openGenericServiceTypes = new HashSet(); foreach (var descriptor in descriptors) { var serviceType = descriptor.ServiceType; exactServiceTypes.Add(serviceType); if (serviceType.IsGenericTypeDefinition) { openGenericServiceTypes.Add(serviceType); } } return new FrozenServiceTypeIndex(exactServiceTypes, openGenericServiceTypes); } /// /// 判断当前索引是否声明了目标服务键。 /// /// 要检查的服务类型。 /// 命中精确服务键或可闭合的开放泛型服务键时返回 public bool Contains(Type requestedType) { ArgumentNullException.ThrowIfNull(requestedType); return _exactServiceTypes.Contains(requestedType) || requestedType.IsConstructedGenericType && _openGenericServiceTypes.Contains(requestedType.GetGenericTypeDefinition()); } } /// /// 获取底层的服务集合 /// 提供对内部IServiceCollection的访问权限,用于高级配置和自定义操作 /// /// 底层的IServiceCollection实例 public IServiceCollection GetServicesUnsafe { get; } = serviceCollection ?? new ServiceCollection(); /// /// 创建一个新的服务作用域 /// 作用域内的 Scoped 服务将共享同一实例 /// /// 服务作用域实例 /// 当容器未冻结时抛出 public IServiceScope CreateScope() { ThrowIfDisposed(); EnterReadLockOrThrowDisposed(); try { // 在锁内检查,避免竞态条件 if (!_frozen || _provider == null) { const string errorMsg = "Cannot create scope before container is frozen"; _logger.Error(errorMsg); throw new InvalidOperationException(errorMsg); } var scope = _provider.CreateScope(); _logger.Debug("Service scope created"); return scope; } finally { _lock.ExitReadLock(); } } /// /// 释放容器持有的服务提供者、注册状态和同步原语。 /// /// /// 冻结后的根 会拥有 DI 创建的单例与作用域根缓存,因此 benchmark、 /// 测试宿主或短生命周期架构在结束时需要显式释放容器,避免这些对象与内部 /// 一起滞留。 /// 释放是幂等的;首次释放后所有后续访问都会抛出 。 /// public void Dispose() { if (_disposed) { return; } var lockTaken = false; try { try { _lock.EnterWriteLock(); lockTaken = true; } catch (ObjectDisposedException) when (_disposed) { return; } if (_disposed) { return; } _disposed = true; (_provider as IDisposable)?.Dispose(); _provider = null; _frozenServiceTypeIndex = null; GetServicesUnsafe.Clear(); _registeredInstances.Clear(); _frozen = false; _logger.Info("IOC Container disposed"); } finally { if (lockTaken) { _lock.ExitWriteLock(); DisposeLockWhenQuiescent(); } } } #endregion }