.Net core 下Swagger如何隐藏接口的显示

    Swagger是这个很是强大的api文档工具,一般能够用来测试接口,和查看接口,就像这样:api

很是的好用和快捷,这是一个小小的demo,咱们在完成系统时,发布后,外部依旧能够用/swagger访问到这个接口文档,这就形成了系统不安全的问题,那么咱们应该怎么办呢,咱们须要隐藏这些接口。安全

如何隐藏呢,首先咱们须要创建一个新类:工具

namespace TR.Unit
{
    using System;
    using System.Linq;
    using Microsoft.AspNetCore.Mvc.ApiExplorer;
    using Swashbuckle.AspNetCore.Swagger;
    using Swashbuckle.AspNetCore.SwaggerGen;

    /// <summary>
    /// 隐藏接口,不生成到swagger文档展现
    /// </summary>
    [AttributeUsage(AttributeTargets.Method | AttributeTargets.Class)] //此特性能够在方法上和类上使用

    public partial class HiddenAttribute : Attribute
    {
    }

    public class HiddenFilter : IDocumentFilter
    {
        public void Apply(SwaggerDocument swaggerDoc, DocumentFilterContext context)
        {
            foreach (ApiDescription apiDescription in context.ApiDescriptionsGroups.Items.SelectMany(e => e.Items))
            {
                if (apiDescription.ControllerAttributes().OfType<HiddenAttribute>().Count() == 0
                    && apiDescription.ActionAttributes().OfType<HiddenAttribute>().Count() == 0)
                {
                    continue;
                }

                var key = "/" + apiDescription.RelativePath.TrimEnd('/');
                if (!key.Contains("/test/") && swaggerDoc.Paths.ContainsKey(key))
                {
                    swaggerDoc.Paths.Remove(key);
                }
            }
        }
    }
}

首先咱们须要引入 using Swashbuckle.AspNetCore.Swagger; using Swashbuckle.AspNetCore.SwaggerGen;测试

这两个命名空间,其次咱们须要继承Attribute 类,方便咱们在须要隐藏的接口中使用隐藏特性,这里命名能够随意,不过仍是直观一点。spa

而后就须要在你想要隐藏的控制器上加上 [Hidden]就能够啦,就像这样code

    [Hidden]
    [Route("Home")]
    [Authorize]
    public class HomeController : Controller
    {
        /// <summary>
        /// 主界面
        /// </summary>
        /// <returns></returns>
        [HttpGet]
        public IActionResult Home()
        {
            return View();
        }
    }

这里是隐藏控制器下全部的接口,也能够方法上隐藏,从而暴露一些接口。blog

这就是隐藏Swagger接口的显示了,下次写Swagger接口文档如何分组。继承

相关文章
相关标签/搜索