- A+
原文地址:https://www.cnblogs.com/jingjiangtao/p/14711003.html
准备工作
为了演示自定义过滤器,需要新建一个 ASP.NET Core Web API 项目,项目配置可以按照自己的习惯来,也可以参考下面的配置,总之能让项目跑起来就可以。
Startup类:
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.AddControllers(); } // This method gets called by the runtime. Use this method to configure the HTTP request pipeline. public void Configure(IApplicationBuilder app, IWebHostEnvironment env) { if (env.IsDevelopment()) { app.UseDeveloperExceptionPage(); } app.UseRouting(); app.UseAuthorization(); app.UseEndpoints(endpoints => { endpoints.MapControllers(); }); } }
launchSettings.json
{ "$schema": "http://json.schemastore.org/launchsettings.json", "profiles": { "CustomizeActionFilter": { "commandName": "Project", "dotnetRunMessages": "true", "launchBrowser": true, "launchUrl": "", "applicationUrl": "http://localhost:5000", "environmentVariables": { "ASPNETCORE_ENVIRONMENT": "Development" } } } }
在Controllers目录下新建SampleController控制器类。这个类只是用来演示,没有业务上的意义:
public class SampleController : ControllerBase { [HttpGet("NeedVersionFilter")] public IActionResult NeedVersionFilter() { return Ok("OK: Need Version"); } [HttpGet("NoNeedVersionFilter")] public IActionResult NoNeedVersionFilter() { return Ok("OK: No Need Version"); } }
自定义过滤器
接下来开始编写自定义Action Filter。在项目根目录下新建ActionFilters目录,在此目录下新建类VersionCheckAttribute,该类继承自Attribute并实现了IActionFilter接口。继承Attribute可以让自定义过滤器以特性的方式使用,也就是用方括号括起来的形式,也有人叫标签;实现IActionFilter接口可以在请求的不同阶段添加处理逻辑。
[Serializable, AttributeUsage(AttributeTargets.Method | AttributeTargets.Class)] public class VersionCheckAttribute : Attribute, IActionFilter { public void OnActionExecuted(ActionExecutedContext context) { } public void OnActionExecuting(ActionExecutingContext context) { } }
其中OnActionExecuting方法在请求进入控制器Action之前执行,拦截代码也在这个方法中实现。
为做演示,假设需要实现这样一种拦截器:每个http请求的header中都应该带有自定义的参数version,如果version的值正确,则请求正常进入控制器执行,如果不正确,则直接返回,不再进入控制器。代码如下:
[Serializable, AttributeUsage(AttributeTargets.Method | AttributeTargets.Class)] public class VersionCheckAttribute : Attribute, IActionFilter { public void OnActionExecuted(ActionExecutedContext context) { } public void OnActionExecuting(ActionExecutingContext context) { if (!context.HttpContext.Request.Headers.ContainsKey("version")) { context.Result = new BadRequestObjectResult("Error: incorrect version"); return; } string headVersionStr = context.HttpContext.Request.Headers["version"].FirstOrDefault(); if (headVersionStr != "1.0.0") { context.Result = new BadRequestObjectResult("Error: incorrect version"); return; } } }
首先判断header中有没有version参数,没有的话直接返回错误。如果有version参数,则判断值是否正确,如果不正确也直接返回错误,如果正确,则继续向下执行,请求会进入控制器的Action。给context.Result赋值可以将请求短路,不会再进入控制器的Action中执行,而是直接返回。
接下来在SampleController上应用VersionCheck过滤器,让它过滤控制器中的所有请求:
[Route("[controller]")] [ApiController] [VersionCheck] // 自定义过滤器 public class SampleController : ControllerBase { [HttpGet("NeedVersionFilter")] public IActionResult NeedVersionFilter() { return Ok("OK: Need Version"); } [HttpGet("NoNeedVersionFilter")] public IActionResult NoNeedVersionFilter() { return Ok("OK: No Need Version"); } }
运行项目,用postman请求看看结果:
没有添加自定义header,返回 Error: incorrect version
添加了自定义header,返回正常:
至此,一个简单的Action Filter已经完成了。
排除指定的Action
上面实现的控制器有两个Action,假设有这样一种需求:NeedVersionFilter接口需要过滤版本号,NoNeedVersionFilter接口不需要过滤版本号。这样的话,[VersionCheck]放到SampleController类上就不行了,可以删掉控制器上的[VersionCheck]特性,转而放到NeedVersionFilter方法上,这样就实现了这个需求。
[Route("[controller]")] [ApiController] public class SampleController : ControllerBase { [VersionCheck] // 将过滤器放到方法上 [HttpGet("NeedVersionFilter")] public IActionResult NeedVersionFilter() { return Ok("OK: Need Version"); } [HttpGet("NoNeedVersionFilter")] public IActionResult NoNeedVersionFilter() { return Ok("OK: No Need Version"); } }
但是,如果Controller中的Action非常多,而大部分Action都需要版本过滤器,只有少数几个不需要,用这种形式就要在每个方法上应用[VersionCheck]特性,有点麻烦,还可能漏加。这时候如果把[VersionCheck]应用到Controller上,同时可以排除几个不需要过滤器的方法,写起来会更简洁。这是可以做到的,通过给不需要过滤器的方法做标记,就可以在过滤器中跳过有标记的方法了。
在ActionFilters目录下新建类IgnoreVersionCheckAttribute,继承自Attribute类和IFilterMetadata接口。IFilterMetadata接口没有需要实现的方法,仅作为标记:
[Serializable, AttributeUsage(AttributeTargets.Method)] public class IgnoreVersionCheckAttribute : Attribute, IFilterMetadata { }
修改VersioncheckAttribute类的代码,让过滤器跳过标记为IgnoreVersionCheck的方法:
[Serializable, AttributeUsage(AttributeTargets.Method | AttributeTargets.Class)] public class VersionCheckAttribute : Attribute, IActionFilter { public void OnActionExecuted(ActionExecutedContext context) { } public void OnActionExecuting(ActionExecutingContext context) { if (HasIgnoreVersionCheck(context)) { return; } if (!context.HttpContext.Request.Headers.ContainsKey("version")) { context.Result = new BadRequestObjectResult("Error: incorrect version"); return; } string headVersionStr = context.HttpContext.Request.Headers["version"].FirstOrDefault(); if (headVersionStr != "1.0.0") { context.Result = new BadRequestObjectResult("Error: incorrect version"); return; } } private bool HasIgnoreVersionCheck(ActionExecutingContext context) { IList<IFilterMetadata> filters = context.Filters; foreach (IFilterMetadata filter in filters) { if (filter is IgnoreVersionCheckAttribute) { return true; } } return false; } }
可以看到,在刚进入Action时就判断是否有IgnoreVersionCheck,如果有,则直接退出过滤器,继续执行Controller中的代码,如果没有则继续执行过滤器。HasIgnoreVersionCheck方法从ActionExecutingContext中拿到当前Action上的所有filter,遍历查找有没有IgnoreVersionCheckAttribute,有则返回true,没有则返回false。
修改SampleController的代码,把[VersionCheck]放到控制器上,在NoNeedVersionFilter方法上添加[IgnoreVersionCheck]
[Route("[controller]")] [ApiController] [VersionCheck] public class SampleController : ControllerBase { [HttpGet("NeedVersionFilter")] public IActionResult NeedVersionFilter() { return Ok("OK: Need Version"); } [IgnoreVersionCheck] [HttpGet("NoNeedVersionFilter")] public IActionResult NoNeedVersionFilter() { return Ok("OK: No Need Version"); } }
测试一下是否生效。
NeedVersionFilter接口不添加version头:
NeedVersionFilter接口添加version头:
NoNeedVersionFilter不添加version头:
可以看到确实生效了。至此,带排除项的过滤器就完成了。
在过滤器中获取服务
上面的过滤器代码为了方便起见,判断版本号是否正确时直接用了 "1.0.0" 这种硬编码的字符串,实际项目中这个字符串可能是会变化的,最好写在配置文件中。在appsetting.Development.json中添加字段 "VersionFilter" 并赋值:
{ "Logging": { "LogLevel": { "Default": "Information", "Microsoft": "Warning", "Microsoft.Hosting.Lifetime": "Information" } }, "VersionFilter": "1.0.0" }
修改VersionCheckAttribute的代码,通过ActionExecutedContext中的属性获取IConfiguration服务,再从IConfiguration实例中获取字符串:
[Serializable, AttributeUsage(AttributeTargets.Method | AttributeTargets.Class)] public class VersionCheckAttribute : Attribute, IActionFilter { public void OnActionExecuted(ActionExecutedContext context) { } public void OnActionExecuting(ActionExecutingContext context) { if (HasIgnoreVersionCheck(context)) { return; } if (!context.HttpContext.Request.Headers.ContainsKey("version")) { context.Result = new BadRequestObjectResult("Error: incorrect version"); return; } string headVersionStr = context.HttpContext.Request.Headers["version"].FirstOrDefault();
// 获取配置服务 var configuration = context.HttpContext.RequestServices.GetRequiredService<IConfiguration>(); string confVersionStr = configuration.GetValue<string>("VersionFilter"); if (headVersionStr != confVersionStr) { context.Result = new BadRequestObjectResult("Error: incorrect version"); return; } } private bool HasIgnoreVersionCheck(ActionExecutingContext context) { IList<IFilterMetadata> filters = context.Filters; foreach (IFilterMetadata filter in filters) { if (filter is IgnoreVersionCheckAttribute) { return true; } } return false; } }
这样,一个比较灵活的自定义ActionFilter就完成了。