async-task-kit 0.1.13__tar.gz → 0.1.14__tar.gz

{async_task_kit-0.1.13 → async_task_kit-0.1.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: async-task-kit
-Version: 0.1.13
+Version: 0.1.14
 Summary: A powerful async task processing kit based on RabbitMQ with Coroutine, Thread, and Process support.
 Author-email: realwrtoff <realwrtoff@gmail.com>
 Classifier: Programming Language :: Python :: 3

{async_task_kit-0.1.13 → async_task_kit-0.1.14}/async_task_kit/consumer/coroutine.py

@@ -18,6 +18,7 @@ class CoroutineConsumer(BaseConsumer):
         concurrency: int = 1,
         max_retry: int = 3,
         retry_delay: int = 30,
+        rmq: RabbitMQ | None = None,
     ):
         self.amqp_url = amqp_url
         self.queue_name = queue_name
@@ -26,7 +27,8 @@ class CoroutineConsumer(BaseConsumer):
         self.max_retry = max_retry
         self.retry_delay = retry_delay
 
-        self._rmq = RabbitMQ(amqp_url)
+        self._owns_rmq = rmq is None
+        self._rmq = rmq or RabbitMQ(amqp_url)
         self._stop_event = asyncio.Event()
 
     async def start(self):
@@ -93,5 +95,6 @@ class CoroutineConsumer(BaseConsumer):
 
     async def stop(self):
         self._stop_event.set()
-        await self._rmq.close()
-        logger.info(f"[CoroutineConsumer] stopped | queue={self.queue_name}")
+        if self._owns_rmq:
+            await self._rmq.close()
+        logger.info(f"[CoroutineConsumer] stopped | queue={self.queue_name}")

{async_task_kit-0.1.13 → async_task_kit-0.1.14}/async_task_kit/consumer/process.py

@@ -104,4 +104,4 @@ class ProcessConsumer(BaseConsumer):
 
     async def stop(self):
         self._stop_event.set()
-        logger.info(f"[ProcessConsumer] stopped | queue={self.queue_name}")
+        logger.info(f"[ProcessConsumer] stopped | queue={self.queue_name}")

{async_task_kit-0.1.13 → async_task_kit-0.1.14}/async_task_kit/consumer/thread.py

@@ -104,4 +104,4 @@ class ThreadConsumer(BaseConsumer):
 
     async def stop(self):
         self._stop_event.set()
-        logger.info(f"[ThreadConsumer] stopped | queue={self.queue_name}")
+        logger.info(f"[ThreadConsumer] stopped | queue={self.queue_name}")

{async_task_kit-0.1.13 → async_task_kit-0.1.14}/async_task_kit/core/rabbitmq.py

@@ -12,6 +12,7 @@ from aio_pika.exceptions import (
     ChannelClosed,
     AMQPError
 )
+from aiormq.exceptions import ChannelInvalidStateError
 from aio_pika.pool import Pool
 
 logger = logging.getLogger(__name__)
@@ -63,6 +64,7 @@ class RabbitMQ(BasicQueue):
         channel_size: int = DEFAULT_CHANNEL_SIZE,
         heartbeat: int = 60,
         max_retry_count: int = DEFAULT_MAX_RETRY_COUNT,
+        
     ):
         self.rabbit_url = rabbit_url
         self.pool_size = pool_size
@@ -157,6 +159,10 @@ class RabbitMQ(BasicQueue):
             logger.warning("[PUSH] 连接断开，准备重连")
             await self._invalidate_pools()
             return None
+        except ChannelInvalidStateError:
+            logger.warning("[PUSH] 信道 RPC 异常，准备重建 pool")
+            await self._invalidate_pools()
+            return None
         except Exception as e:
             logger.error(f"[PUSH] 异常 queue={queue_name}: {e}", exc_info=True)
             return None
@@ -172,13 +178,15 @@ class RabbitMQ(BasicQueue):
         try:
             async with self._channel_pool.acquire() as channel:
                 queue = await channel.declare_queue(queue_name, durable=durable)
-                return await asyncio.wait_for(
-                    queue.get(), timeout=self.DEFAULT_POP_TIMEOUT
+                # 使用 aio_pika 自带 timeout，勿用 asyncio.wait_for 取消 get（会关闭 channel）
+                return await queue.get(
+                    timeout=self.DEFAULT_POP_TIMEOUT,
+                    fail=False,
                 )
         except (QueueEmpty, asyncio.TimeoutError):
             return None
-        except (ConnectionClosed, ChannelClosed):
-            logger.warning("[POP] 连接断开，准备重连")
+        except (ConnectionClosed, ChannelClosed, ChannelInvalidStateError):
+            logger.warning("[POP] 连接/信道异常，准备重建 pool")
             await self._invalidate_pools()
             return None
         except Exception as e:
@@ -201,6 +209,10 @@ class RabbitMQ(BasicQueue):
             logger.warning("[QUEUE_LENGTH] 连接断开，准备重连")
             await self._invalidate_pools()
             return self.QUEUE_LENGTH_UNAVAILABLE
+        except ChannelInvalidStateError:
+            logger.warning("[QUEUE_LENGTH] 信道 RPC 异常，准备重建 pool")
+            await self._invalidate_pools()
+            return self.QUEUE_LENGTH_UNAVAILABLE
         except Exception as e:
             logger.error(f"[QUEUE_LENGTH] 异常 queue={queue_name}: {e}", exc_info=True)
             return self.QUEUE_LENGTH_UNAVAILABLE

{async_task_kit-0.1.13 → async_task_kit-0.1.14}/async_task_kit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: async-task-kit
-Version: 0.1.13
+Version: 0.1.14
 Summary: A powerful async task processing kit based on RabbitMQ with Coroutine, Thread, and Process support.
 Author-email: realwrtoff <realwrtoff@gmail.com>
 Classifier: Programming Language :: Python :: 3

{async_task_kit-0.1.13 → async_task_kit-0.1.14}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "async-task-kit"
-version = "0.1.13"
+version = "0.1.14"
 description = "A powerful async task processing kit based on RabbitMQ with Coroutine, Thread, and Process support."
 readme = "README.md"
 requires-python = ">=3.12"