Asp.Net Core 2.x 和 3.x WebAPI 使用 Swagger 时 API Controller 控制器 Action 方法 隐藏 hidden 与 and 分组 group

2021-02-01 02:16

阅读:391

标签:ica   访问   需要   collect   path   注入   tty   public   uil   

1、前言

为什么我们要隐藏部分接口?

因为我们在用swagger代替接口的时候,难免有些接口会直观的暴露出来,比如我们结合Consul一起使用的时候,会将健康检查接口以及报警通知接口暴露出来,这些接口有时候会出于方便考虑,没有进行加密,这个时候我们就需要把接口隐藏起来,只有内部的开发者知道。

为什么要分组?

通常当我们写前后端分离的项目的时候,难免会遇到编写很多接口供前端页面进行调用,当接口达到几百个的时候就需要区分哪些是框架接口,哪些是业务接口,这时候给swaggerUI的接口分组是个不错的选择。

技术图片

 

swagger的基本使用这里将不再赘述,可以阅读微软官方文档,即可基本使用

2、swaggerUI中加入授权请求

  • 新建 HttpHeaderOperationFilter 操作过滤器,继承 Swashbuckle.AspNetCore.SwaggerGen.IOperationFilter 接口,实现 Apply 方法
    技术图片
    /// 
    /// swagger请求头
    /// 
    public class HttpHeaderOperationFilter : IOperationFilter
    {
        public void Apply(Operation operation, OperationFilterContext context)
        {
            #region 新方法
            if (operation.Parameters == null)
            {
                operation.Parameters = new List();
            }
    
            if (context.ApiDescription.TryGetMethodInfo(out MethodInfo methodInfo))
            {
                if (!methodInfo.CustomAttributes.Any(t => t.AttributeType == typeof(AllowAnonymousAttribute))
                        &&!(methodInfo.ReflectedType.CustomAttributes.Any(t => t.AttributeType == typeof(AuthorizeAttribute))))
                {
                    operation.Parameters.Add(new NonBodyParameter
                    {
                        Name = "Authorization",
                        In = "header",
                        Type = "string",
                        Required = true,
                        Description = "请输入Token,格式为bearer XXX"
                    });
                }
            }
            #endregion
    
            #region 已过时
            //if (operation.Parameters == null)
            //{
            //    operation.Parameters = new List();
            //}
            //var actionAttrs = context.ApiDescription.ActionAttributes().ToList();
            //var isAuthorized = actionAttrs.Any(a => a.GetType() == typeof(AuthorizeAttribute));
            //if (isAuthorized == false)
            //{
            //    var controllerAttrs = context.ApiDescription.ControllerAttributes();
            //    isAuthorized = controllerAttrs.Any(a => a.GetType() == typeof(AuthorizeAttribute));
            //}
            //var isAllowAnonymous = actionAttrs.Any(a => a.GetType() == typeof(AllowAnonymousAttribute));
            //if (isAuthorized && isAllowAnonymous == false)
            //{
            //    operation.Parameters.Add(new NonBodyParameter
            //    {
            //        Name = "Authorization",
            //        In = "header",
            //        Type = "string",
            //        Required = true,
            //        Description = "请输入Token,格式为bearer XXX"
            //    });
            //}
            #endregion
        }
    }
    技术图片
  • 然后修改 Startup.cs 中的 ConfigureServices 方法,添加我们自定义的 HttpHeaderOperationFilter 过滤器
    技术图片
    public IServiceProvider ConfigureServices(IServiceCollection services)
    {
        ...
        services.AddSwaggerGen(c =>
        {
            ...
            c.OperationFilter();
        });
        ...
    }
    技术图片

    这时候我们再访问swaggerUI就可以输入Token了
    技术图片

