44 lines
1.3 KiB
C#
44 lines
1.3 KiB
C#
|
||
|
||
using Google.Apis.Discovery;
|
||
using Microsoft.AspNetCore.Mvc.Controllers;
|
||
using Microsoft.OpenApi.Models;
|
||
using Swashbuckle.AspNetCore.Swagger;
|
||
using Swashbuckle.AspNetCore.SwaggerGen;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
|
||
namespace LongHuParkApi.Common
|
||
{
|
||
/// <summary>
|
||
/// swagger过滤操作
|
||
/// </summary>
|
||
public class SiteIdHeaderParameter : IOperationFilter
|
||
{
|
||
/// <summary>
|
||
/// 使用
|
||
/// </summary>
|
||
/// <param name="operation"></param>
|
||
/// <param name="context"></param>
|
||
public void Apply(OpenApiOperation operation, OperationFilterContext context)
|
||
{
|
||
if (operation.Parameters == null)
|
||
{
|
||
operation.Parameters = new List<OpenApiParameter>();
|
||
}
|
||
//判断是否存在AllowAnonymousAttribute属性
|
||
var count = context.MethodInfo.CustomAttributes.Where(c=>c.AttributeType.Name == "AllowAnonymousAttribute").Count();
|
||
if (count == 0)
|
||
{
|
||
//不存在属性，则需要token验证
|
||
operation.Parameters.Add(new OpenApiParameter
|
||
{
|
||
Name = "token",
|
||
In = ParameterLocation.Header,
|
||
Required = false
|
||
});
|
||
}
|
||
}
|
||
}
|
||
}
|