ASP.NET Core 中间件(Middleware)的使用及其源码解析（三）- 对中间件管道 - ToB企服应用市场:ToB评测及商务社交产业平台

using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using NETCoreMiddleware.Middlewares;
namespace NETCoreMiddleware
{
public class Startup
{
public Startup(IConfiguration configuration)
{
Configuration = configuration;
}
public IConfiguration Configuration { get; }
//服务注册（往容器中添加服务）
// This method gets called by the runtime. Use this method to add services to the container.
public void ConfigureServices(IServiceCollection services)
{
services.AddControllersWithViews();
}
/// <summary>
/// 配置Http请求处理管道
/// Http请求管道模型---就是Http请求被处理的步骤
/// 所谓管道，就是拿着HttpContext，经过多个步骤的加工，生成Response，这就是管道
/// </summary>
/// <param name="app"></param>
/// <param name="env"></param>
// This method gets called by the runtime. Use this method to configure the HTTP request pipeline.
public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
{
#region 环境参数
if (env.IsDevelopment())
{
app.UseDeveloperExceptionPage();
}
else
{
app.UseExceptionHandler("/Home/Error");
}
#endregion 环境参数
//静态文件中间件
app.UseStaticFiles();
#region 创建管道分支
//Map管道分支
app.Map("/map1", HandleMapTest1);
app.Map("/map2", HandleMapTest2);
//MapWhen管道分支
app.MapWhen(context => context.Request.Query.ContainsKey("mapwhen"), HandleBranch);
//UseWhen管道分支
//UseWhen 与 MapWhen 不同的是，如果这个分支不发生短路或包含终端中间件，则会重新加入主管道
app.UseWhen(context => context.Request.Query.ContainsKey("usewhen"), HandleBranchAndRejoin);
#endregion 创建管道分支
#region Use中间件
//中间件1
app.Use(next =>
{
Console.WriteLine("middleware 1");
return async context =>
{
await Task.Run(() =>
{
Console.WriteLine("");
Console.WriteLine("===================================Middleware===================================");
Console.WriteLine($"This is middleware 1 Start");
});
await next.Invoke(context);
await Task.Run(() =>
{
Console.WriteLine($"This is middleware 1 End");
Console.WriteLine("===================================Middleware===================================");
});
};
});
//中间件2
app.Use(next =>
{
Console.WriteLine("middleware 2");
return async context =>
{
await Task.Run(() => Console.WriteLine($"This is middleware 2 Start"));
await next.Invoke(context); //可通过不调用 next 参数使请求管道短路
await Task.Run(() => Console.WriteLine($"This is middleware 2 End"));
};
});
//中间件3
app.Use(next =>
{
Console.WriteLine("middleware 3");
return async context =>
{
await Task.Run(() => Console.WriteLine($"This is middleware 3 Start"));
await next.Invoke(context);
await Task.Run(() => Console.WriteLine($"This is middleware 3 End"));
};
});
//中间件4
//Use方法的另外一个重载
app.Use(async (context, next) =>
{
await Task.Run(() => Console.WriteLine($"This is middleware 4 Start"));
await next();
await Task.Run(() => Console.WriteLine($"This is middleware 4 End"));
});
#endregion Use中间件
#region UseMiddleware中间件
app.UseMiddleware<CustomMiddleware>();
#endregion UseMiddleware中间件
#region 终端中间件
//app.Use(_ => handler);
app.Run(async context =>
{
await Task.Run(() => Console.WriteLine($"This is Run"));
});
#endregion 终端中间件
#region 最终把请求交给MVC
app.UseRouting();
app.UseAuthorization();
app.UseEndpoints(endpoints =>
{
endpoints.MapControllerRoute(
name: "areas",
pattern: "{area:exists}/{controller=Home}/{action=Index}/{id?}");
endpoints.MapControllerRoute(
name: "default",
pattern: "{controller=Home}/{action=Index}/{id?}");
});
#endregion 最终把请求交给MVC
}
#region 创建管道分支
/// <summary>
/// Map管道分支1
/// </summary>
/// <param name="app"></param>
static void HandleMapTest1(IApplicationBuilder app)
{
//中间件
app.Use(next =>
{
Console.WriteLine("HandleMapTest1");
return async context =>
{
await Task.Run(() =>
{
Console.WriteLine("");
Console.WriteLine($"This is HandleMapTest1 Start");
});
await next.Invoke(context); //可通过不调用 next 参数使请求管道短路
await Task.Run(() => Console.WriteLine($"This is HandleMapTest1 End"));
};
});
}
/// <summary>
/// Map管道分支2
/// </summary>
/// <param name="app"></param>
static void HandleMapTest2(IApplicationBuilder app)
{
//中间件
app.Use(next =>
{
Console.WriteLine("HandleMapTest2");
return async context =>
{
await Task.Run(() =>
{
Console.WriteLine("");
Console.WriteLine($"This is HandleMapTest2 Start");
});
await next.Invoke(context); //可通过不调用 next 参数使请求管道短路
await Task.Run(() => Console.WriteLine($"This is HandleMapTest2 End"));
};
});
}
/// <summary>
/// MapWhen管道分支
/// </summary>
/// <param name="app"></param>
static void HandleBranch(IApplicationBuilder app)
{
//中间件
app.Use(next =>
{
Console.WriteLine("HandleBranch");
return async context =>
{
await Task.Run(() =>
{
Console.WriteLine("");
Console.WriteLine($"This is HandleBranch Start");
});
await next.Invoke(context); //可通过不调用 next 参数使请求管道短路
await Task.Run(() => Console.WriteLine($"This is HandleBranch End"));
};
});
}
/// <summary>
/// UseWhen管道分支
/// </summary>
/// <param name="app"></param>
static void HandleBranchAndRejoin(IApplicationBuilder app)
{
//中间件
app.Use(next =>
{
Console.WriteLine("HandleBranchAndRejoin");
return async context =>
{
await Task.Run(() =>
{
Console.WriteLine("");
Console.WriteLine($"This is HandleBranchAndRejoin Start");
});
await next.Invoke(context); //可通过不调用 next 参数使请求管道短路
await Task.Run(() => Console.WriteLine($"This is HandleBranchAndRejoin End"));
};
});
}
#endregion 创建管道分支
}
}

