AspNetCoreRateLimit应用于MVC项目求助

怼烎@ 2024-03-22 13:47 62阅读 0赞

就是我的需求是评论接口限流，然后触发限流后要回到文章页面告诉用户你触发了限流，但是，使用这个组件，他会将返回信息以页面的形式返回给你，我并不知道该如何去让他回到文章页面，也是琢磨了很久，用中间件去实现了这个效果，但是感觉不是很理想，如果有大佬知道更好的办法，可以私信或评论,感激不尽。

## 实现的效果图 ##

![44eff57d08d72a041439926bb93cc8ab.png][]

## 评论接口 ##

![aaca9e8e1e3c6adc24f5b2d2140e31ce.png][]

## 文章接口 ##

![c364d7b5fb58721837660e030acf2836.png][]

`_messages.Warning`是博客开源作者封装的提示信息组件,可以采用别的方式去提示，问题不大。这里就是将从缓存中的提示信息提取出来，然后因为这里用的是缓存，用session做的唯一值处理，所以用session去取出来，如果从缓存中查出来存在，则提示被限流。

## 代码实现 ##

原理就是把组件自带的信息提示设置为空字符串，自己在中间件中去使用。

这是限流规则:

![9bbb20d9ac8cf2bda313064024f7d6f7.png][]

这里参数就不做多的解释，可以去看之前发布的那篇文章。只要把Content设置为空字符串即可。

然后就开始去写中间件去处理触发了限流该怎么做

需要注册缓存服务

`builder.Services.AddMemoryCache();`

app.Use(async (context, next) =>
    {
        var cache = context.RequestServices.GetRequiredService<IMemoryCache>();
        
        // 保存原始响应流
        var originalBody = context.Response.Body;
        
        // 创建一个新的响应流
        using var responseBody = new MemoryStream();
        context.Response.Body = responseBody;
        
        // 加载当前用户的 Session 对象
        await context.Session.LoadAsync();
        
        await next.Invoke();
        
        if (context.Response.StatusCode == 429)
        {
            var referer = context.Request.Headers["Referer"].ToString();
            
            // 从 Session 中获取一个字符串值
            var value = context.Session.GetString("key");
            if (string.IsNullOrEmpty(value))
            {
                // 如果 Session 中没有值，则设置一个字符串值
                context.Session.SetString("key", "value");
            }
            var sessionId = context.Session.Id;
            if (!cache.TryGetValue("Errors", out Dictionary<string, string> errors))
            {
                errors = new Dictionary<string, string>();
                cache.Set("Errors", errors, TimeSpan.FromSeconds(10));
            }
            
            errors[sessionId] = "您的请求已被限流，请稍后再试。";
            
            // 重置响应流位置
            responseBody.Seek(0, SeekOrigin.Begin);
            
            // 读取响应内容
            // var bodyText = new StreamReader(responseBody).ReadToEnd();
            
            // 设置新的响应流
            context.Response.Body = originalBody;
            
            // 设置新的响应状态码
            context.Response.StatusCode = 302;
    
            context.Response.Headers["Location"] = referer;
            
        }
        else
        {
            // 将响应流写回到原始响应流中
            responseBody.Seek(0, SeekOrigin.Begin);
            await responseBody.CopyToAsync(originalBody);
            
        }
    });

注意这个中间件处理要放在`app.UseRateLimit();`前面。

## 结尾 ##

AspNetCoreRateLimit原本就讲限流的ip存放在redis当中了的，但是我就是查不出来，如果能用该组件自带的方法查询出来，就不需要再写一个中间件，当429的时候再用缓存存一次会话了。

总之暂且先用这种办法吧，如果有更好的方法可以评论哟~

[44eff57d08d72a041439926bb93cc8ab.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/22/becbcdb0c43e48c388de3ff547f49489.png
[aaca9e8e1e3c6adc24f5b2d2140e31ce.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/22/3d9289f0a09e445ab00563c2f86afdfd.png
[c364d7b5fb58721837660e030acf2836.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/22/0c8590db73194eb2809fb5617a1a208e.png
[9bbb20d9ac8cf2bda313064024f7d6f7.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/22/4be9b03acde64655a15f0ce04d47c529.png