python - 如何设置SocketServer的最大连接数

Question

我正在尝试找到一种方法来设置限制 - SocketServer 中的最大连接数。

class ThreadedTCPServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer):
    daemon_threads = True


class ThreadedTCPRequestHandler(SocketServer.BaseRequestHandler):
    def handle(self):
        some code 

有什么方法可以做到这一点吗？例如最多 3 个客户端。

Answer 1

最简单的解决方案是计算事件连接数，并在达到限制时停止为新连接提供服务。缩小规模是新的 TCP 连接被接受然后关闭。希望没问题。

我在不基于SocketServer的服务器中使用的代码基本上如下所示:

需要一个变量:

accepted_sockets = weakref.WeakSet()

WeakSet 的优点是不再使用的元素会被自动删除。

一个简单的辅助函数:

def count_connections():
     return sum(1 for sock in accepted_sockets if sock.fileno() >= 0)

请求处理代码中的用法很简单:

   if count_connections() >= MAX_CONN:
        _logger.warning("Too many simultaneous connections")
        sock.close()
        return
    accepted_sockets.add(sock)
    # handle the request

您必须对其进行调整以与SocketServer一起使用(例如，TCP套接字在那里被称为request，可能是服务器处理socket.close等)，但是希望有帮助。