龙卷风: 确定 / 跟踪网络套接字的连接?

Question

我有一个基本的龙卷风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键，值是WSHandler对象。。。思想？我不确定Sec websocket密钥的唯一性有多可靠。

Answer 1

当您只需要列出所有连接时，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 2

最简单的方法是保存WSHandler实例的列表或dict：

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 3

如果你的用户有一个访问令牌，它可以附加到你的websocket端点上，比如说，甚至在你的套接字打开之前就初始化它（但是请使用SSL）。

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

但是，这样做时，访问令牌应该与具有标识符的用户相关联，然后甚至在套接字打开之前，该标识符就可以绑定到套接字。标识符可以用作字典键，其值是绑定到此用户的一组套接字。