saq 0.24.12__tar.gz → 0.24.13__tar.gz

{saq-0.24.12/saq.egg-info → saq-0.24.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: saq
-Version: 0.24.12
+Version: 0.24.13
 Summary: Distributed Python job queue with asyncio and redis
 Home-page: https://github.com/tobymao/saq
 Author: Toby Mao

{saq-0.24.12 → saq-0.24.13}/saq/__init__.py

@@ -14,4 +14,4 @@ __all__ = [
     "Worker",
 ]
 
-__version__ = "0.24.12"
+__version__ = "0.24.13"

{saq-0.24.12 → saq-0.24.13}/saq/queue/postgres.py

@@ -888,7 +888,21 @@ class PostgresQueue(Queue):
 
     @asynccontextmanager
     async def _get_dequeue_conn(self) -> t.AsyncGenerator:
+        """
+        This context manager is used instead of the pool.connection() context manager because we need
+        to reuse specific connections outside of a context to properly manager advisory locks.
+        If advisory locks are not used, this context manager can be replaced with pool.connection().
+        """
         async with self._connection_lock:
+            # Check if the connection is still associated with a connection pool.
+            if not getattr(self._dequeue_conn, "_pool", None):
+                # The only known case of having a connection without a pool is when the connection was closed
+                # Therefore this close check shouldn't be needed but it doesn't hurt to close anyways
+                # to cover potential case where the connection is not closed and we would leave an open connection.
+                if self._dequeue_conn:
+                    await self._dequeue_conn.close()
+                self._dequeue_conn = None
+
             if self._dequeue_conn:
                 try:
                     # Pool normally performs this check when getting a connection.

{saq-0.24.12 → saq-0.24.13/saq.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: saq
-Version: 0.24.12
+Version: 0.24.13
 Summary: Distributed Python job queue with asyncio and redis
 Home-page: https://github.com/tobymao/saq
 Author: Toby Mao