using Microsoft.AspNetCore.Authorization; using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Mvc; using Microsoft.AspNetCore.Mvc.Filters; using Microsoft.Extensions.DependencyInjection; using System; using System.Threading.Tasks; using Znyc.Cloudcar.Admin.WebApi.Auth; namespace Znyc.Cloudcar.Admin.WebApi.Attributes { /// /// 定时任务启动密钥 /// [AttributeUsage(AttributeTargets.Class | AttributeTargets.Method, AllowMultiple = true)] public class QuartzAttribute : AuthorizeAttribute, IAuthorizationFilter, IAsyncAuthorizationFilter { private IApiHandler _apiHandler; /// /// /// /// public Task OnAuthorizationAsync(AuthorizationFilterContext context) { OnAuthorization(context); return Task.CompletedTask; } /// /// /// public async void OnAuthorization(AuthorizationFilterContext context) { string apiUrl = context.HttpContext.Request.Path.Value; _apiHandler = context.HttpContext.RequestServices.GetService(); Security.Entitys.ApiEntity entity = await _apiHandler.ValidateAsync(apiUrl); //key/value 存入数据库去查询,不同的api对应不同的key/value string authHeader = context.HttpContext.Request.Headers[entity.AuthKey]; if (authHeader != null && authHeader.Equals(entity.AuthValue)) { return; } context.Result = new StatusCodeResult(StatusCodes.Status401Unauthorized); } } }