【代码】flask对asyncio的支持

最新推荐文章于 2025-04-15 21:58:22 发布

原创最新推荐文章于 2025-04-15 21:58:22 发布 · 3.9k 阅读

2 ·

CC 4.0 BY-SA版权

本文探讨了如何在 Flask 应用中集成和利用 asyncio，以实现异步处理，提升 Web 服务的性能和响应速度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from time import sleep

from flask import Flask, jsonify, has_request_context, copy_current_request_context
from functools import wraps
from concurrent.futures import Future, ThreadPoolExecutor
import asyncio


def run_async(func):
    @wraps(func)
    def _wrapper(*args, **kwargs):
        call_result = Future()

        def _run():
            loop = asyncio.new_event_loop()
            try:
                result = loop.run_until_complete(func(*args, **kwargs))
            except Exception as error:
                call_result.set_exception(error)
            else:
                call_result.set_result(result)
            finally:
                loop.close()

        loop_executor = ThreadPoolExecutor(max_workers=10)
        if has_request_context():
            _run = copy_current_request_context(_run)
        loop_future = loop_executor.submit(_run)
        loop_future.result()
        return call_result.result()

    return _wrapper


app = Flask(__name__)


@app.route('/async')
@run_async
async def index():
    await asyncio.sleep(0.2)
    return jsonify('hello')


@app.route('/sync')
def aindex():
    sleep(0.2)
    return jsonify('hello')


import logging

# logging.getLogger('werkzeug').setLevel(logging.ERROR)
app.run(port=8000)