在 Flask 中返回立即响应，但在新线程中完成处理

我有一个从 10 秒开始的过程，但是在 10 秒之后，我不需要将信息返回给用户，而是将其写在 dynamoDB 上，这就是为什么我希望用户不必等待 10 秒。相反，我希望在发布请求后立即得到“成功”响应。

我读了几篇文章并在this https://stackoverflow.com/questions/17293311/flask-end-response-and-continue-processing一，答案是使用 Teardown Callback，但没有示例。

我当时读到this https://stackoverflow.com/questions/30521112/how-teardown-request-works-with-python-flask，但这并不能帮助我解决我的问题。

我当然读过拆卸回调 http://flask.pocoo.org/docs/0.12/reqcontext/#teardown-callbacks and 这个图案 http://flask.pocoo.org/snippets/53/但我不知道如何以其他方式使用它。

我的代码如下所示：

@app.route('/ocr/read_image', methods=['POST'])
def get_text():    
    return jsonify('Success')

@app.teardown_request
def teardown_request(response):        
    time.sleep(10)

它实际上返回“成功”消息，但就在 10 秒之后。

有没有办法在 10 秒之前返回“成功”消息？

我一直在读到芹菜也许是可能的，但如果可以的话我很想避免它。

有谁知道该怎么做？

恕我直言，线程在这里也使事情变得不必要的复杂化。要在不引入线程的情况下完成您想要做的事情，您可以挂接到WSGI响应关闭方法 https://www.python.org/dev/peps/pep-0333/#specification-details。最简单的方法是使用 werkzeug关闭迭代器 http://werkzeug.pocoo.org/docs/0.14/wsgi/#werkzeug.wsgi.ClosingIterator helper.

import traceback
from werkzeug.wsgi import ClosingIterator

class AfterThisResponse:
    def __init__(self, app=None):
        self.callbacks = []
        if app:
            self.init_app(app)

    def __call__(self, callback):
        self.callbacks.append(callback)
        return callback

    def init_app(self, app):
        # install extensioe
        app.after_this_response = self

        # install middleware
        app.wsgi_app = AfterThisResponseMiddleware(app.wsgi_app, self)

    def flush(self):
        try:
            for fn in self.callbacks:
                try:
                    fn()
                except Exception:
                    traceback.print_exc()
        finally:
            self.callbacks = []

class AfterThisResponseMiddleware:
    def __init__(self, application, after_this_response_ext):
        self.application = application
        self.after_this_response_ext = after_this_response_ext

    def __call__(self, environ, start_response):
        iterator = self.application(environ, start_response)
        try:
            return ClosingIterator(iterator, [self.after_this_response_ext.flush])
        except Exception:
            traceback.print_exc()
            return iterator

然后您可以像这样使用这个扩展：

import flask
import time
app = flask.Flask("after_response")
AfterThisResponse(app)

@app.route("/")
def home():
    @app.after_this_response
    def post_process():
        time.sleep(2)
        print("after_response")

    return "Success!\n"

当您卷曲时，您将立即看到成功，然后 2 秒后在日志中您将看到“after_response”消息：

127.0.0.1 - - [25/Jun/2018 16:15:01] "GET / HTTP/1.1" 200 -
after_response

该解决方案是根据我的答案改编的摘要：

• Flask 中返回响应后需要执行一个函数 https://stackoverflow.com/questions/18082683/need-to-execute-a-function-after-returning-the-response-in-flask/51027637#51027637
• Flask结束响应并继续处理 https://stackoverflow.com/questions/17293311/flask-end-response-and-continue-processing/51027454#51027454
• Flask 返回响应后执行函数 https://stackoverflow.com/questions/48994440/execute-a-function-after-flask-returns-response/51013358#51013358