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

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'