3、API分组

  • 修改 Startup.cs 中的 ConfigureServices 方法,添加多个swagger文档
    技术图片
    public IServiceProvider ConfigureServices(IServiceCollection services)
    {
        ...
        services.AddSwaggerGen(c =>
        {
            c.SwaggerDoc("v1", new Info
            {
                Version = "v1",
                Title = "接口文档",
                Description =  "接口文档-基础",
                TermsOfService = "",
                Contact = new Contact
                {
                    Name = "XXX1111",
                    Email = "XXX1111@qq.com",
                    Url = ""
                }
            });
    
            c.SwaggerDoc("v2", new Info
            {
                Version = "v2",
                Title = "接口文档",
                Description =  "接口文档-基础",
                TermsOfService = "",
                Contact = new Contact
                {
                    Name = "XXX2222",
                    Email = "XXX2222@qq.com",
                    Url = ""
                }
            });
    
            //反射注入全部程序集说明
            GetAllAssemblies().Where(t => t.CodeBase.EndsWith("Controller.dll")).ToList().ForEach(assembly =>
                {
                    c.IncludeXmlComments(assembly.CodeBase.Replace(".dll", ".xml"));
                });
    
            c.OperationFilter();
            //c.DocumentFilter();
        });
        ...
    }
    技术图片
  • 修改 Startup.cs 中的 Configure 方法,加入
    技术图片
    public void Configure(IApplicationBuilder app, ILoggerFactory loggerFactory)
    {
        ...
        app.UseSwagger();
        app.UseSwaggerUI(c =>
        {
            c.SwaggerEndpoint("/swagger/v2/swagger.json", "接口文档-基础");//业务接口文档首先显示
            c.SwaggerEndpoint("/swagger/v1/swagger.json", "接口文档-业务");//基础接口文档放后面后显示
            c.RoutePrefix = string.Empty;//设置后直接输入IP就可以进入接口文档
        });
        ...
    
    }
    技术图片

     

  • 然后还要在我们的控制器上面标注swagger文档的版本
    技术图片

    这时候我们就可以将接口文档进行分组显示了
    技术图片

4、API隐藏

    • 创建自定义隐藏特性 HiddenApiAttribute.cs 
      技术图片
      /// 
      /// 隐藏swagger接口特性标识
      /// 
      [AttributeUsage(AttributeTargets.Method | AttributeTargets.Class)]
      public class HiddenApiAttribute:System.Attribute
      {
      }
      技术图片
    • 创建API隐藏过滤器 HiddenApiFilter 继承 Swashbuckle.AspNetCore.SwaggerGen.IDocumentFilter 接口,实现 Apply 方法
      技术图片
      /// 
      /// 自定义Swagger隐藏过滤器
      /// 
      public class HiddenApiFilter : IDocumentFilter
      {
          public void Apply(SwaggerDocument swaggerDoc, DocumentFilterContext context)
          {
              foreach (ApiDescription apiDescription in context.ApiDescriptions)
              {
                  if (apiDescription.TryGetMethodInfo(out MethodInfo method))
                  {
                      if (method.ReflectedType.CustomAttributes.Any(t=>t.AttributeType==typeof(HiddenApiAttribute))
                              || method.CustomAttributes.Any(t => t.AttributeType == typeof(HiddenApiAttribute)))
                      {
                          string key = "/" + apiDescription.RelativePath;
                          if (key.Contains("?"))
                          {
                              int idx = key.IndexOf("?", System.StringComparison.Ordinal);
                              key = key.Substring(0, idx);
                          }
                          swaggerDoc.Paths.Remove(key);
                      }
                  }
              }
          }
      }
      技术图片
    • 在 Startup.cs 中使用 HiddenApiFilter 
      技术图片
      public IServiceProvider ConfigureServices(IServiceCollection services)
      {
          ...
          services.AddSwaggerGen(c =>
          {
              c.SwaggerDoc("v1", new Info
              {
                  Version = "v1",
                  Title = "接口文档",
                  Description =  "接口文档-基础",
                  TermsOfService = "",
                  Contact = new Contact
                  {
                      Name = "XXX1111",
                      Email = "XXX1111@qq.com",
                      Url = ""
                  }
              });
      
              c.SwaggerDoc("v2", new Info
              {
                  Version = "v2",
                  Title = "接口文档",
                  Description =  "接口文档-基础",
                  TermsOfService = "",
                  Contact = new Contact
                  {
                      Name = "XXX2222",
                      Email = "XXX2222@qq.com",
                      Url = ""
                  }
              });
      
              //反射注入全部程序集说明
              GetAllAssemblies().Where(t => t.CodeBase.EndsWith("Controller.dll")
                  && !t.CodeBase.Contains("Common.Controller.dll")).ToList().ForEach(assembly =>
                  {
                      c.IncludeXmlComments(assembly.CodeBase.Replace(".dll", ".xml"));
                  });
      
              c.OperationFilter();
              c.DocumentFilter();
          });
          ...
      }
      技术图片
    • 示例:
      我这里提供了Consul的心跳检车接口
      技术图片

      但是在接口文档中并没有显示出来
      技术图片

       

Asp.Net Core 2.x 和 3.x WebAPI 使用 Swagger 时 API Controller 控制器 Action 方法 隐藏 hidden 与 and 分组 group

标签:ica   访问   需要   collect   path   注入   tty   public   uil   

原文地址:https://www.cnblogs.com/webenh/p/11605718.html


评论


亲,登录后才可以留言!