Question

我有一个基本的Tornado websocket测试：

import tornado.httpserver
import tornado.websocket
import tornado.ioloop
import tornado.web

class WSHandler(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'/ws', WSHandler),
])


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

我希望能够处理多个连接（它似乎已经存在），但也能够引用其他连接。我没有看到识别和跟踪各个连接的方法，只是为了能够处理连接打开，消息接收和连接关闭的事件。

[编辑]
想到创建一个dict，其中键是Sec-websocket-key，而值是WSHandler对象......想法？我不确定Sec-websocket-key的独特性是多么可靠。

Answer 1

最简单的方法就是保留WSHandler实例的列表或字典：

class WSHandler(tornado.websocket.WebSocketHandler):
    clients = []

    def open(self):
        self.clients.append(self)
        print 'new connection'
        self.write_message("Hello World")

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

    def on_close(self):
        self.clients.remove(self)
        print 'closed connection'

如果您想识别连接，例如按用户，您可能必须通过套接字发送该信息。

Answer 2

当您需要所有连接列表时，Cole Maclean asnwer是一个很好的简单解决方案。但是，如果你想要更复杂的东西，可以在WSHandler实例之外进行监控 - 勇敢地这样做：

class WSHandler(tornado.websocket.WebSocketHandler):

    def open(self):
        self.id = uuid.uuid4()
        external_storage[self.id] = {'id':self.id}
        print 'new connection'
        self.write_message("Hello World")

    def on_message(self, message):
        #Some message parsing here
        if message.type == 'set_group':
           external_storage[self.id]['group'] = message.group
        print 'message received %s' % message

    def on_close(self):
        external_storage.remove(self.id)
        print 'closed connection'

Answer 3

如果您的用户拥有访问令牌，可以将其添加到您的websocket端点，比如说，并在启动套接字时将其设置为初始化（但请通过SSL工作）。

如果访问令牌不可用，或者因为用户未提供访问令牌或者他们提供的令牌已过期，则用户未经过身份验证，您将尽早终止该套接字。

但是，执行此操作时，访问令牌应与具有标识符的用户相关联，然后该标识符即使在打开之前也可以绑定到套接字。标识符可以作为字典键，其值是与该用户绑定的一组套接字。

Answer 4

我通过检查conexion的来源解决了这个问题。因此，覆盖方法def check_origin(self, origin)可能会有所帮助。例如：

clients = {}

class WSHandler(tornado.websocket.WebSocketHandler):


    def check_origin(self, origin): 

        clients[origin] = self
        print(clients)
        return True

龙卷风：识别/跟踪websockets的连接？

4 个答案: