Revert .close removal

Author: insani7y

docs/examples/aiohttp/start_example.py

@@ -16,6 +16,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(PSQLPool, app.db_pool)
+    await db_pool.close()
+
+
 async def pg_pool_example(request: web.Request) -> Any:
     db_pool = cast(PSQLPool, request.app["db_pool"])
     connection = await db_pool.connection()

docs/examples/fastapi/advanced_example.py

@@ -19,6 +19,7 @@ async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
     """Startup database connection pool and close it on shutdown."""
     app.state.db_pool = db_pool
     yield
+    await db_pool.close()
 
 
 app = FastAPI(lifespan=lifespan)

docs/examples/fastapi/start_example.py

@@ -18,6 +18,7 @@ async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
     )
     app.state.db_pool = db_pool
     yield
+    await db_pool.close()
 
 
 app = FastAPI(lifespan=lifespan)

python/psqlpy/_internal/__init__.pyi

@@ -882,6 +882,12 @@ class PSQLPool:
         - `max_db_pool_size`: maximum size of the connection pool
         - `conn_recycling_method`: how a connection is recycled.
         """
+    async def close(self: Self) -> None:
+        """Close the connection pool.
+
+        By default it will be closed automatically,
+        but you can call it manually.
+        """
     async def execute(
         self: Self,
         querystring: str,

python/tests/test_connection_pool.py

@@ -1,6 +1,7 @@
 import pytest
 
 from psqlpy import Connection, ConnRecyclingMethod, PSQLPool, QueryResult
+from psqlpy.exceptions import RustPSQLDriverPyBaseError
 
 pytestmark = pytest.mark.anyio
 
@@ -56,3 +57,17 @@ async def test_pool_conn_recycling_method(
     )
 
     await pg_pool.execute("SELECT 1")
+
+
+async def test_close_connection_pool() -> None:
+    """Test that `close` method closes connection pool."""
+    pg_pool = PSQLPool(
+        dsn="postgres://postgres:postgres@localhost:5432/psqlpy_test",
+    )
+
+    await pg_pool.execute("SELECT 1")
+
+    await pg_pool.close()
+
+    with pytest.raises(expected_exception=RustPSQLDriverPyBaseError):
+        await pg_pool.execute("SELECT 1")