利用flask-socketio实现前后端实时通信的功能

今天上上签

已于 2022-11-10 15:25:09 修改

阅读量3.5k

点赞数 8

分类专栏：前端项目实战文章标签： flask 后端前端 websocket

于 2022-11-10 15:16:46 首次发布

本文链接：https://blog.csdn.net/bradym/article/details/127787708

版权

前端项目实战专栏收录该内容

11 篇文章 4 订阅

订阅专栏

文章目录

一、需求
二、解决方法
三、具体实现
四、注意事项

一、需求

准备实现一个前后端实时发送消息的功能，即后端发送一条消息，前端就可以接收到该消息并显示在页面上。

二、解决方法

利用Flask-SocketIO实现，Flask-SocketIO使Flask应用程序能够访问客户端和服务器之间的低延迟双向通信。能够很好的解决我们的问题

三、具体实现

步骤一:安装

pip install eventlet
pip install flask-socketio

注意：flask-socketio对版本要求比较严格，如果报错，请更换如下包的版本：

Flask 1.1.4
Flask-SocketIO 4.3.1
python-engineio 3.13.2
python-socketio 4.6.0

步骤二：服务器代码

import time
from flask import Flask, request, render_template
from flask_socketio import SocketIO
from threading import Lock

thread = None
thread_lock = Lock()
app = Flask(__name__)
socketio = SocketIO()
socketio.init_app(app, cors_allowed_origins='*') # 用socketio初始化flask的app

flag1 = 0
flag2 = 0
flag3 = 0


@app.route('/', methods=['POST', 'GET'])
def index():
    global flag1
    global flag2
    global flag3
    if request.method == 'POST':
        # 模拟执行你的业务逻辑一
        time.sleep(5)
        # 利用全局变量表明业务逻辑一已完成
        flag1 = 1
        print("步骤1完成")
        # 模拟执行你的业务逻辑二
        time.sleep(2)
        # 利用全局变量表明业务逻辑二已完成
        flag2 = 1
        print("步骤2完成")
        # 模拟执行你的业务逻辑三
        time.sleep(3)
        # 利用全局变量表明业务逻辑三已完成
        flag3 = 1
        print("步骤3完成")

    return render_template('t1.html')


@socketio.on('connect', namespace='/test_conn')
def test_connect():
    """
    此函数在建立socket连接时被调用
    """
    print("socket 建立连接")
    global thread
    with thread_lock:
        print(thread)
        if thread is None:
            # 如果socket连接，则开启一个线程，专门给前端发送消息
            thread = socketio.start_background_task(target=background_thread)


@socketio.on('disconnect', namespace='/test_conn')
def disconnect_msg():
    """
    此函数在socket断开时被调用
    """
    print('client disconnected!')


def background_thread():
    """
    该线程专门用来给前端发送消息
    :return:
    """
    global flag1
    global flag2
    global flag3
    while True:
        if flag1 == 1:
            print("进入判断一")
            socketio.emit('server_response',
                          {'data': '步骤一已完成'}, namespace='/test_conn')
            flag1 = 2
        socketio.sleep(2)
        if flag2 == 1 and flag1 == 2:
            print("进入判断二")
            socketio.emit('server_response',
                          {'data': '步骤二已完成'}, namespace='/test_conn')
            flag2 = 2
        socketio.sleep(2)
        if flag3 == 1 and flag2 == 2 and flag1 == 2:
            print("进入判断三")
            socketio.emit('server_response',
                          {'data': '步骤三已完成'}, namespace='/test_conn')
            flag3 = 2
        socketio.sleep(2)


if __name__ == '__main__':
    socketio.run(app, host='127.0.0.1', port=5000, debug=True)

步骤三：前端代码

前端代码是用vue+element ui编写。(代码中所涉及的js文件可以在百度上自行下载，有很多)

<!DOCTYPE html>
<html>

<head>
    <meta charset="UTF-8" />
    <meta name="viewport" content="width=device-width,initial-scale=1.0" />

    <link rel="stylesheet" href="static/css/index.css"> <!-- 引入element的css文件-->
    <script src="static/js/vue.js"></script> <!-- 引入vue文件-->
    <script src="static/js/index.js"></script> <!-- 引入element的js文件-->
    <script src="static/js/axios.min.js"></script> <!--引入axios文件--> 
    <script type="text/javascript" src="static/js/jquery-1.4.2.min.js"></script> <!--引入jquery文件--> 
    <script type="text/javascript" src="static/js/socket.io.min.js"></script> <!--引入socket文件--> 
</head>

<body>
    <div id="test">
        <el-button type="primary" size="small" @click="onSubmit()">提交</el-button>
        <div id = "t"></div>
    </div>

</body>

<script>
	// 新建vue环境
    new Vue({
        el: '#test',
        data() {return{} },
        methods: {
            onSubmit() {
            	// 向后端发送post请求
                axios.post('http://127.0.0.1:5000/');
                // 建立socket连接，此时会触发后端的@socketio.on('connect', namespace='/test_conn')路由，执行test_connect函数
                let socket = io.connect('http://127.0.0.1:5000/test_conn');
                socket.on('server_response', function (res) {
                	//接收到后端发送过来的消息
                    var t = res.data;
                    console.log(t)
                    if (t) {
                    	// 显示在页面上
                        $("#t").append(t).append('<br/>');
                    }
                });
            }
        }
    })

</script>

</html>

结果展示

启动flask项目，并访问：
在这里插入图片描述
点击提交后，前端发送post请求给后端，并建立socket连接：

同时执行index()函数：

同时前端页面接收到后端发送的消息：

至此，前后端交互功能完成！

四、注意事项

1、我们不能直接在socket连接函数中写一个while循环去发送信息，例如这样：

@socketio.on('connect', namespace='/test_conn')
def test_connect():
    while True:
        socketio.sleep(5)
        t = random.randint(1, 100)
        socketio.emit('server_response',
                      {'data': t},namespace='/test_conn')

虽然这样写，emit函数每次都会执行，理论上客户端应该可以定时收到服务端的随机数。但是结果是客户端根本接收不到，连soketio.on函数都没有触发运行。
原因应该是当服务端陷入死循环，会影响与客户端之间的websocket连接，总之写while true需谨慎。
所以，在flask_socketio的示例程序中，我们 用后台线程 进行while循环以解决这个问题。

2、websocket连接的一个典型应用是实现一个网络聊天室，所以就会有群发和私聊的功能，我们此处演示的示例，是一个群发功能，也就是我们并没有指定socket的room空间，所以当我们从后端发送给前端消息时，所有打开此页面的人，都可以看到我们发送的消息，也就是一个群发功能。