在Flask中如何实现WebSocket通信

发布时间：2024-04-30 09:57:14 作者：小樊
来源：亿速云阅读：265

在Flask中实现WebSocket通信通常需要使用WebSocket扩展。常用的是Flask-SocketIO和Flask-SocketIO-Client库。

以下是一个简单的示例代码，演示了如何在Flask应用中实现WebSocket通信：

from flask import Flask, render_template
from flask_socketio import SocketIO

app = Flask(__name__)
socketio = SocketIO(app)

@app.route('/')
def index():
    return render_template('index.html')

@socketio.on('message')
def handle_message(message):
    print('received message: ' + message)
    socketio.send(message)

if __name__ == '__main__':
    socketio.run(app)

在上面的示例中，我们首先初始化了一个Flask应用和一个SocketIO实例。然后定义了一个路由’/‘，并在该路由中渲染了一个简单的HTML模板。接着我们定义了一个事件处理器handle_message，用于处理客户端发送过来的消息，并发送响应消息。

在客户端，我们可以使用JavaScript的SocketIO客户端库来连接WebSocket服务，并发送接收消息。以下是一个简单的HTML模板示例：

<!DOCTYPE html>
<html>
<head>
    <title>WebSocket Example</title>
    <script src="https://cdn.socket.io/socket.io-4.0.1.min.js"></script>
</head>
<body>
    <script>
        var socket = io();
        socket.on('connect', function() {
            console.log('connected');
            socket.send('Hello, server!');
        });
        
        socket.on('message', function(data) {
            console.log('received message: ' + data);
        });
    </script>
</body>
</html>

在上面的示例中，我们引入了SocketIO的JavaScript客户端库，并通过io()方法初始化了一个SocketIO实例。然后我们监听了connect事件，连接成功后向服务端发送了一个消息，并监听了message事件，用于处理服务端发送过来的消息。

通过以上代码示例，可以实现WebSocket通信的功能。在实际应用中，你可以根据需求扩展功能，如实现实时聊天、实时数据更新等。

在Flask中如何实现WebSocket通信

相关阅读