Net6 DI源碼分析Part3 CallSiteRuntimeResolver,CallSiteVisitor

• 2022 年 2 月 11 日
• 筆記
• DI, netcore

CallSiteRuntimeResolver

CallSiteRuntimeResolver是實現了CallSiteVisitor之一。
提供的方法主要分三個部分

1. 自有成員方法

   1. Resolve提供服務
   2. VisitCache
   3. 私有構造函數

2. 重寫父類方法

   1. VisitDisposeCache(ServiceCallSite transientCallSite, RuntimeResolverContext context)
      調用父類VisitCallSiteMain ，並把創建出來的服務添加到context.Scope.CaptureDisposable

      return context.Scope.CaptureDisposable(VisitCallSiteMain(transientCallSite, context));
      

   2. object VisitRootCache(ServiceCallSite callSite, RuntimeResolverContext context)

      1. 先找對應callSite.Value是否有值如果有直接返回
      2. 如果沒有直接調用 VisitCallSiteMain創建服務並把對象rootServiceProviderEngine內。
      3. 創建的服務放入callSite.Value。

          if (callSite.Value is object value) 
          {
              return value;
          }
      
      var lockType = RuntimeResolverLock.Root;
      ServiceProviderEngineScope serviceProviderEngine = context.Scope.RootProvider.Root;
      if (callSite.Value is object resolved)
      {
          return resolved;
      }
      resolved = VisitCallSiteMain(callSite, new RuntimeResolverContext
      {
          Scope = serviceProviderEngine,
          AcquiredLocks = context.AcquiredLocks | lockType
      });
      serviceProviderEngine.CaptureDisposable(resolved);
      callSite.Value = resolved;
      return resolved;
      

   3. VisitCache(ServiceCallSite callSite, RuntimeResolverContext context, ServiceProviderEngineScope serviceProviderEngine, RuntimeResolverLock lockType)

      1. 先去serviceProviderEngine.resolvedServices拿,有的話直接返回如果沒拿到執行步驟2去創建。serviceProviderEngine.resolvedServices.TryGetValue(callSite.Cache.Key, out object resolved)
      2. VisitCallSiteMain創建對象
      3. 把創建後的對象加入對應Scope的 disposable列表
      4. 把創建後的對象加入對應Scope的 resolvedServices列表

          if (resolvedServices.TryGetValue(callSite.Cache.Key, out object resolved))
          {
              return resolved;
          }
          resolved = VisitCallSiteMain(callSite, new RuntimeResolverContext
          {
              Scope = serviceProviderEngine,
              AcquiredLocks = context.AcquiredLocks | lockType
          });
          serviceProviderEngine.CaptureDisposable(resolved);
          resolvedServices.Add(callSite.Cache.Key, resolved);
          return resolved;
      

3. 實現父類抽象方法

   1. VisitConstructor
      1. 給構造方法提供參數（如果有參數遞歸調用VisitCallSite創建參數對象）
      2. 直接調用ServiceCallSite的ConstructorInfo反射創建對象。

         object[] parameterValues;
          if (constructorCallSite.ParameterCallSites.Length == 0)
          {
              parameterValues = Array.Empty<object>();
          }
          else
          {
              parameterValues = new object[constructorCallSite.ParameterCallSites.Length];
              for (int index = 0; index < parameterValues.Length; index++)
              {
                  parameterValues[index] = VisitCallSite(constructorCallSite.ParameterCallSites[index], context);
              }
          }
      
          return constructorCallSite.ConstructorInfo.Invoke(BindingFlags.DoNotWrapExceptions, binder: null, parameters: parameterValues, culture: null);
      

   2. override object VisitConstant(ConstantCallSite constantCallSite, RuntimeResolverContext context)
      直接返回CallSite對象的DefaultValue
      return constantCallSite.DefaultValue;
   3. VisitServiceProvider
      return context.Scope;
   4. override object VisitIEnumerable(IEnumerableCallSite enumerableCallSite, RuntimeResolverContext context)
      根據ServiceCallSites直接循環調用VisitCallSite創建實例，然後根據實例創建集合返回

        var array = Array.CreateInstance(
        enumerableCallSite.ItemType,
        enumerableCallSite.ServiceCallSites.Length);
        for (int index = 0; index < enumerableCallSite.ServiceCallSites.Length; index++)
        {
            object value = VisitCallSite(enumerableCallSite.ServiceCallSites[index], context);
            array.SetValue(value, index);
        }
        return array;
      

   5. VisitFactory
      return factoryCallSite.Factory(context.Scope);

