限制来自 IP 的连接数

在编写服务器端应用程序时，限制来自特定 IP 的连接数量是一个很常见的需求。这可以帮助防止某个客户端占用过多服务器资源，从而保护服务器的稳定性和安全性。

以下是一些常见的方法，用于限制来自 IP 的连接数量。它们涵盖了不同的编程语言和服务器软件。

使用iptables

iptables 是 Linux 上常用的防火墙软件，它可以轻松地限制来自特定 IP 的连接数量。我们可以使用如下命令：

iptables -A INPUT -p tcp --syn --dport 80 -m connlimit --connlimit-above 10 -j REJECT

这条命令的含义是限制 TCP 协议、目标端口为 80、来自同一 IP 的连接数量不超过 10。如果超过了这个限制，就拒绝新的连接。

使用nginx

nginx 是一种轻量级的 Web 服务器和反向代理服务器，它也支持限制来自特定 IP 的连接数量。我们可以在 nginx 的配置文件中添加如下代码：

http {
  limit_conn_zone $binary_remote_addr zone=conn_limit_per_ip:10m;

  server {
    location / {
      limit_conn conn_limit_per_ip 10;
      # ...
    }
  }
}

这段代码的含义是，在每个对远程 IP 的连接中，最多同时允许 10 个连接。如果超过了这个限制，nginx 就会返回 503 状态码，表示服务器过载。

使用Java Servlet API

在 Java Servlet API 中，我们可以使用如下代码：

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class ConnectionLimiterServlet extends HttpServlet {
    private static final int MAX_CONNECTIONS_PER_IP = 10;
    private final Map<String, Integer> connectionsPerIp = new ConcurrentHashMap<>();

    @Override
    public void service(HttpServletRequest request, HttpServletResponse response) throws ServletException {
        String remoteIp = request.getRemoteAddr();
        int connections = connectionsPerIp.getOrDefault(remoteIp, 0);

        if (connections >= MAX_CONNECTIONS_PER_IP) {
            response.setStatus(503);
            return;
        }

        connectionsPerIp.put(remoteIp, connections + 1);
        try {
            // 处理请求
        } finally {
            connectionsPerIp.put(remoteIp, connections);
        }
    }
}

这段代码的含义是，对于每个请求，我们先获取远程 IP，然后检查该 IP 的连接数是否超过了最大限制。如果超过了最大限制，我们就返回 503 状态码。如果请求被接受，则在最后更新连接数。

注意，这里我们使用了 ConcurrentHashMap 来存储每个 IP 的连接数，并且使用了 try-finally 块确保在任何情况下都能准确更新连接数。

以上是三种常见的限制来自 IP 的连接数的方法，开发者可以在具体情况下灵活使用。