python - 如何修复 Flask-uwsgi-Nginx 设置中的 Post 请求延迟的 websocket 消息？

问题描述

我正在开发一个 Flask 应用程序，该应用程序具有由 Flask 的 render_template 函数呈现的单个网页。此网页包含一个表单，用于通过 post 请求 + WebSocket 连接提交数据以获取任务状态。提交表单时，会触发长时间运行任务的发布请求。在这个长时间运行的任务的执行过程中，很少有 WebSocket 消息从烧瓶后端发送到这个网页作为任务的状态。然而，即使这些消息是从后端以正确的顺序发送的，它们也只有在长时间运行的任务完成时才会在网页中接收到（换句话说，当网页收到表单发布请求的响应时）。据我了解，WebSockets 是异步的。那么为什么会这样呢？

我在做什么：

Post request handler --> websocket msg 1 --> websocket msg2 --> celery --> post request的响应

我尝试在 app.ini 中添加进程、线程等。

uwsgi 的 app.ini

[uwsgi]
module = wsgi:app
master = true
http-socket = 127.0.0.1:7001
gevent = 10
http-websockets = true

nginx 配置

server {
    listen 7000;
    server_name _;  

    location / {
        include proxy_params;
    proxy_pass http://127.0.0.1:7001;
    uwsgi_buffering off;
    }

    location /socket.io {
    include proxy_params;
        proxy_http_version 1.1;
    uwsgi_buffering off;
        proxy_buffering off;
        proxy_set_header Upgrade $http_upgrade;
        proxy_set_header Connection "Upgrade";
        proxy_pass http://127.0.0.1:7001/socket.io;

    }
}

网页 websocket 代码

<script type="text/javascript" charset="utf-8">
        var socket = io.connect('http://' + document.domain + ':' + location.port);

         socket.on('connect', function() {
            socket.emit('get_status')
            socket.emit('get_message')
            console.log('Websocket connected!');
            });

        socket.on('response_status',function(data){
        document.getElementById("status").innerHTML = "Status : "+data.status;
        console.log("status received");
    });


        socket.on('response_last_message',function(data){
        document.getElementById("message").innerHTML = "Message : "+data.last_message;
        console.log("Message received");
    });


</script>

Python代码：

主班

socketio = SocketIO()

def create_app():
    app = Flask(__name__)
    api = Api(app)
    socketio.init_app(app)


    with open('config.json', 'r') as conf:
        configObject = json.load(conf)

    commonConfig = commons.Common(configObject)
    socketWork = commons.SocketWork(socketio, commonConfig)


    api.add_resource(routes.IndexPage,'/', resource_class_kwargs={ 'commonConfig': commonConfig, 'socketWork': socketWork})
    api.add_resource(routes.UploadFile, '/uploadFile', resource_class_kwargs={ 'commonConfig': commonConfig, 'socketWork': socketWork})


    return app

套接字类

class SocketWork():

    def __init__(self, socketio = None, commonObj=None):
        self.commonObj = commonObj
        self.socketio = socketio
        self.socketio.on_event('get_status',self.send_last_status)
        self.socketio.on_event('get_message',self.send_last_message)

    def send_last_status(self):
        print("Send status now")
        status = self.commonObj.status
        self.socketio.emit("response_status", {"status": status})

    def send_last_message(self):
        print("Send last message")
        msg = self.commonObj.msg
        self.socketio.emit("response_last_message", {"last_message": msg})

    def send_message(self,msg):
        self.commonObj.msg = msg
        self.send_last_message()

    def send_status(self,statusVal):
        self.commonObj.status = statusVal
        self.send_last_status()

长时间运行的帖子方法：

class UploadFile(Resource):

    def __init__(self, **kwargs):
        self.commonConfig = kwargs['commonConfig']
        self.socketWork    = kwargs['socketWork']

    @use_args(args.argsUploadData)
    def post(self, args):
        msg = "This is socket message"
        socketWork.send_message(msg)
        socketWork.send_status(commonConfig.config['Status']['READY'])
        sleep(15)       ###long running task
        return constants.generic_msg.PROCESSING_QUERY.name

网页中的预期 WebSocket 会立即从烧瓶后端接收消息，但在发布请求完成后接收消息。不会生成错误消息。

标签： pythonflaskwebsocketuwsgi

python - 如何修复 Flask-uwsgi-Nginx 设置中的 Post 请求延迟的 websocket 消息？

问题描述

解决方案

推荐阅读