feat: adds get_connection methods (#4)

Author: cofin

pyproject.toml

@@ -26,7 +26,7 @@ license = { text = "MIT" }
 name = "litestar-oracledb"
 readme = "README.md"
 requires-python = ">=3.8"
-version = "0.1.2"
+version = "0.1.3"
 
 [project.urls]
 Changelog = "https://litestar-org.github.io/litesatr-oracledb/latest/changelog"

src/litestar_oracledb/config/_asyncio.py

@@ -224,3 +224,16 @@ async def provide_connection(
             async with pool.acquire() as connection:
                 set_scope_state(scope, self.connection_scope_key, connection)
                 yield connection
+
+    @asynccontextmanager
+    async def get_connection(
+        self,
+    ) -> AsyncGenerator[AsyncConnection, None]:
+        """Create a connection instance.
+
+        Returns:
+            A connection instance.
+        """
+        pool = await self.create_pool()
+        async with pool.acquire() as connection:
+            yield connection

src/litestar_oracledb/config/_sync.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from contextlib import asynccontextmanager
+from contextlib import asynccontextmanager, contextmanager
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Generator, Optional, cast
 
@@ -193,7 +193,7 @@ async def lifespan(
         try:
             yield
         finally:
-            db_pool.close(force=True)
+            db_pool.close()
 
     def provide_connection(
         self,
@@ -219,3 +219,16 @@ def provide_connection(
             with pool.acquire() as connection:
                 set_scope_state(scope, self.connection_scope_key, connection)
                 yield connection
+
+    @contextmanager
+    def get_connection(
+        self,
+    ) -> Generator[Connection, None, None]:
+        """Create a connection instance.
+
+        Returns:
+            A connection instance.
+        """
+        pool = self.create_pool()
+        with pool.acquire() as connection:
+            yield connection