C＃ Tcp服务器如何限制同一个IP的连接数量？

---

一、说明

今天有小伙伴提问，他在使用RRQMSocket创建服务器后，想实现一个功能，就是“限制同一个IP的连接数量”。我和他说，自定义一个插件即可，结果。。。。他不会。

那ok，我给大家示例一下吧。

---

二、程序集源码

2.1 源码位置

• Gitee
• Github

2.2 说明文档

文档首页

三、安装

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

VS、Unity安装和使用Nuget包

四、创建插件

class LimitNumberOfConnectionsPlugin : TcpPluginBase
{
    
    [DependencyInject(2)]
    public LimitNumberOfConnectionsPlugin(int max,ILog logger)
    {
    
        this.Max = max;
        this.Logger = logger;

        logger.Message($"限制连接插件生效，同一IP限制{
      max}个连接");
    }

    readonly ConcurrentDictionary<string, Count> m_ipToCount = new ConcurrentDictionary<string, Count>();

    public int Max {
     get; }

    protected override void OnConnecting(ITcpClientBase client, ClientOperationEventArgs e)
    {
    
        Count count = m_ipToCount.GetOrAdd(client.IP, (s) => {
     return new Count(); });

        if (count.Increment() > this.Max)
        {
    
            count.Decrement();
            e.IsPermitOperation = false;//表示不许连接
            e.Handled = true;//并且已经处理该消息。
            this.Logger.Warning($"IP={
      client.IP}的客户端，连接数达到设置阈值。已拒绝连接。");
            return;
        }
        base.OnConnecting(client, e);
    }

    protected override void OnDisconnected(ITcpClientBase client, ClientDisconnectedEventArgs e)
    {
    
        if (m_ipToCount.TryGetValue(client.IP,out Count count))
        {
    
            if (count.Decrement() == 0)
            {
    
                m_ipToCount.TryRemove(client.IP, out _);
            }
        }
        base.OnDisconnected(client, e);
    }
}

class Count
{
    
    private int num;

    public int Num
    {
    
        get {
     return num; }
    }

    public int Increment()
    {
    
        return Interlocked.Increment(ref num);
    }

    public int Decrement()
    {
    
        return Interlocked.Decrement(ref num);
    }
}

五、启动服务器

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()//载入配置 
        .SetListenIPHosts(new IPHost[] {
     new IPHost("127.0.0.1:7789"), new IPHost(7790) })//同时监听两个地址
        .SetMaxCount(10000)
        .SetThreadCount(100)
        .UsePlugin())
        .Start();//启动

    service.AddPlugin<LimitNumberOfConnectionsPlugin>();

    Console.ReadKey();
}

六、效果

因为代码全在这里了，所以，也就不用上传项目了。