刷新客户端网页时如何重新连接到 Web Socket

Question

当我刷新客户端网页时，我停止从龙卷风套接字服务器接收数据。如何重新连接到流？

我尝试将连接对象附加到列表中，然后在关闭时从列表中删除连接，但事实证明，当您刷新页面时，连接永远不会关闭，它根据服务器保持活动状态，但它客户端也不再接收数据：(

这是我的龙卷风服务器

# python 3
from tornado import web, httpserver, ioloop, websocket, options
from time import time, sleep


class ChannelHandler(websocket.WebSocketHandler):
    """Handler that handles a websocket channel"""
    connections = list()

    @classmethod
    def urls(cls):
        return [(r'/websocket', cls, {})]

    def initialize(self):
        self.channel = None

    def open(self):
        # When Client opens a websocket
        # add the new connnection to connections
        self.connections.append(self)

    def on_message(self, message):
        # Message received on channel
        # keep sending all connected clients the time info
        while True:
            [client.write_message({'time()': str(time())}) for client in self.connections]
            sleep(1)
            print('still sending')

    def on_close(self):
        # Channel is closed
        # delete client from active connections if they close connection
        self.connections.remove(self)
        print('CLOSED connection?')

    def check_origin(self, origin):
        # Override the origin check if needed
        return True


def main():
    # Create tornado application and supply URL routes
    app = web.Application(ChannelHandler.urls())
    # Setup HTTP Server
    http_server = httpserver.HTTPServer(app)
    http_server.listen(8000, 'localhost')
    # Start IO/Event loop
    ioloop.IOLoop.instance().start()


if __name__ == '__main__':
    main()

而socket客户端是

<script type="text/javascript">
    var ws = new WebSocket("ws://localhost:8000/websocket");
    ws.onopen = function () {
        ws.send("Hello, world");
    };
    ws.onmessage = function (evt) {
        console.log(evt.data);
    };
</script>

那么，当我刷新客户端网页时，如何继续从套接字服务器接收数据？

Answer 1

while 循环阻塞了整个服务器。您在循环中添加了一个 sleep(1) 调用以暂停循环一秒钟，但 time.sleep 是一个阻塞函数，因此也没有任何帮助。

您需要使用time.sleep 的异步替代方案——Tornado 的gen.sleep。

您还需要将on_message 函数转换为协程。

例子：

from tornado import gen


async def on_message(...):
    ...
    while True:
        ...
        await gen.sleep(1)
        ...

---

作为附加提示，请使用 set() 而不是 list() 来存储您的连接，以免意外添加重复的连接。

在这种情况下，您还需要稍微修改您的代码。 set 没有 append 方法，而是有一个 add 方法。

connections = set()
...
self.connections.add(self) # to add connection
...
self.connections.remove(self) # to remove connetion