在 Tornado 中保留 websocket 连接列表 [重复]

Question

我想跟踪多个 websocket 连接，我必须将 WebSocket 处理程序对象存储在一个列表中。但我有多个处理程序 - 每个 WS URI（端点）一个。假设我的项目有三个端点 - A、B、C

ws://www.example.com/A
ws://www.example.com/B
ws://www.example.com/C

因此，为了处理每一个的连接，我有三个处理程序。所以我很困惑在哪里创建列表来存储稍后使用的处理程序对象。

我在添加列表之前的代码是--

class WSHandlerA(tornado.websocket.WebSocketHandler):
    def open(self):
        print 'new connection'
        self.write_message("Hello World")

    def on_message(self, message):
        print 'message received %s' % message

    def on_close(self):
        print 'connection closed'


class WSHandlerB(tornado.websocket.WebSocketHandler):
    def open(self):
        print 'new connection'
        self.write_message("Hello World")

    def on_message(self, message):
        print 'message received %s' % message

    def on_close(self):
        print 'connection closed'

class WSHandlerC(tornado.websocket.WebSocketHandler):
    def open(self):
        print 'new connection'
        self.write_message("Hello World")

    def on_message(self, message):
        print 'message received %s' % message

    def on_close(self):
        print 'connection closed'

application = tornado.web.Application([
    (r'/A', WSHandlerA),
    (r'/B', WSHandlerB),
    (r'/C', WSHandlerC),
])


if __name__ == "__main__":
    http_server = tornado.httpserver.HTTPServer(application)
    http_server.listen(8888)
    tornado.ioloop.IOLoop.instance().start()

那么我在哪里创建这个列表并确保它对所有创建的对象都是可见的？我也是 python 新手，所以在理解它时遇到了一些麻烦。

我也意识到我可能只使用一个 URI（端点）并将多个命令作为消息本身的一部分发送。但是后来我不想把 WebSocket 变成二进制协议。鉴于我们有 URI，我应该使用它们。

感谢您的帮助。

Answer 1

这取决于你想用处理程序做什么，但你可以很容易地让它们在三个列表中访问：

# Global variables.
a_handlers = []
b_handlers = []
c_handlers = []

WSHandlerA.open() 执行a_handlers.append(self)，WSHandlerB.open() 执行b_handlers.append(self)，以此类推。 WSHandlerA.on_close() 是 a_handlers.remove(self)。

如果你想对所有 A 处理程序做某事：

for handler in a_handlers:
    handler.write_message("message on A")

对所有处理程序做某事：

for handler in a_handlers + b_handlers + c_handlers:
    # do something....
    pass

顺便说一句，如果您使用 Python set() 而不是每组处理程序的列表，它会好一点。对于集合而不是列表，请使用 add 和 discard 而不是 append 和 remove。