Merge pull request #311 from plangrid/jonls/async-handler

Async handler support

examples/todo/todo/handlers/todo_handlers.py

@@ -1,4 +1,4 @@
-from enum import Enum
+import asyncio
 
 from todo.database import todo_id_sequence, todo_database
 from todo.app import rebar

flask_rebar/rebar.py

@@ -176,7 +176,7 @@ def wrapped(*args: P.args, **kwargs: P.kwargs) -> Union[T, Response]:
         if headers_schema:
             g.validated_headers = get_header_params_or_400(schema=headers_schema)
 
-        rv: Any = f(*args, **kwargs)
+        rv: Any = current_app.ensure_sync(f)(*args, **kwargs)
 
         if not response_body_schema:
             return rv

setup.py

@@ -20,6 +20,7 @@
     "sphinx_rtd_theme==1.2.2",
     "types-jsonschema==4.17.0.10",
     "types-setuptools==68.0.0.3",
+    "flask[async]>=2,<4",
 ]
 
 install_requires = [
@@ -42,7 +43,11 @@
         packages=find_packages(exclude=("test*", "examples")),
         package_data={"flask_rebar": ["py.typed"]},
         include_package_data=True,
-        extras_require={"dev": development, "enum": ["marshmallow-enum~=1.5"]},
+        extras_require={
+            "dev": development,
+            "enum": ["marshmallow-enum~=1.5"],
+            "async": ["flask[async]>=2,<4"],
+        },
         install_requires=install_requires,
         url="https://github.com/plangrid/flask-rebar",
         classifiers=[

tests/test_rebar.py

@@ -7,6 +7,7 @@
     :copyright: Copyright 2018 PlanGrid, Inc., see AUTHORS.
     :license: MIT, see LICENSE for details.
 """
+import asyncio
 import json
 import unittest
 
@@ -864,3 +865,20 @@ def post_foo(*args, **kwargs):
         self.assertIsNotNone(swagger)  # really only care that it didn't barf
         swagger = SwaggerV3Generator().generate(registry)
         self.assertIsNotNone(swagger)
+
+    def test_async_handler(self):
+        rebar = Rebar()
+        registry = rebar.create_handler_registry()
+
+        @registry.handles(
+            rule="/async",
+            method="GET",
+            response_body_schema=DefaultResponseSchema(),
+        )
+        async def get_response():
+            await asyncio.sleep(0)
+            return DEFAULT_RESPONSE
+
+        app = create_rebar_app(rebar)
+        resp = app.test_client().get(path="/async")
+        self.assertEqual(resp.status_code, 200)