Owin Katana 的底层源码分析
最近看了一下开源项目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; }
好的，到这里了，谢谢大家阅读，如果有任何不理解的欢迎交流：）
