Owin Katana 的底層原始碼分析

NeilHu發表於2020-05-11

最近看了一下開源專案asp.net katana,感覺公開的介面非常的簡潔優雅,channel 9 說是受到node.js的啟發設計的,Katana是一個比較老的專案,現在已經整合到asp.net core中。

github克隆下來的專案,這個部落格專門是從程式碼角度去理解katana專案,所以本篇隨筆針對已經對OWIN有所瞭解的人,如果只是入門的話可以跑一下MSDN的原始碼再來閱讀本篇文章。

程式碼結構如上,簡單分析一下各個資料夾的含義,這對於理解katana專案的整體結構有一個大的輪廓。

  .build資料夾顧名思義就是編譯的資料夾,在沒使用vs的時候你可以單擊build.cmd 去編譯這個專案,十分的方便。

  .Nuget就是包管理工具的配置檔案,這個我們可以忽略。同理.Prerelease。

  Development是本次的研究重點,當你開啟這個資料夾的時候你會發現一個類庫Microsoft.Owin的類庫,這個是OWIN元件的經典實現。

  FunctionTests是單元測試的類庫

  Hosting 是server的抽象層,OWIN 將伺服器進行抽象化,Hosting 就是能夠管理Server的一層,像WebApp就能開啟一個httplister服務,詳細稍後再講。

  Middleware是一些中介軟體的實現,在katana已經將管道模型虛擬化成中介軟體

  Performance 和Sandbox 是微軟的一些測試工具

  Security 是微軟已經寫好的驗證中介軟體,其中包括JWT和Oauth的驗證方式

  Server 就是伺服器的實現

  Owin.Analysis是我本人建的web程式用來debug

上面已經介紹了各個資料夾所對應的功能,相必大部分人都是一臉矇蔽,但是不用擔心,下面就來看看具體的程式碼,當然是從最小的例子出發。點選 getting started with owin and katan 你就能跳到MSDN得到最小的例子。裡面的一系列操作就為了新增下面的一個類和幾個reference.現在我們看一下這個類。

 

 1 using Microsoft.Owin;
 2 
 3 [assembly: OwinStartup(typeof(Owin.Analysis.Startup))]
 4 namespace Owin.Analysis
 5 {
 6     public class Startup
 7     {
 8         public void Configuration(IAppBuilder app)
 9         {
10             app.Run(context =>
11             {
12                 context.Response.ContentType = "text/plain";
13                 return context.Response.WriteAsync("Hello World");
14             });
15         }
16     }
17 }

看起來這個程式碼十分的優雅,新增幾個reference和一個類就讓請求到達Hello World。我們先分析這個類,首先程式集特性OwinStartupAtribute將當前類儲存在後設資料中。然後寫了一個Configuration方法,獲取一個IAppBuilder 引數呼叫Run方法,Run方法傳遞一個委託進去,我們的處理邏輯就在這一個委託裡。

這裡面我們分析一下核心介面IAppBuilder的經典實現者AppBuilder,IAppBuilder的介面如下,

using System;
using System.Collections.Generic;

namespace Owin
{
    public interface IAppBuilder
    {
        IDictionary<string, object> Properties { get; }//請求的引數

        object Build(Type returnType);//中介軟體連結
        IAppBuilder New();//建立一個新的物件
        IAppBuilder Use(object middleware, params object[] args);//註冊中介軟體
    }
}

好的我們來分析一下AppBuilder中介軟體的註冊實現。在app.Run 打完break point你就可以進入app.use方法,首先在AppBuilderUseExtensions這個類裡對use的入口寫了一大堆擴充套件方法。app.Run就是其中的一個,當你用app.Run註冊中介軟體的時候是沒有下一個中介軟體的引用的。

        public static void Run(this IAppBuilder app, Func<IOwinContext, Task> handler)
        {
            if (app == null)
            {
                throw new ArgumentNullException("app");
            }
            if (handler == null)
            {
                throw new ArgumentNullException("handler");
            }

            app.Use<UseHandlerMiddleware>(handler);
        }

在經典的實現中,引數middleware會有兩種情況,一種是delegate,一種是type,如果是type型別,則他的構造方法接受next為引數,並且裡面有一個公開的Invoke方法。如果是委託,當前委託作為引數傳遞到next中。

 public IAppBuilder Use(object middleware, params object[] args)
        {
            _middleware.Add(ToMiddlewareFactory(middleware, args));
            return this;
        }

 