复制代码

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Builder.Extensions;
namespace Microsoft.AspNetCore.Builder
{
/// <summary>
/// Extension methods for the <see cref="MapMiddleware"/>.
/// </summary>
public static class MapExtensions
{
/// <summary>
/// Branches the request pipeline based on matches of the given request path. If the request path starts with
/// the given path, the branch is executed.
/// </summary>
/// <param name="app">The <see cref="IApplicationBuilder"/> instance.</param>
/// <param name="pathMatch">The request path to match.</param>
/// <param name="configuration">The branch to take for positive path matches.</param>
/// <returns>The <see cref="IApplicationBuilder"/> instance.</returns>
public static IApplicationBuilder Map(this IApplicationBuilder app, PathString pathMatch, Action<IApplicationBuilder> configuration)
{
if (app == null)
{
throw new ArgumentNullException(nameof(app));
}
if (configuration == null)
{
throw new ArgumentNullException(nameof(configuration));
}
if (pathMatch.HasValue && pathMatch.Value.EndsWith("/", StringComparison.Ordinal))
{
throw new ArgumentException("The path must not end with a '/'", nameof(pathMatch));
}
// create branch
var branchBuilder = app.New();
configuration(branchBuilder);
var branch = branchBuilder.Build();
var options = new MapOptions
{
Branch = branch,
PathMatch = pathMatch,
};
return app.Use(next => new MapMiddleware(next, options).Invoke);
}
}
}

