Merge pull request #117 from QueryaHub/issue-116-pool-management

feat(db): connection pool management in AppState (Closes #116)

Author: ZhuchkaTriplesix

oxyroute/app.py

@@ -387,8 +387,8 @@ async def __rsgi_init__(self, *args: Any, **kwargs: Any) -> None:
         return None
 
     async def __rsgi_del__(self, *args: Any, **kwargs: Any) -> None:
-        """RSGI worker teardown (no-op in the base class)."""
-        return None
+        """RSGI worker teardown. Closes the global connection pool if it exists."""
+        await self.close_database()
 
     async def __rsgi__(self, scope: Any, protocol: Any) -> Any:
         """

tests/test_database_pool.py

@@ -0,0 +1,22 @@
+import pytest
+from oxyroute import App
+
+
+@pytest.mark.anyio
+async def test_database_pool_failure():
+    app = App()
+
+    # Attempting to connect to a non-existent database should fail fast
+    with pytest.raises(RuntimeError, match="DB connect error:"):
+        await app.setup_database("postgresql://foo:bar@127.0.0.1:12345/baz")
+
+
+@pytest.mark.anyio
+async def test_database_pool_teardown_no_op():
+    app = App()
+
+    # Calling close without setup should be a no-op and not raise any errors
+    await app.close_database()
+
+    # __rsgi_del__ should also gracefully do nothing if no db was set up
+    await app.__rsgi_del__()