diff options
Diffstat (limited to 'lib/sqlalchemy/pool')
| -rw-r--r-- | lib/sqlalchemy/pool/__init__.py | 2 | ||||
| -rw-r--r-- | lib/sqlalchemy/pool/impl.py | 8 |
2 files changed, 9 insertions, 1 deletions
diff --git a/lib/sqlalchemy/pool/__init__.py b/lib/sqlalchemy/pool/__init__.py index eb0d37517..353f34333 100644 --- a/lib/sqlalchemy/pool/__init__.py +++ b/lib/sqlalchemy/pool/__init__.py @@ -28,6 +28,7 @@ from .base import reset_rollback from .dbapi_proxy import clear_managers from .dbapi_proxy import manage from .impl import AssertionPool +from .impl import AsyncAdaptedQueuePool from .impl import NullPool from .impl import QueuePool from .impl import SingletonThreadPool @@ -44,6 +45,7 @@ __all__ = [ "AssertionPool", "NullPool", "QueuePool", + "AsyncAdaptedQueuePool", "SingletonThreadPool", "StaticPool", ] diff --git a/lib/sqlalchemy/pool/impl.py b/lib/sqlalchemy/pool/impl.py index 0fe7612b9..e1a9f00db 100644 --- a/lib/sqlalchemy/pool/impl.py +++ b/lib/sqlalchemy/pool/impl.py @@ -33,6 +33,8 @@ class QueuePool(Pool): """ + _queue_class = sqla_queue.Queue + def __init__( self, creator, @@ -95,7 +97,7 @@ class QueuePool(Pool): """ Pool.__init__(self, creator, **kw) - self._pool = sqla_queue.Queue(pool_size, use_lifo=use_lifo) + self._pool = self._queue_class(pool_size, use_lifo=use_lifo) self._overflow = 0 - pool_size self._max_overflow = max_overflow self._timeout = timeout @@ -215,6 +217,10 @@ class QueuePool(Pool): return self._pool.maxsize - self._pool.qsize() + self._overflow +class AsyncAdaptedQueuePool(QueuePool): + _queue_class = sqla_queue.AsyncAdaptedQueue + + class NullPool(Pool): """A Pool which does not pool connections. |