有时,我们可能在WebAPI中包含一些具有调试功能的请求。比如我们上次的文章中“晕了!这个配置值从哪来的?”使用的获取配置值的功能:
endpoints.MapGet("/test2/{key:alpha}",asynccontext={varkey=context.Request.RouteValues["key"].ToString();foreach(varproviderinConfiguration.Providers.Reverse()){if(provider.TryGet(key,outstringvalue)){awaitcontext.Response.WriteAsync(provider.ToString());awaitcontext.Response.WriteAsync("\r\n");awaitcontext.Response.WriteAsync(value);break;}}});
但你绝不会想在生产环境中暴露它们。要想实现此目的,有多种方案:
用户权限验证编译成单独dll,不发布到生产环境这些方案各有利弊,这里我们介绍一种使用Middleware实现的简单方案。
实现从上图可以看到,请求要访问到实际路由,需要先经过Middleware,我们可以在最外层的Middleware进行检查,只有满足条件的请求才能通过,否则返回错误。
Middleware代码如下:
publicclassDebugMiddleware:IMiddleware{publicasyncTaskInvokeAsync(HttpContextcontext,RequestDelegatenext){varisDebugEndpoint=context.Request.Path.Value.Contains("/test");vardebugKey=context.Request.Query["debugKey"].ToString();vardebugKeyInConfig="";//来自于配置if(isDebugEndpointdebugKey!=debugKeyInConfig){context.SetEndpoint(newEndpoint((context)={context.Response.StatusCode=StatusCodes.StatusForbidden;returnTask.CompletedTask;},EndpointMetadataCollection.Empty,"无权访问"));}awaitnext(context);}}
当请求地址包含“/test”时,检查请求参数debugKey是否和配置的值一样,如果不一样就返回错误。
效果如下图:
结论如果你觉得这篇文章对你有所启发,请