首页 文章

Flask应用程序中的asyncio event_loop

提问于
浏览 次
3

在Flask应用程序中运行asyncio事件循环的最佳方法是什么？

我的main.py看起来像这样：

if __name__ == '__main__':
    try:
        app.run(host='0.0.0.0', port=8000, debug=True)
    except:
        logging.critical('server: CRASHED: Got exception on main handler')
        logging.critical(traceback.format_exc())
        raise

要添加异步任务选项，我需要在运行 app 之前创建 event_loop ，但即使停止应用程序运行，后台线程仍会挂起（在调试器中可观察）

if __name__ == '__main__':
    try:
        app.event_loop = asyncio.get_event_loop()
        app.run(host='0.0.0.0', port=8000, debug=True)
    except:
        logging.critical('server: CRASHED: Got exception on main handler')
        logging.critical(traceback.format_exc())
        raise
    finally:
        app.event_loop.stop()
        app.event_loop.run_until_complete(app.event_loop.shutdown_asyncgens())
        app.event_loop.close()

并使用以下命令创建异步任务：

def future_callback(fut):
    if fut.exception():
        logging.error(fut.exception())

def fire_and_forget(func, *args, **kwargs):
    if callable(func):
        future = app.event_loop.run_in_executor(None, func, *args, **kwargs)
        future.add_done_callback(future_callback)
    else:
        raise TypeError('Task must be a callable')

我能找到的唯一解决方案是在 finally 块的末尾添加 exit() ，但我不认为它是正确的解决方案 .

python flask python-3.6 python-asyncio event-loop

2 回答

  • 0

    我最初为整个烧瓶应用程序采用了1 event_loop的方法，但由于几个不同的原因不喜欢它 .

    相反，我创建了一个辅助文件 custom_flask_async.py ，其中包含2个函数，

    import asyncio


def get_set_event_loop():
    try:
        return asyncio.get_event_loop()
    except RuntimeError as e:
        if e.args[0].startswith('There is no current event loop'):
            asyncio.set_event_loop(asyncio.new_event_loop())
            return asyncio.get_event_loop()
        raise e


def run_until_complete(tasks):
    return get_set_event_loop().run_until_complete(asyncio.gather(*tasks))

    我只检查和 get_set event_loop如果's going to be used, and waited upon in a particular request. And because the main way I'使用这个是带有聚集的 run_until_complete ，我也在帮助器中 .

    我也不确定这是不是最好的方法，或者这个方法有什么缺点，但它绝对适用于我的目的 .

    回复于
  • 1

    最终我找到的最佳解决方案是在 uwsgi 中托管我的Flask应用程序，它可以使用 mules 和 spooler 进行异步任务

    https://uwsgi-docs.readthedocs.io/en/latest/Spooler.html

    https://uwsgi-docs.readthedocs.io/en/latest/Mules.html

    回复于

相关问题