ASP.NET Core - 实现自定义WebApi模型验证

  Framework时代  

  在Framework时代,咱们通常进行参数验证的时候,如下代码是很是常见的框架

        [HttpPost]
        public async Task<JsonResult> SaveNewCustomerAsnyc(AddCustomerInput input)
        {
            if (!ModelState.IsValid)
            {
                return Json(Result.FromCode(ResultCode.InvalidParams));
            }

           .....
        }

  或者高级一点是实现IActionFilter进行拦截,以下:  async

  public class ApiValidationFilter : IActionFilter
    {
        public bool AllowMultiple => false;

        public async Task<HttpResponseMessage> ExecuteActionFilterAsync(HttpActionContext actionContext, CancellationToken cancellationToken, Func<Task<HttpResponseMessage>> continuation)
        {
            var method = actionContext.ActionDescriptor.GetMethodInfoOrNull();
            if (method == null)
            {
                return await continuation();
            }           

            if (!actionContext.ModelState.IsValid)
            {
                var error = actionContext.ModelState.GetValidationSummary();
                var result = Result.FromError($"参数验证不经过:{error}", ResultCode.InvalidParams);
                return actionContext.Request.CreateResponse(result);
            }

            return await continuation();
        }
}
 public static class ModelStateExtensions
    {
        /// <summary>
        /// 获取验证消息提示并格式化提示
        /// </summary>
        public static string GetValidationSummary(this ModelStateDictionary modelState, string separator = "\r\n")
        {
            if (modelState.IsValid) return null;

            var error = new StringBuilder();

            foreach (var item in modelState)
            {
                var state = item.Value;
                var message = state.Errors.FirstOrDefault(p => !string.IsNullOrWhiteSpace(p.ErrorMessage))?.ErrorMessage;
                if (string.IsNullOrWhiteSpace(message))
                {
                    message = state.Errors.FirstOrDefault(o => o.Exception != null)?.Exception.Message;
                }
                if (string.IsNullOrWhiteSpace(message)) continue;

                if (error.Length > 0)
                {
                    error.Append(separator);
                }

                error.Append(message);
            }

            return error.ToString();
        }
    } 

  而后在启动项把这个拦截注册进来使用便可 ui

  .Net Core时代  

  自动模型状态验证

  在.Net Core的时代中,框架会帮你自动验证model的state,也就是ModelState。框架会为你自动注册ModelStateInvalidFilter,这个会运行在OnActionExecuting事件里面。this

  基于现有框架的代码编写的话,因此咱们再也不须要在业务中耦合这样的模型判断代码,系统内部会检查ModelState是否为Valid,若是为InValid会直接返回400 BadRequest,这样就没有必要执行后面的代码,提升效率。所以,操做方法中再也不须要如下代码: spa

if (!ModelState.IsValid)
{
    return BadRequest(ModelState);
}

  问题引入  

  在咱们的真实开发中,当咱们碰到参数验证没经过400错误时,咱们但愿的是后台返回一个可理解的Json结果返回,而不是直接在页面返回400错误。因此咱们须要替换掉默认的BadRequest响应结果,把结果换成咱们想要的Json结果返回。code

  自定义 BadRequest 响应

  咱们如何改变 ASP.NET Core WEB API 模型验证的默认行为呢?具体的作法是在经过Startup的ConfigureServices方法配置ApiBehaviorOptions来实现,先来看一下这个类。 orm

  public class ApiBehaviorOptions
    {
        public Func<ActionContext, IActionResult> InvalidModelStateResponseFactory { get; set; }

        public bool SuppressModelStateInvalidFilter { get; set; }

        public bool SuppressInferBindingSourcesForParameters { get; set; }

        public bool SuppressConsumesConstraintForFormFileParameters { get; set; }
    }

  全部bool类型的属性默认都是false。blog

  方案一

  当 SuppressModelStateInvalidFilter 属性设置为 true 时,会禁用默认行为  事件

  public void ConfigureServices(IServiceCollection services)
        {           
            services
                  .AddMvc()
                .AddXmlSerializerFormatters() //设置支持XML格式输入输出
                .SetCompatibilityVersion(CompatibilityVersion.Version_2_1);

            //禁用默认行为
            services.Configure<ApiBehaviorOptions>(options =>
            {
                options.SuppressModelStateInvalidFilter = true;
            });
        }

  当咱们禁用完以后,须要咱们自定义的返回结果了,咱们使用上面的定义的ApiValidationFilter进行拦截和返回。须要在ConfigureServices方法里面把这个拦截器注册进来ip

     public void ConfigureServices(IServiceCollection services)
        {
            .....
            services
                  .AddMvc(options =>
                  {
                      options.Filters.Add<ApiValidationFilter>();
                  })
                .AddXmlSerializerFormatters() //设置支持XML格式输入输出
                .SetCompatibilityVersion(CompatibilityVersion.Version_2_1);

        }

  方案二  

  这也是官网的推荐的作法是,若要自定义验证错误引起的响应,请使用InvalidModelStateResponseFactory。这个InvalidModelStateResponseFactory是一个参数为ActionContext,返回值为IActionResult的委托,具体实现以下:  

    public void ConfigureServices(IServiceCollection services)
        {           
            services
                  .AddMvc()
                .AddXmlSerializerFormatters() //设置支持XML格式输入输出
                .SetCompatibilityVersion(CompatibilityVersion.Version_2_1);

            //参数验证
            services.Configure<ApiBehaviorOptions>(options =>
            {
                options.InvalidModelStateResponseFactory = (context) =>
                {
                    var error = context.ModelState.GetValidationSummary();
                    
                    return new JsonResult(Result.FromError($"参数验证不经过:{error.ToString()}", ResultCode.InvalidParams));
                };
            });
        }

   上面的代码是覆盖ModelState管理的默认行为(ApiBehaviorOptions),当数据模型验证失败时,程序会执行这段代码。没经过验证的ModelState,把它抛出的错误信息经过格式化利用JsonResult返回给客户端。

  总结

  咱们在实际应用过程当中,针对WebApi的开发基本上对于全部的请求都是要返回自定义结果的,因此咱们须要覆盖默认的覆盖默认的模型认证行为,上面给出了两种方案:

  第一种方案:符合Framework时代的风格,须要额外在指定覆盖原有的模型验证(SuppressModelStateInvalidFilter = true

  第二种方案:官方建议作法,符合Core时代的风格,只需复写InvalidModelStateResponseFactory委托便可,我的也推荐第二种方案。

 

  若是你有更好的想法,欢迎给我留言交流。

相关文章
相关标签/搜索