diff options
Diffstat (limited to 'lib/sqlalchemy/pool/impl.py')
| -rw-r--r-- | lib/sqlalchemy/pool/impl.py | 8 |
1 files changed, 7 insertions, 1 deletions
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. |