复制代码

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Http.Features;
using Microsoft.Extensions.Internal;
namespace Microsoft.AspNetCore.Builder
{
public class ApplicationBuilder : IApplicationBuilder
{
private const string ServerFeaturesKey = "server.Features";
private const string ApplicationServicesKey = "application.Services";
private readonly IList<Func<RequestDelegate, RequestDelegate>> _components = new List<Func<RequestDelegate, RequestDelegate>>();
public ApplicationBuilder(IServiceProvider serviceProvider)
{
Properties = new Dictionary<string, object>(StringComparer.Ordinal);
ApplicationServices = serviceProvider;
}
public ApplicationBuilder(IServiceProvider serviceProvider, object server)
: this(serviceProvider)
{
SetProperty(ServerFeaturesKey, server);
}
private ApplicationBuilder(ApplicationBuilder builder)
{
Properties = new CopyOnWriteDictionary<string, object>(builder.Properties, StringComparer.Ordinal);
}
public IServiceProvider ApplicationServices
{
get
{
return GetProperty<IServiceProvider>(ApplicationServicesKey);
}
set
{
SetProperty<IServiceProvider>(ApplicationServicesKey, value);
}
}
public IFeatureCollection ServerFeatures
{
get
{
return GetProperty<IFeatureCollection>(ServerFeaturesKey);
}
}
public IDictionary<string, object> Properties { get; }
private T GetProperty<T>(string key)
{
object value;
return Properties.TryGetValue(key, out value) ? (T)value : default(T);
}
private void SetProperty<T>(string key, T value)
{
Properties[key] = value;
}
public IApplicationBuilder Use(Func<RequestDelegate, RequestDelegate> middleware)
{
_components.Add(middleware);
return this;
}
public IApplicationBuilder New()
{
return new ApplicationBuilder(this);
}
public RequestDelegate Build()
{
RequestDelegate app = context =>
{
// If we reach the end of the pipeline, but we have an endpoint, then something unexpected has happened.
// This could happen if user code sets an endpoint, but they forgot to add the UseEndpoint middleware.
var endpoint = context.GetEndpoint();
var endpointRequestDelegate = endpoint?.RequestDelegate;
if (endpointRequestDelegate != null)
{
var message =
$"The request reached the end of the pipeline without executing the endpoint: '{endpoint.DisplayName}'. " +
$"Please register the EndpointMiddleware using '{nameof(IApplicationBuilder)}.UseEndpoints(...)' if using " +
$"routing.";
throw new InvalidOperationException(message);
}
context.Response.StatusCode = 404;
return Task.CompletedTask;
};
foreach (var component in _components.Reverse())
{
app = component(app);
}
return app;
}
}
}

复制代码

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Http;
namespace Microsoft.AspNetCore.Builder.Extensions
{
/// <summary>
/// Represents a middleware that maps a request path to a sub-request pipeline.
/// </summary>
public class MapMiddleware
{
private readonly RequestDelegate _next;
private readonly MapOptions _options;
/// <summary>
/// Creates a new instance of <see cref="MapMiddleware"/>.
/// </summary>
/// <param name="next">The delegate representing the next middleware in the request pipeline.</param>
/// <param name="options">The middleware options.</param>
public MapMiddleware(RequestDelegate next, MapOptions options)
{
if (next == null)
{
throw new ArgumentNullException(nameof(next));
}
if (options == null)
{
throw new ArgumentNullException(nameof(options));
}
_next = next;
_options = options;
}
/// <summary>
/// Executes the middleware.
/// </summary>
/// <param name="context">The <see cref="HttpContext"/> for the current request.</param>
/// <returns>A task that represents the execution of this middleware.</returns>
public async Task Invoke(HttpContext context)
{
if (context == null)
{
throw new ArgumentNullException(nameof(context));
}
PathString matchedPath;
PathString remainingPath;
if (context.Request.Path.StartsWithSegments(_options.PathMatch, out matchedPath, out remainingPath))
{
// Update the path
var path = context.Request.Path;
var pathBase = context.Request.PathBase;
context.Request.PathBase = pathBase.Add(matchedPath);
context.Request.Path = remainingPath;
try
{
await _options.Branch(context);
}
finally
{
context.Request.PathBase = pathBase;
context.Request.Path = path;
}
}
else
{
await _next(context);
}
}
}
}

