SwaggerIgnore

复制代码
    /// <summary>
    /// SwaggerIgnoreAttribute
    /// </summary>
    [AttributeUsage(AttributeTargets.Class | AttributeTargets.Method|AttributeTargets.Field|AttributeTargets.Property, AllowMultiple = false, Inherited = true)]
    public class SwaggerIgnoreAttribute : Attribute, IFilterMetadata
    {
        /// <summary>
        /// construct
        /// </summary>
        /// <param name="ignore"></param>
        public SwaggerIgnoreAttribute(bool ignore = true)
        {
            IgnoreApi = ignore;
        }
        /// <summary>
        /// 
        /// </summary>
        public string GroupName { get; set; }

        /// <summary>
        /// 
        /// </summary>
        public bool IgnoreApi { get; set; }
    }
复制代码
复制代码
    /// <summary>
    /// SwaggerIgnoreFilter
    /// </summary>
    public class SwaggerIgnoreFilter : Swashbuckle.AspNetCore.SwaggerGen.IDocumentFilter
    {
        public void Apply(OpenApiDocument swaggerDoc, DocumentFilterContext context)
        {
            var ignoreApis =
                context.ApiDescriptions.Where(o =>
                {
                    if (o.TryGetMethodInfo(out MethodInfo methodInfo))
                    {
                        if (methodInfo == null) return false;
                        if (methodInfo.CustomAttributes.Any(any => any.AttributeType == typeof(SwaggerIgnoreAttribute)))
                        {
                            var attribute = methodInfo.GetCustomAttribute<SwaggerIgnoreAttribute>();
                            return attribute?.IgnoreApi ?? false;
                        }
                        else if (methodInfo.DeclaringType?.CustomAttributes.Any(any =>
                            any.AttributeType == typeof(SwaggerIgnoreAttribute)) ?? false)
                        {
                            var attribute = methodInfo.DeclaringType.GetCustomAttribute<SwaggerIgnoreAttribute>();
                            return attribute?.IgnoreApi ?? false;
                        }
                        return false;
                    }

                    return false;
                }).ToList();
            if (ignoreApis.Count > 0)
            {
                foreach (var ignoreApi in ignoreApis)
                {
                    swaggerDoc.Paths.Remove("/" + ignoreApi.RelativePath);
                }
            }
        }
    }
复制代码
 services.AddSwaggerGen(
                options =>
                {
                    options.DocumentFilter<SwaggerIgnoreFilter>();
                });

 

posted @   向萧  阅读(56)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示