使用 python socketserver 如何将变量传递给处理程序类的构造函数

Question

我想将数据库连接传递给 EchoHandler 类，但是我根本不知道如何执行此操作或访问 EchoHandler 类。

class EchoHandler(SocketServer.StreamRequestHandler):
    def handle(self):
        print self.client_address, 'connected'

if __name__ == '__main__':
    conn = MySQLdb.connect (host = "10.0.0.5", user = "user", passwd = "pass", db = "database")

    SocketServer.ForkingTCPServer.allow_reuse_address = 1

    server = SocketServer.ForkingTCPServer(('10.0.0.6', 4242), EchoHandler)

    print "Server listening on localhost:4242..."
    try:
        server.allow_reuse_address
        server.serve_forever()
    except KeyboardInterrupt:
        print "\nbailing..."

Answer 1

不幸的是，确实没有一种简单的方法可以直接从服务器外部访问处理程序。

您有两种选择来获取 EchoHandler 实例的信息：

1. 将连接存储为服务器的属性（在调用 server_forever() 之前添加 server.conn = conn）以及然后通过 self.server.conn 访问 EchoHandler.handler 中的该属性。
2. 您可以覆盖服务器的finish_request并在那里分配值（您必须将其传递给EchoHandler的构造函数并覆盖EchoHandler.__init__）。这是一个更加混乱的解决方案，而且它几乎要求您将连接存储在服务器上。

我的最佳选择是：

class EchoHandler(SocketServer.StreamRequestHandler):
    def handle(self):
        # I have no idea why you would print this but this is an example
        print( self.server.conn );
        print self.client_address, 'connected'

if __name__ == '__main__':
    SocketServer.ForkingTCPServer.allow_reuse_address = 1

    server = SocketServer.ForkingTCPServer(('10.0.0.6', 4242), EchoHandler)
    server.conn = MySQLdb.connect (host = "10.0.0.5", 
                     user = "user", passwd = "pass", db = "database")
    # continue as normal

Answer 2

Mark T 在 python list archive 上有如下说法

在 handler 类中， self.server 指的是服务器对象，所以 subclass
服务器并覆盖 init 以获取任何其他服务器参数
并将它们存储为实例变量。

import SocketServer

class MyServer(SocketServer.ThreadingTCPServer):

    def __init__(self, server_address, RequestHandlerClass, arg1, arg2):
        SocketServer.ThreadingTCPServer.__init__(self, 
                                                 server_address, 
                                                 RequestHandlerClass)
        self.arg1 = arg1
        self.arg2 = arg2


class MyHandler(SocketServer.StreamRequestHandler):

    def handle(self):
        print self.server.arg1
        print self.server.arg2

Answer 3

我认为更Pythonic的另一种方法是执行以下操作：

class EchoHandler(SocketServer.StreamRequestHandler):
  def __init__(self, a, b):
    self.a = a
    self.b = b

  def __call__(self, request, client_address, server):
    h = EchoHandler(self.a, self.b)
    SocketServer.StreamRequestHandler.__init__(h, request, client_address, server)

您现在可以将处理程序的实例提供给TCPServer：

SocketServer.ForkingTCPServer(('10.0.0.6', 4242), EchoHandler("aaa", "bbb"))

TCPServer通常会创建EchoHandler的新实例> 每个请求，但在这种情况下，将调用 __call__ 方法而不是构造函数（它已经是一个实例。）

在 call 方法中，我显式创建了一个副本当前 EchoHandler 并将其传递给超级构造函数符合“每个请求一个处理程序实例”的原始逻辑。

值得查看 SocketServer 模块以了解此处发生的情况： https://github.com/python/cpython/blob/2.7/Lib/SocketServer.py

Answer 4

我目前正在解决同样的问题，但我使用了略有不同的解决方案，我觉得它稍微好一点，更通用（受到@aramaki的启发）。

在 EchoHandler 中，您只需覆盖 __init__ 并指定自定义 Creator 方法。

class EchoHandler(SocketServer.StreamRequestHandler):
    def __init__(self, request, client_address, server, a, b):
        self.a = a
        self.b = b
        # super().__init__() must be called at the end
        # because it's immediately calling handle method
        super().__init__(request, client_address, server)

    @classmethod
    def Creator(cls, *args, **kwargs):
        def _HandlerCreator(request, client_address, server):
            cls(request, client_address, server, *args, **kwargs)
        return _HandlerCreator

然后您可以调用 Creator 方法并传递您需要的任何内容。

SocketServer.ForkingTCPServer(('10.0.0.6', 4242), EchoHandler.Creator(0, "foo"))

主要好处是，通过这种方式，您不会创建不必要的实例，并且可以以更易于管理的方式扩展类 - 您不需要再次更改 Creator 方法。

Answer 5

似乎您无法使用 ForkingServer 来共享变量，因为当进程尝试修改共享变量时会发生写入时复制。
将其更改为 ThreadingServer，您将能够共享全局变量。