下面的程式碼是ToMiddlewareFactory的實現,在第9行和第27行分別判斷了中介軟體物件是委託型別還是type型別,由於本題例子是type物件,我們分析一下ToConstructorMiddlewareFactory方法。

 1 private static Tuple<Type, Delegate, object[]> ToMiddlewareFactory(object middlewareObject, object[] args)
 2         {
 3             if (middlewareObject == null)
 4             {
 5                 throw new ArgumentNullException("middlewareObject");
 6             }
 7 
 8             var middlewareDelegate = middlewareObject as Delegate;
 9             if (middlewareDelegate != null)
10             {
11                 return Tuple.Create(GetParameterType(middlewareDelegate), middlewareDelegate, args);
12             }
13 
14             Tuple<Type, Delegate, object[]> factory = ToInstanceMiddlewareFactory(middlewareObject, args);
15             if (factory != null)
16             {
17                 return factory;
18             }
19 
20             factory = ToGeneratorMiddlewareFactory(middlewareObject, args);
21             if (factory != null)
22             {
23                 return factory;
24             }
25 
26             if (middlewareObject is Type)
27             {
28                 return ToConstructorMiddlewareFactory(middlewareObject, args, ref middlewareDelegate);
29             }
30 
31             throw new NotSupportedException(string.Format(CultureInfo.CurrentCulture,
32                 Resources.Exception_MiddlewareNotSupported, middlewareObject.GetType().FullName));
33         }

 

 

在第5行獲取type型別的所有構造方法,第8行獲取構造方法的所有引數,在第14行有個trick,用zip方法判斷引數型別是否和構造方法型別是一致的。如果是一致的則繼續往下走,在第22行和第23行利用委託將構造方法建立成lambda表示式,然後生成元祖,第一個是next的型別,第二個是type的構造方法,第三個是type構造方法所需的引數。然後將元祖加入到AppBuild所維護的中介軟體物件。

 1  
 2         private static Tuple<Type, Delegate, object[]> ToConstructorMiddlewareFactory(object middlewareObject, object[] args, ref Delegate middlewareDelegate)
 3         {
 4             var middlewareType = middlewareObject as Type;
 5             ConstructorInfo[] constructors = middlewareType.GetConstructors();
 6             foreach (var constructor in constructors)
 7             {
 8                 ParameterInfo[] parameters = constructor.GetParameters();
 9                 Type[] parameterTypes = parameters.Select(p => p.ParameterType).ToArray();
10                 if (parameterTypes.Length != args.Length + 1)
11                 {
12                     continue;
13                 }
14                 if (!parameterTypes
15                     .Skip(1)
16                     .Zip(args, TestArgForParameter)
17                     .All(x => x))
18                 {
19                     continue;
20                 }
21 
22                 ParameterExpression[] parameterExpressions = parameters.Select(p => Expression.Parameter(p.ParameterType, p.Name)).ToArray();
23                 NewExpression callConstructor = Expression.New(constructor, parameterExpressions);
24                 middlewareDelegate = Expression.Lambda(callConstructor, parameterExpressions).Compile();
25                 return Tuple.Create(parameters[0].ParameterType, middlewareDelegate, args);
26             }
27 
28             throw new MissingMethodException(string.Format(CultureInfo.CurrentCulture,
29                 Resources.Exception_NoConstructorFound, middlewareType.FullName, args.Length + 1));
30         }

 

這個時候我們已經將中介軟體註冊到AppBuilder物件了。註冊完中介軟體的物件我們還需要做一件事就是將這些中介軟體chained together,這些實現就是Build 方法中,而Build方法BuildInternal方法,這個時候會產生一個entry point供呼叫。

現在我們重點看一下這個build方法。

    public object Build(Type returnType)
        {
            return BuildInternal(returnType);
        }

Build方法呼叫私有的BuildInternal的方法。

private object BuildInternal(Type signature)
        {
            object app;
            if (!_properties.TryGetValue(Constants.BuilderDefaultApp, out app))
            {
                app = NotFound;
            }

            foreach (var middleware in _middleware.Reverse())
          {
                Type neededSignature = middleware.Item1;
                Delegate middlewareDelegate = middleware.Item2;
                object[] middlewareArgs = middleware.Item3;

                app = Convert(neededSignature, app);
                object[] invokeParameters = new[] { app }.Concat(middlewareArgs).ToArray();
                app = middlewareDelegate.DynamicInvoke(invokeParameters);
                app = Convert(neededSignature, app);
            }

            return Convert(signature, app);
        }

我們可以看到它是怎樣將中介軟體chained together的,在我們之前註冊的時候實際上middleware元祖會儲存三個資訊,第一個type就是建構函式的第一個型別,第二個委託是useHandlerMiddleware的構造方法,第三個是構造方法的引數(除了第一個),Reverse的方法會將中介軟體逆序,這樣保證呼叫的順序就是你註冊的順序,後面的是chain的邏輯,app的變數實際上就是下一個中介軟體建構函式的next,當得到第一個中介軟體的時候,裡面的next會儲存第二個中介軟體的處理邏輯,同樣第二個next就是第三個...,這樣chained together得到的就是第一個中介軟體的邏輯,所以你們在用app.Use的方法就會有一個引數next,並且需要手動呼叫一下。

