Question

我正在开发一个非常简单的服务器。该服务器应该能够执行clients请求的一些预定义命令。

我想分别存储有关已连接客户端的信息。例如，我希望服务器添加一些特定的clients请求。

以下是一个例子：

SERVER
CLIENT1 
CLIENT2

CLIENT1> print 'stuff'
SERVER>> REQUESTS: 1 OUTPUT: stuff
CLIENT2> print 'simple sentence'
SERVER>> REQUESTS: 1 OUTPUT: simple sentence
CLIENT> print 'hilarious'
SERVER>> REQUESTS: 2 OUTPUT: hilarious

我的代码很简单：

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

class ThreadedTCPRequestHandler(SocketServer.BaseRequestHandler):
    def handle(self):
        data = self.request.recv(1024)
        command = data.split(' ')[0]
        arguments = data.split(' ')[1:]
        cur_thread = threading.current_thread()

        output = do_command(command,arguments)
        response = "{}: {}".format(cur_thread.name, output)
        self.request.sendall(response)


if __name__ == "__main__":
    commands.register_commands()
    HOST, PORT = _host, int(_port)

    server = ThreadedTCPServer((HOST, PORT), ThreadedTCPRequestHandler)
    ip, port = server.server_address

    # start server
    print "Running on: %s:%s" % (HOST, PORT)
    server.serve_forever()

所以我想知道的是如何存储有关每个客户的信息。我正在考虑创建一个类Client并在每个连接后创建一个object，但我不知道我应该在哪里创建这个对象。

Answer 1

您可以使用处理程序的client_address属性来标识客户端，并跟踪全局字典中的请求：

from collections import defaultdict

client_requests = defaultdict(int)

class ThreadedTCPRequestHandler(SocketServer.BaseRequestHandler):
    def handle(self):
        ...
        client_requests[self.client_address[0]] += 1
        response = "{} ({}): {}".format(cur_thread.name, client_requests[self.client_address[0]], output)

请注意，由于您使用的是线程服务器，因此您可能需要添加一些锁定代码来保护对client_requests的写入，但这是留给读者的练习。

SocketServer - 如何识别客户端

1 个答案: