NET CORE 基于缓存策略的SignalR控制推送频率(每多少秒/多少次)API接口控制(限流)...

Posted 2022-11-21 dotNET跨平台

ASP.NET Core SignalR 概述，自行去官网搜。

SignalR 没有控制和前端推送频率的功能，就是后端一旦发送请求，前端立马响应。或者前端发送请求，后端立马响应，但是如果误操作，或者业务原因，对产生的信息频繁的推送，此时就会对系统的性能产生一定影响。

鉴于以上情况，我对SignalR 发送请求的时候，做了简单的改造。我们可以通过配置文件，或者前后端传入的参数，对SignalR 推送的频率做了限制。

这里我设置的参数是： 每多少秒/每多少次，当时你也可以 每小时/每天...

NET CORE默认自带 Microsoft.Extensions.Caching.Memory 缓存功能，于是我基于缓存的策略，来控制SignalR控制推送频率。

构造函数如下：

private readonly IMemoryCache _memoryCache;

如果你们目前使用Redis把 IMemoryCache 替换成你们的Redis也可以，因为MemoryCache 使用 key-value操作时也是  Set、Get

相关代码如下

public int SignalRxxxxxx(int count = 10, int seconds = 60)
    
        var signalRFrequencyCount = _memoryCache.Get<int>("Count");
        var signalRFrequencyMinutes = _memoryCache.Get<int>("Minutes");
        if (signalRFrequencyMinutes == 0)
        
            _memoryCache.Set("Count", 1);
            _memoryCache.Set("Minutes", seconds, TimeSpan.FromSeconds(seconds));
            //_signalRClients.AddMessageToAll(你的业务);
        
        if (signalRFrequencyCount < count && signalRFrequencyMinutes != 0)
        
            _memoryCache.Set("Count", signalRFrequencyCount + 1);
           // _signalRClients.AddMessageToAll(你的业务);
        
        return signalRFrequencyCount;
    

这个逻辑，其实就是API网关的限流的原理一样，同一个接口，频繁的请求指定次数，就会返回一个状态的提示。这个方法用在全局上，就会对系统的整体请求做处理，当然我这只是仅仅对SignalR的推送做一个频率的控制。