如何计算 TCPServer 中连接的客户端？答案

【问题标题】：How to count connected clients in TCPServer?如何计算 TCPServer 中连接的客户端？
【发布时间】：2011-03-20 19:12:21
【问题描述】：

我正在使用 Pythons SocketServer.ThreadingTCPServer。现在我想知道某个时刻有多少客户端连接。

如何解决？

【问题讨论】：

每次接受连接时计数器加一，结束时减一？

标签： python tcp network-programming

【解决方案1】：

SocketServer.ThreadingTCPServer 为每个客户端连接交换一个新线程，因此知道某个时刻连接的客户端数量与知道在那一刻有多少线程处于活动状态相同，因此只需使用threading.activeCount 和客户端数量将是：

num_client = threading.activeCount() - 1 # Don't count the main thread.

当然，如果您的代码的其他部分也交换线程，这不会给您正确的结果，因此要解决此问题，您可以通过为客户端添加一个计数器来覆盖 process_request() 和 process_request_thread() 方法。

从例子here我写了这个sn-p的代码来测试这两种方法

import time
import socket
import threading
from SocketServer import ThreadingTCPServer, BaseRequestHandler


def client(ip, port, message):
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.connect((ip, port))
    sock.send(message)
    response = sock.recv(1024)
    sock.close()


class ThreadedTCPRequestHandler(BaseRequestHandler):

    def handle(self):
        data = self.request.recv(1024)
        cur_thread = threading.currentThread()
        response = "%s: %s" % (cur_thread.getName(), data)
        self.request.send(response)
        time.sleep(1)


class MyServer(ThreadingTCPServer):

    def __init__(self, *args, **kws):
        self._num_client = 0
        ThreadingTCPServer.__init__(self, *args, **kws)

    def process_request(self, *args, **kws):
        print "swap thread"
        self._num_client += 1
        ThreadingTCPServer.process_request(self, *args, **kws)

    def process_request_thread(self, *args, **kws):
        ThreadingTCPServer.process_request_thread(self, *args, **kws)
        print "kill thread"
        self._num_client -= 1

    def get_client_number(self):
        return self._num_client


def my_client_count(ignore=1):
    return  threading.activeCount() - ignore


if __name__ == '__main__':
    server = MyServer(("localhost", 0), ThreadedTCPRequestHandler)

    server_thread = threading.Thread(target=server.serve_forever)
    ip, port = server.server_address

    server_thread.setDaemon(True)
    server_thread.start()

    print "client 1 connected"
    client(ip, port, "Hello World 1")
    print "number of client get_client_number : %s,  enumerate : %s" \
       %  (server.get_client_number(), my_client_count())
    print "client 2 connected"
    client(ip, port, "Hello World 2")
    print "number of client get_client_number : %s,  enumerate : %s" \
       %  (server.get_client_number(), my_client_count())

    time.sleep(3)
    print "client 3 connected"
    client(ip, port, "Hello World 3")
    print "number of client get_client_number : %s,  enumerate : %s" \
       %  (server.get_client_number(), my_client_count())

输出：

client 1 connected
swap client thread
number of client get_client_number : 1,  enumerate : 2
client 2 connected
swap client thread
number of client get_client_number : 2,  enumerate : 3
kill client thread
kill client thread
client 3 connected
swap client thread
number of client get_client_number : 1,  enumerate : 2

正如您所见，第二种方式给出了更精确的值，两种方式之间的区别在于，因为我的服务器使用线程运行，所以我总是 +1 线程来解释 1 的差异。

希望这可以帮助:)

【讨论】：

【解决方案2】：

在“服务”客户端的线程中，使用一些全局计数，当客户端连接时增加，断开连接时减少。

如果您想从操作系统级别计数，请使用 nestat -an 和适当的 grep 过滤器和 wc -l（在 Windows 上使用 grep 和 wc 的端口）

【讨论】：