增加防重复提交

This commit is contained in:
18942506660
2024-03-30 17:31:58 +08:00
parent c0aa46faa2
commit 7e18d3c6d6
4 changed files with 93 additions and 0 deletions

View File

@@ -82,6 +82,8 @@ namespace WMS.Web.Api
//ͳһ<CDB3><EFBFBD><ECB3A3><EFBFBD><EFBFBD><EFBFBD>м<EFBFBD><D0BC><EFBFBD>
app.UseMiddleware<AuthorizationTokenSecurityPolicy>();
app.UseMiddleware<ErrorHandlingMiddleware>();
app.UseMiddleware<PlatformActionMiddleware>();
app.UseRouting();
app.UseEndpoints(endpoints =>
{

View File

@@ -719,6 +719,12 @@
<param name="dto"></param>
<returns></returns>
</member>
<member name="M:WMS.Web.Api.Controllers.TestController.TestPost">
<summary>
测试重复提交过滤器
</summary>
<returns></returns>
</member>
<member name="M:WMS.Web.Api.Controllers.TestController.Refresh">
<summary>
出库任务单手动同步金蝶数据

View File

@@ -0,0 +1,85 @@
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.Filters;
using Microsoft.Extensions.Caching.Memory;
using Microsoft.Extensions.Logging;
using System;
using System.Collections.Generic;
using System.Net;
using System.Text;
using WMS.Web.Core.Internal.Results;
using WMS.Web.Domain.Services.Public;
using Microsoft.AspNetCore.Http;
using Newtonsoft.Json;
using System.IO;
using System.Threading.Tasks;
namespace WMS.Web.Repositories.DependencyInjection
{
/// <summary>
/// 防重复提交
/// </summary>
public class PlatformActionMiddleware
{
private readonly RequestDelegate next;
private readonly ILogger<PlatformActionMiddleware> _logger;
private readonly RedisClientService _redisClientService;
/// <summary>
/// 统一错误处理中间件
/// </summary>
public PlatformActionMiddleware(RequestDelegate next, ILogger<PlatformActionMiddleware> logger, RedisClientService redisClientService)
{
this.next = next;
_logger = logger;
_redisClientService = redisClientService;
}
private string bodyStr = "";
/// <summary>
/// 激活
/// </summary>
/// <param name="context"></param>
/// <returns></returns>
public async Task Invoke(HttpContext context)
{
//默认 请求body只能读取一次 所以在这里需要手动把他设置成读取多次 并且 不能使用using 释放
context.Request.EnableBuffering();
StreamReader requestReader = new StreamReader(context.Request.Body, Encoding.UTF8);
bodyStr = await requestReader.ReadToEndAsync();
context.Request.Body.Seek(0, SeekOrigin.Begin);
string httpMethod = WebUtility.HtmlEncode(context.Request.Method);
if (httpMethod == "POST")
{
//使用请求路径作为唯一key
string path = context.Request.Path;
string authorization = context.Request.Headers["Authorization"];
string cacheToken = $"{authorization}_{path}";
string keyValue = string.IsNullOrEmpty(bodyStr) ? "yc" : bodyStr;
if (path != null)
{
//var cache = iZen.Utils.Core.iCache.CacheManager.GetCacheValue(cacheToken);
string cv = _redisClientService.GetStringKey(cacheToken);
if (cv == null && bodyStr != cv)
{
//iZen.Utils.Core.iCache.CacheManager.SetChacheValueSeconds(cacheToken, keyValue, 1);
//设置缓存10秒过期
_redisClientService.SetStringKey(cacheToken, keyValue, TimeSpan.FromSeconds(10));
}
else
{
context.Response.StatusCode = 401;
var result = JsonConvert.SerializeObject(new { status = 4012343, data = string.Empty, message = "重复提交" });
context.Response.ContentType = "application/json;charset=utf-8";
await context.Response.WriteAsync(result);
return;
}
}
}
await next(context);
}
}
}