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

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

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


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

有什么办法可以做到吗？例如，最多3个客户端。

最佳答案

最简单的解决方案是对 Activity 连接进行计数，并在达到限制时停止提供新连接。缩小的尺寸是接受新的TCP连接，然后将其关闭。希望没事。

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

需要一个变量:

accepted_sockets = weakref.WeakSet()

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

一个简单的辅助函数:

def count_connections(self):
     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等)，但希望能有所帮助。