C＃ TCP如何限制单个客户端的访问流量

---

一、说明

使用RRQMSocket创建服务器后，想实现一个限流功能，应该如何实现呢？

---

二、程序集源码

2.1 源码位置

• Gitee
• Github

2.2 说明文档

文档首页

三、安装

Nuget安装RRQMSocket即可，具体步骤详看链接博客。

VS、Unity安装和使用Nuget包

四、创建插件

public class MyThrottlingPlugin : TcpPluginBase
{
    
    private readonly int m_max;

    [DependencyInject(10)]
    public MyThrottlingPlugin(int max)
    {
    
        this.m_max = max;
        this.Order = int.MaxValue;//提升优先级
    }

    protected override void OnConnected(ITcpClientBase client, RRQMEventArgs e)
    {
    
        client.InitFlowGate(this.m_max);//初始化流量计数器。
        base.OnConnected(client, e);
    }

    protected override void OnReceivedData(ITcpClientBase client, ReceivedDataEventArgs e)
    {
    
        client.GetFlowGate().AddCheckWait(e.ByteBlock.Len);//此处假设接收的是ByteBlock数据。如果是自定义适配器，按需增量即可。
        base.OnReceivedData(client, e);
    }
}

五、创建扩展类

 /// <summary>
 /// 一个流量计数器扩展。
 /// </summary>
 static class DependencyExtensions
 {
    
     public static readonly DependencyProperty FlowGateProperty =
         DependencyProperty.Register("FlowGate", typeof(FlowGate), typeof(DependencyExtensions), null);

     public static void InitFlowGate(this IRRQMDependencyObject dependencyObject, int max)
     {
    
         dependencyObject.SetValue(FlowGateProperty, new FlowGate() {
     Maximum = max });
     }

     public static FlowGate GetFlowGate(this IRRQMDependencyObject dependencyObject)
     {
    
         return dependencyObject.GetValue<FlowGate>(FlowGateProperty);
     }
 }

六、启动服务器

static void Main(string[] args)
{
    
    TcpService service = new TcpService();
    service.Connecting += (client, e) => {
     };//有客户端正在连接
    service.Connected += (client, e) => {
     };//有客户端连接
    service.Disconnected += (client, e) => {
     };//有客户端断开连接
    service.Received += (client, byteBlock, requestInfo) =>
    {
    
        //从客户端收到信息
        string mes = Encoding.UTF8.GetString(byteBlock.Buffer, 0, byteBlock.Len);
        Console.WriteLine($"已从{
      client.ID}接收到信息：{
      mes}");

        client.Send(mes);//将收到的信息直接返回给发送方
    };

    service.Setup(new RRQMConfig()//载入配置 
        .UsePlugin()
        .SetListenIPHosts(new IPHost[] {
     new IPHost("127.0.0.1:7789"), new IPHost(7790) })//同时监听两个地址
        .SetMaxCount(10000)
        .SetThreadCount(100))
        .Start();//启动
    service.AddPlugin<MyThrottlingPlugin>();

    Console.ReadLine();
}

七、效果

本文示例demo

实际上，该插件也能用于客户端。同时，也能限制发送流量限制。