18

我有一个基本的 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()

我希望能够处理多个连接（它似乎已经这样做了），但也能够引用其他连接。我看不到一种识别和跟踪单个连接的方法，只是为了能够处理连接打开、接收消息和连接关闭的事件。

[编辑]
考虑创建一个字典，其中键是 Sec-websocket-key，值是 WSHandler 对象......想法？我不确定 Sec-websocket-key 的唯一性有多可靠。

4

4 回答 4

24

最简单的方法就是保留 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'

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

于 2012-07-28T00:25:00.543 回答
19

当您只需要所有连接的列表时，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'
于 2012-07-29T07:01:02.017 回答
1

如果您的用户有访问令牌，则可以将其附加到您的 websocket 端点，并在初始化您的套接字时获取它，甚至在它打开之前（但请通过 SSL 工作）。

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

无论您如何执行此操作，访问令牌都应与将拥有标识符的用户相关联，并且该标识符甚至可以在套接字打开之前绑定到套接字。标识符可以用作字典键，其值是与该用户绑定的一组套接字。

于 2013-07-22T01:04:18.127 回答
0

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

clients = {}

class WSHandler(tornado.websocket.WebSocketHandler):


    def check_origin(self, origin): 

        clients[origin] = self
        print(clients)
        return True
于 2015-10-28T19:20:42.807 回答