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

Question

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

我在做什么:

发布请求处理程序 --> websocket msg 1 --> websocket msg2 --> celery --> 发布请求的响应

我尝试在 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()

长时间运行的post方法:

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

网页中的预期 WebSockets 会立即从 flask 后端接收消息，但在 post 请求完成后接收消息。不会生成错误消息。

Answer 1

在搜索了很多文章之后，我才知道这个错误是由于 NOT monkey patching 我的程序造成的。我应该仔细看看 Gevent 本身的文档。此链接将帮助遇到相同问题的人:GeventDoc .

此外，请记住猴子补丁应该是您代码中的第一件事。 在任何其他导入之前。