Actualize docs a bit

Author: insani7y

docs/usage/frameworks/aiohttp.md

@@ -26,6 +26,12 @@ async def start_db_pool(app: web.Application) -> None:
     app["db_pool"] = db_pool
 
 
+async def stop_db_pool(app: web.Application) -> None:
+    """Close database connection pool."""
+    db_pool = cast(ConnectionPool, app.db_pool)
+    db_pool.close()
+
+
 async def pg_pool_example(request: web.Request):
     db_pool = cast(ConnectionPool, request.app["db_pool"])
     connection = await db_pool.connection()

docs/usage/frameworks/blacksheep.md

@@ -32,6 +32,17 @@ async def on_start(app: Application) -> None:
     app.services.add_instance(db_pool)
 
 
+@app.on_stop
+async def on_stop(app: Application) -> None:
+    """Close a database pool if it exists in app scope."""
+    try:
+        db_pool = app.services.resolve(ConnectionPool)
+    except Exception:
+        ...
+    else:
+        db_pool.close()
+
+
 @get("/")
 async def pg_pool_example(db_pool: ConnectionPool) -> list[dict[Any, Any]]:
     connection = await db_pool.connection()

docs/usage/frameworks/fastapi.md

@@ -30,6 +30,8 @@ async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
         max_db_pool_size=10,
     )
     app.state.db_pool = db_pool
+    yield
+    db_pool.close()
 
 
 app = FastAPI(lifespan=lifespan)
@@ -82,6 +84,8 @@ db_pool = ConnectionPool(
 async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
     """Startup database connection pool and close it on shutdown."""
     app.state.db_pool = db_pool
+    yield
+    db_pool.close()
 
 
 app = FastAPI(lifespan=lifespan)

docs/usage/frameworks/litestar.md

@@ -33,6 +33,13 @@ def start_db_pool(app: Litestar) -> ConnectionPool:
     return cast("ConnectionPool", app.state.db_pool)
 
 
+async def stop_db_pool(app: Litestar) -> None:
+    """Close database connection pool."""
+    if getattr(app.state, "db_pool", None):
+        db_pool = cast(ConnectionPool, app.state.db_pool)
+        db_pool.close()
+
+
 @get("/")
 async def pg_pool_example(request: Request) -> list[dict[Any, Any]]:
     db_pool = cast(ConnectionPool, request.app.state.db_pool)
@@ -46,6 +53,7 @@ async def pg_pool_example(request: Request) -> list[dict[Any, Any]]:
 app = Litestar(
     [pg_pool_example],
     on_startup=[start_db_pool],
+    on_shutdown=[stop_db_pool],
 )
 
 

docs/usage/frameworks/robyn.md

@@ -36,7 +36,10 @@ async def pg_pool_example(request: Request) -> list[dict[Any, Any]]:
 
 
 async def main() -> None:
-    await app.start(host="127.0.0.1", port=8000)
+    try:
+        await app.start(host="127.0.0.1", port=8000)
+    finally:
+        db_pool.close()
 
 
 if __name__ == "__main__":

src/driver/connection_pool.rs

@@ -422,7 +422,7 @@ impl ConnectionPool {
         Ok(Connection::new(Some(Arc::new(db_connection)), None))
     }
 
-    /// Return new single connection.
+    /// Close connection pool.
     ///
     /// # Errors
     /// May return Err Result if cannot get new connection from the pool.