复制代码

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Builder.Extensions;
namespace Microsoft.AspNetCore.Builder
{
using Predicate = Func<HttpContext, bool>;
/// <summary>
/// Extension methods for the <see cref="MapWhenMiddleware"/>.
/// </summary>
public static class MapWhenExtensions
{
/// <summary>
/// Branches the request pipeline based on the result of the given predicate.
/// </summary>
/// <param name="app"></param>
/// <param name="predicate">Invoked with the request environment to determine if the branch should be taken</param>
/// <param name="configuration">Configures a branch to take</param>
/// <returns></returns>
public static IApplicationBuilder MapWhen(this IApplicationBuilder app, Predicate predicate, Action<IApplicationBuilder> configuration)
{
if (app == null)
{
throw new ArgumentNullException(nameof(app));
}
if (predicate == null)
{
throw new ArgumentNullException(nameof(predicate));
}
if (configuration == null)
{
throw new ArgumentNullException(nameof(configuration));
}
// create branch
var branchBuilder = app.New();
configuration(branchBuilder);
var branch = branchBuilder.Build();
// put middleware in pipeline
var options = new MapWhenOptions
{
Predicate = predicate,
Branch = branch,
};
return app.Use(next => new MapWhenMiddleware(next, options).Invoke);
}
}
}

复制代码

// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System;
using System.Threading.Tasks;
using Microsoft.AspNetCore.Http;
namespace Microsoft.AspNetCore.Builder.Extensions
{
/// <summary>
/// Represents a middleware that runs a sub-request pipeline when a given predicate is matched.
/// </summary>
public class MapWhenMiddleware
{
private readonly RequestDelegate _next;
private readonly MapWhenOptions _options;
/// <summary>
/// Creates a new instance of <see cref="MapWhenMiddleware"/>.
/// </summary>
/// <param name="next">The delegate representing the next middleware in the request pipeline.</param>
/// <param name="options">The middleware options.</param>
public MapWhenMiddleware(RequestDelegate next, MapWhenOptions options)
{
if (next == null)
{
throw new ArgumentNullException(nameof(next));
}
if (options == null)
{
throw new ArgumentNullException(nameof(options));
}
_next = next;
_options = options;
}
/// <summary>
/// Executes the middleware.
/// </summary>
/// <param name="context">The <see cref="HttpContext"/> for the current request.</param>
/// <returns>A task that represents the execution of this middleware.</returns>
public async Task Invoke(HttpContext context)
{
if (context == null)
{
throw new ArgumentNullException(nameof(context));
}
if (_options.Predicate(context))
{
await _options.Branch(context);
}
else
{
await _next(context);
}
}
}
}

复制代码

// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
using System;
using Microsoft.AspNetCore.Http;
namespace Microsoft.AspNetCore.Builder
{
using Predicate = Func<HttpContext, bool>;
/// <summary>
/// Extension methods for <see cref="IApplicationBuilder"/>.
/// </summary>
public static class UseWhenExtensions
{
/// <summary>
/// Conditionally creates a branch in the request pipeline that is rejoined to the main pipeline.
/// </summary>
/// <param name="app"></param>
/// <param name="predicate">Invoked with the request environment to determine if the branch should be taken</param>
/// <param name="configuration">Configures a branch to take</param>
/// <returns></returns>
public static IApplicationBuilder UseWhen(this IApplicationBuilder app, Predicate predicate, Action<IApplicationBuilder> configuration)
{
if (app == null)
{
throw new ArgumentNullException(nameof(app));
}
if (predicate == null)
{
throw new ArgumentNullException(nameof(predicate));
}
if (configuration == null)
{
throw new ArgumentNullException(nameof(configuration));
}
// Create and configure the branch builder right away; otherwise,
// we would end up running our branch after all the components
// that were subsequently added to the main builder.
var branchBuilder = app.New();
configuration(branchBuilder);
return app.Use(main =>
{
// This is called only when the main application builder
// is built, not per request.
branchBuilder.Run(main);
var branch = branchBuilder.Build();
return context =>
{
if (predicate(context))
{
return branch(context);
}
else
{
return main(context);
}
};
});
}
}
}

复制代码

链接：https://pan.baidu.com/s/18I66dBmKZUpfPCNn85HI2g
提取码：2xcj

复制代码