得到這些之後需要的就是要將中介軟體註冊到application管道事件呢。因為asp.net的是一個大的切面框架。

 public void Initialize(HttpApplication application)
        {
            for (IntegratedPipelineBlueprintStage stage = _blueprint.FirstStage; stage != null; stage = stage.NextStage)
            {
                var segment = new IntegratedPipelineContextStage(this, stage);
                switch (stage.Name)
                {
                    case Constants.StageAuthenticate:
                        application.AddOnAuthenticateRequestAsync(segment.BeginEvent, segment.EndEvent);
                        break;
                    case Constants.StagePostAuthenticate:
                        application.AddOnPostAuthenticateRequestAsync(segment.BeginEvent, segment.EndEvent);
                        break;
                    case Constants.StageAuthorize:
                        application.AddOnAuthorizeRequestAsync(segment.BeginEvent, segment.EndEvent);
                        break;
                    case Constants.StagePostAuthorize:
                        application.AddOnPostAuthorizeRequestAsync(segment.BeginEvent, segment.EndEvent);
                        break;
                    case Constants.StageResolveCache:
                        application.AddOnResolveRequestCacheAsync(segment.BeginEvent, segment.EndEvent);
                        break;
                    case Constants.StagePostResolveCache:
                        application.AddOnPostResolveRequestCacheAsync(segment.BeginEvent, segment.EndEvent);
                        break;
                    case Constants.StageMapHandler:
                        application.AddOnMapRequestHandlerAsync(segment.BeginEvent, segment.EndEvent);
                        break;
                    case Constants.StagePostMapHandler:
                        application.AddOnPostMapRequestHandlerAsync(segment.BeginEvent, segment.EndEvent);
                        break;
                    case Constants.StageAcquireState:
                        application.AddOnAcquireRequestStateAsync(segment.BeginEvent, segment.EndEvent);
                        break;
                    case Constants.StagePostAcquireState:
                        application.AddOnPostAcquireRequestStateAsync(segment.BeginEvent, segment.EndEvent);
                        break;
                    case Constants.StagePreHandlerExecute:
                        application.AddOnPreRequestHandlerExecuteAsync(segment.BeginEvent, segment.EndEvent);
                        break;
                    default:
                        throw new NotSupportedException(
                            string.Format(CultureInfo.InvariantCulture, Resources.Exception_UnsupportedPipelineStage, stage.Name));
                }
            }
            // application.PreSendRequestHeaders += PreSendRequestHeaders; // Null refs for async un-buffered requests with bodies.
            application.AddOnEndRequestAsync(BeginFinalWork, EndFinalWork);
        }

這裡面有一個概念就是IntegratedPipelineBlueprintStage,這個是一個連結串列結構,每個對應的就是管道事件,每個stage都有entry point,這樣方便我們在不同的管道事件中執行中介軟體,在BeginEvent裡我們得到stage 的entry point,然後非同步呼叫得到結果。entry point 就是我們上例build得到的結果。

  private async Task RunApp(AppFunc entryPoint, IDictionary<string, object> environment, TaskCompletionSource<object> tcs, StageAsyncResult result)
        {
            try
            {
                await entryPoint(environment);
                tcs.TrySetResult(null);
                result.TryComplete();
            }
            catch (Exception ex)
            {
                // Flow the exception back through the OWIN pipeline.
                tcs.TrySetException(ex);
                result.TryComplete();
            }
        }

然後我們分析一下怎麼在不同的管道中註冊事件。在MSDN的文件描述的。

app.UseStageMarker(PipelineStage.Authenticate)

這個api會建立一個IntegratedPipelineBlueprintStage,上文說這是一個連結串列結構,之間用next屬性連線。在不同的stage中會有entry point,然後在上面的例子中註冊到不同的管道中去呼叫。下圖是api的程式碼。

  public static IAppBuilder UseStageMarker(this IAppBuilder app, string stageName)
        {
            if (app == null)
            {
                throw new ArgumentNullException("app");
            }

            object obj;
            if (app.Properties.TryGetValue(IntegratedPipelineStageMarker, out obj))
            {
                var addMarker = (Action<IAppBuilder, string>)obj;
                addMarker(app, stageName);
            }
            return app;
        }

好的,到這裡了,謝謝大家閱讀,如果有任何不理解的歡迎交流:)

相關文章