feat: Add Middleware for ASGI applications

Author: amisadmin

sqlalchemy_database/_abc_async_database.py

@@ -27,3 +27,19 @@ def __init__(self) -> None:
             if not asyncio.iscoroutinefunction(func):
                 func = functools.partial(to_thread, func)
             setattr(self, f"async_{func_name}", func)
+
+    async def asgi_dispatch(self, request, call_next):
+        """Middleware for ASGI applications, such as: Starlette, FastAPI, Quart, Sanic, Hug, Responder, etc.
+        Bind a SQLAlchemy session connection to the incoming HTTP request session context,
+        you can access the session object through `self.session`.
+        The instance shortcut method will also try to use this `session` object by default.
+        Example:
+            ```Python
+            app = FastAPI()
+            db = Database.create("sqlite:///test.db")
+            app.add_middleware(BaseHTTPMiddleware, db.asgi_dispatch)
+            ```
+        """
+        async with self:
+            response = await call_next(request)
+        return response

sqlalchemy_database/_abc_async_database.pyi

@@ -91,3 +91,4 @@ class AbcAsyncDatabase(metaclass=abc.ABCMeta):
         executor: Union[Session, Connection, AsyncSession, AsyncConnection, None] = None,
         **kwargs: _P.kwargs,
     ) -> Union[_T, _R]: ...
+    async def asgi_dispatch(self, request, call_next): ...