體力活都給CallSiteFactory了

CallSiteVisitor

1. 方法

   1. VisitCallSite(ServiceCallSite callSite, TArgument argument)

      1. 保證Stack安全
      2. 根據callSite的Cache.Location去調用不同策略的Cache方法。

      switch (callSite.Cache.Location)
           {
               case CallSiteResultCacheLocation.Root:
                   return VisitRootCache(callSite, argument);
               case CallSiteResultCacheLocation.Scope:
                   return VisitScopeCache(callSite, argument);
               case CallSiteResultCacheLocation.Dispose:
                   return VisitDisposeCache(callSite, argument);
               case CallSiteResultCacheLocation.None:
                   return VisitNoCache(callSite, argument);
               default:
                   throw new ArgumentOutOfRangeException();
       }
      

   2. VisitCallSiteMain(ServiceCallSite callSite, TArgument argument)

      1. 根據不同CallSiteKind去調用不同策略的創建服務方法。

          switch (callSite.Kind)
           {
               case CallSiteKind.Factory:
                   return VisitFactory((FactoryCallSite)callSite, argument);
               case  CallSiteKind.IEnumerable:
                   return VisitIEnumerable((IEnumerableCallSite)callSite, argument);
               case CallSiteKind.Constructor:
                   return VisitConstructor((ConstructorCallSite)callSite, argument);
               case CallSiteKind.Constant:
                   return VisitConstant((ConstantCallSite)callSite, argument);
               case CallSiteKind.ServiceProvider:
                   return VisitServiceProvider((ServiceProviderCallSite)callSite, argument);
               default:
                   throw new NotSupportedException(SR.Format(SR.CallSiteTypeNotSupported, callSite.GetType()));
           }
         

2. 虛方法
   1. TResult VisitNoCache(ServiceCallSite callSite, TArgument argument) => VisitCallSiteMain(callSite, argument);
   2. TResult VisitDisposeCache(ServiceCallSite callSite, TArgument argument) -> isitCallSiteMain(callSite, argument);
   3. TResult VisitRootCache(ServiceCallSite callSite, TArgument argument) -> VisitCallSiteMain(callSite, argument);
   4. TResult VisitScopeCache(ServiceCallSite callSite, TArgument argument) => VisitCallSiteMain(callSite, argument);

   根據生命周期策略由子類實現對應存儲機制。

3. 抽象方法
   1. TResult VisitConstructor(ConstructorCallSite constructorCallSite, TArgument argument);
   2. TResult VisitConstant(ConstantCallSite constantCallSite, TArgument argument);
   3. TResult VisitServiceProvider(ServiceProviderCallSite serviceProviderCallSite, TArgument argument);
   4. TResult VisitIEnumerable(IEnumerableCallSite enumerableCallSite, TArgument argument);
   5. TResult VisitFactory(FactoryCallSite factoryCallSite, TArgument argument);

   根據服務註冊時的創建機制策略由子類實現對應的創建對象機制。

CallSiteVisitor負責這活怎麼干！。CallSiteRuntimeResolver負責幹活

1. 調用 VisitCallSite在內部區分出scope並調用具體實現方法快取方法 (CallSiteResultCacheLocation),
   -> VisitCache方法
   -> 但是在快取前得先得到被快取的對象，這時通過調用（VisitCallSiteMain）創建對象
   -> VisitCallSiteMain根據提供的callsite類型去調用實現的抽象方法創建對應對象
   -> 根據cache對應的方法快取步驟1.1的結果。
2. return cahce的結果。

整體的獲取對象流程大約如下

CallSiteRuntimeResolver.Resolve -> CallSiteVisitor->VisitCallSite callSite.Cache.Location VisitRootCache/VisitScopeCache/ VisitDisposeCache/ VisitNoCache -> CallSiteRuntimeResolver.VisitCache/ VisitRootCache -> VisitRootCache -> CallSiteVisitor.VisitCallSiteMain 根據callSite.Kind CallSiteRuntimeResolver.VisitFactory（abstract TResult VisitFactory）遞歸調用 VisitCallSite