ASP.NET Core使用filter和redis实现接口防重

猪毛一小撮

背景

日常开发中，经常需要对一些响应不是很快的关键业务接口增加防重功能，即短时间内收到的多个相同的请求，只处理一个，其余不处理，避免产生脏数据。这和幂等性（idempotency）稍微有点区别，幂等性要求的是对重复请求有相同的效果和结果，通常需要在接口内部执行业务操作前检查状态；而防重可以认为是一个业务无关的通用功能，在ASP.NET Core中我们可以借助过Filter和redis实现。
关于Filter

Filter的由来可以追溯到ASP.NET MVC中的ActionFilter和ASP.NET Web API中的ActionFilterAttribute。ASP.NET Core将这些不同类型的Filter统一为一种类型，称为Filter，以简化API和提高灵活性。ASP.NET Core中Filter可以用于实现各种功能，例如身份验证、日志记录、异常处理、性能监控等。

通过使用Filter，我们可以在请求处理管道的特定阶段之前或者之后运行自定义代码，达到AOP的效果。

编码实现

防重组件的思路很简单，将第一次请求的某些参数作为标识符存入redis中，并设置过期时间，下次请求过来，先检查redis相同的请求是否已被处理；
作为一个通用组件，我们需要能让使用者自定义作为标识符的字段以及过期时间，下面开始实现。
PreventDuplicateRequestsActionFilter

1. public class PreventDuplicateRequestsActionFilter : IAsyncActionFilter
2. {
3.     public string[] FactorNames { get; set; }
4.     public TimeSpan? AbsoluteExpirationRelativeToNow { get; set; }
6.     private readonly IDistributedCache _cache;
7.     private readonly ILogger<PreventDuplicateRequestsActionFilter> _logger;
9.     public PreventDuplicateRequestsActionFilter(IDistributedCache cache, ILogger<PreventDuplicateRequestsActionFilter> logger)
10.     {
11.         _cache = cache;
12.         _logger = logger;
13.     }
15.     public async Task OnActionExecutionAsync(ActionExecutingContext context, ActionExecutionDelegate next)
16.     {
17.         var factorValues = new string?[FactorNames.Length];
19.         var isFromBody =
20.             context.ActionDescriptor.Parameters.Any(r => r.BindingInfo?.BindingSource == BindingSource.Body);
21.         if (isFromBody)
22.         {
23.             var parameterValue = context.ActionArguments.FirstOrDefault().Value;
24.             factorValues = FactorNames.Select(name =>
25.                 parameterValue?.GetType().GetProperty(name)?.GetValue(parameterValue)?.ToString()).ToArray();
26.         }
27.         else
28.         {
29.             for (var index = 0; index < FactorNames.Length; index++)
30.             {
31.                 if (context.ActionArguments.TryGetValue(FactorNames[index], out var factorValue))
32.                 {
33.                     factorValues[index] = factorValue?.ToString();
34.                 }
35.             }
36.         }
38.         if (factorValues.All(string.IsNullOrEmpty))
39.         {
40.             _logger.LogWarning("Please config FactorNames.");
42.             await next();
43.             return;
44.         }
46.         var idempotentKey = $"{context.HttpContext.Request.Path.Value}:{string.Join("-", factorValues)}";
47.         var idempotentValue = await  _cache.GetStringAsync(idempotentKey);
48.         if (idempotentValue != null)
49.         {
50.             _logger.LogWarning("Received duplicate request({},{}), short-circuiting...", idempotentKey, idempotentValue);
51.             context.Result = new AcceptedResult();
52.         }
53.         else
54.         {
55.             await _cache.SetStringAsync(idempotentKey, DateTimeOffset.UtcNow.ToString(),
56.                 new DistributedCacheEntryOptions {AbsoluteExpirationRelativeToNow = AbsoluteExpirationRelativeToNow});
57.             await next();
58.         }
59.     }
60. }

复制代码

PreventDuplicateRequestsActionFilter里，我们首先通过反射从 ActionArguments拿到指定参数字段的值，由于从request body取值略有不同，我们需要分开处理；接下来开始拼接key并检查redis，如果key已经存在，我们需要短路请求，这里直接返回的是 Accepted (202)而不是Conflict (409)或者其它错误状态，是为了避免上游已经调用失败而继续重试。
PreventDuplicateRequestsAttribute

防重组件的全部逻辑在PreventDuplicateRequestsActionFilter中已经实现，由于它需要注入 IDistributedCache和ILogger对象，我们使用IFilterFactory实现一个自定义属性，方便使用。

1. [AttributeUsage(AttributeTargets.Method)]
2. public class PreventDuplicateRequestsAttribute : Attribute, IFilterFactory
3. {
4.     private readonly string[] _factorNames;
5.     private readonly int _expiredMinutes;
7.     public PreventDuplicateRequestsAttribute(int expiredMinutes, params string[] factorNames)
8.     {
9.         _expiredMinutes = expiredMinutes;
10.         _factorNames = factorNames;
11.     }
13.     public IFilterMetadata CreateInstance(IServiceProvider serviceProvider)
14.     {
15.         var filter = serviceProvider.GetService<PreventDuplicateRequestsActionFilter>();
16.         filter.FactorNames = _factorNames;
17.         filter.AbsoluteExpirationRelativeToNow = TimeSpan.FromMinutes(_expiredMinutes);
18.         return filter;
19.     }
20.     public bool IsReusable => false;
21. }

复制代码

注册

为了简单，操作redis，直接使用微软Microsoft.Extensions.Caching.StackExchangeRedis包；注册PreventDuplicateRequestsActionFilter，PreventDuplicateRequestsAttribute无需注册。

1. builder.Services.AddStackExchangeRedisCache(options =>
2. {
3.     options.Configuration = "127.0.0.1:6379,DefaultDatabase=1";
4. });
5. builder.Services.AddScoped<PreventDuplicateRequestsActionFilter>();

复制代码

使用

假设我们有一个接口CancelOrder，我们指定入参中的OrderId和Reason为因子。

1. namespace PreventDuplicateRequestDemo.Controllers
2. {
3.     [Route("api/[controller]")]
4.     [ApiController]
5.     public class OrderController : ControllerBase
6.     {
7.         [HttpPost(nameof(CancelOrder))]
8.         [PreventDuplicateRequests(5, "OrderId", "Reason")]
9.         public async Task<IActionResult> CancelOrder([FromBody] CancelOrderRequest request)
10.         {
11.             await Task.Delay(1000);
12.             return new OkResult();
13.         }
14.     }
16.     public class CancelOrderRequest
17.     {
18.         public Guid OrderId { get; set; }
19.         public string Reason { get; set; }
20.     }
21. }

复制代码

启动程序，多次调用api，除第一次调用成功，其余请求皆被短路

查看redis，已有记录

参考链接

https://learn.microsoft.com/en-us/aspnet/core/mvc/controllers/filters?view=aspnetcore-7.0
https://learn.microsoft.com/en-us/aspnet/core/performance/caching/distributed?view=aspnetcore-7.0

来源:https://www.cnblogs.com/netry/archive/2023/03/15/aspnetcore-prevent-duplicate-requests-filter-redis.html
免责声明：由于采集信息均来自互联网，如果侵犯了您的权益，请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容，谢谢合作！