使用TornadoWebSocket实现客户端与服务端双向通信

发布时间：2023-12-28 06:53:09

Tornado是一个Python的web框架，它可以用来构建高性能的、非阻塞的web应用程序。TornadoWebSocket是Tornado框架中用于实现双向通信的一种方式，它基于WebSocket协议，可以实现服务器与客户端之间的实时双向通信。

下面我们使用TornadoWebSocket来实现一个简单的聊天应用程序，演示客户端与服务端之间的双向通信。

首先，我们需要安装Tornado框架。可以使用pip进行安装：

pip install tornado

接下来，我们创建一个WebSocketHandler类来处理客户端和服务端之间的通信。这个类继承自tornado.websocket.WebSocketHandler，并重写了一些方法来处理不同的事件。

import tornado.websocket

class ChatWebSocketHandler(tornado.websocket.WebSocketHandler):
    # 存储所有已连接的客户端对象
    clients = []

    def open(self):
        # 当有新的客户端连接时，将客户端对象添加到clients列表中
        self.clients.append(self)

    def on_message(self, message):
        # 当接收到客户端发送的消息时，将消息转发给所有已连接的客户端
        for client in self.clients:
            client.write_message(message)

    def on_close(self):
        # 当有客户端断开连接时，将该客户端对象从clients列表中移除
        self.clients.remove(self)

现在我们可以创建一个Tornado应用程序，并将WebSocket处理器与一个路由相结合。

import tornado.web
import tornado.ioloop

app = tornado.web.Application([
    (r"/ws", ChatWebSocketHandler),
])

if __name__ == "__main__":
    app.listen(8888)
    tornado.ioloop.IOLoop.current().start()

在上述代码中，我们使用了8888端口来监听WebSocket连接。

接下来，我们创建一个简单的HTML文件作为聊天应用程序的前端界面。

<!DOCTYPE html>
<html>
<head>
    <title>Chat Application</title>
</head>
<body>
    <div id="chat">
        <input type="text" id="message" />
        <button onclick="send()">Send</button>
    </div>

    <script type="text/javascript">
        var ws = new WebSocket("ws://localhost:8888/ws");

        ws.onmessage = function(event) {
            // 当接收到服务端发送的消息时，将消息显示在聊天窗口中
            var message = document.createElement("p");
            message.innerHTML = event.data;
            document.getElementById("chat").appendChild(message);
        };

        function send() {
            // 当点击发送按钮时，将输入框中的消息发送给服务端
            var message = document.getElementById("message").value;
            ws.send(message);
        }
    </script>
</body>
</html>

在上述代码中，我们创建了一个WebSocket对象并与服务器建立连接。当接收到服务端发送的消息时，将消息显示在聊天窗口中；当点击发送按钮时，将输入框中的消息发送给服务端。

最后，我们可以运行应用程序，在浏览器中打开该HTML文件。当有多个客户端连接时，它们之间的消息可以实时地进行双向通信。

以上就是使用TornadoWebSocket实现客户端与服务端双向通信的一个简单例子。通过TornadoWebSocket，我们可以轻松地实现实时双向通信的功能，非常适合构建聊天室、实时数据展示等需要实时更新的应用程序。