sycommon-python-lib 0.1.29__py3-none-any.whl → 0.1.40__py3-none-any.whl

sycommon/rabbitmq/rabbitmq_client.py

@@ -1,19 +1,19 @@
+from aio_pika import Channel
+from typing import Optional
 import asyncio
 import json
-from typing import Callable, Coroutine, Optional, Dict, Any, Union, Set
+from typing import Callable, Coroutine, Dict, Any, Union
 from aio_pika import Message, DeliveryMode, ExchangeType
 from aio_pika.abc import (
-    AbstractChannel,
     AbstractExchange,
     AbstractQueue,
     AbstractIncomingMessage,
     ConsumerTag,
 )
-from aiormq.exceptions import ChannelInvalidStateError, ConnectionClosed
-
+from sycommon.rabbitmq.rabbitmq_pool import RabbitMQConnectionPool
 from sycommon.logging.kafka_log import SYLogger
 from sycommon.models.mqmsg_model import MQMsgModel
-from sycommon.rabbitmq.rabbitmq_pool import RabbitMQConnectionPool
+
 
 # 最大重试次数限制
 MAX_RETRY_COUNT = 3
@@ -23,8 +23,8 @@ logger = SYLogger
 
 class RabbitMQClient:
     """
-    RabbitMQ客户端（基于连接池），支持集群多节点配置
-    提供自动故障转移、连接恢复和消息可靠性保障
+    RabbitMQ客户端（基于连接池），支持集群、自动重连、消息发布/消费
+    依赖 aio_pika 的内置重连机制，移除手动重连逻辑
     """
 
     def __init__(
@@ -38,646 +38,263 @@ class RabbitMQClient:
         auto_delete: bool = False,
         auto_parse_json: bool = True,
         create_if_not_exists: bool = True,
-        connection_timeout: int = 10,
         rpc_timeout: int = 10,
-        reconnection_delay: int = 1,
-        max_reconnection_attempts: int = 5,
         prefetch_count: int = 2,
-        consumption_stall_threshold: int = 10
+        consumption_stall_threshold: int = 60,
     ):
-        """
-        初始化RabbitMQ客户端（依赖连接池）
-
-        :param connection_pool: 连接池实例
-        """
-        # 连接池依赖
         self.connection_pool = connection_pool
+        if not self.connection_pool._initialized:
+            raise RuntimeError("连接池未初始化，请先调用 connection_pool.init_pools()")
 
-        # 交换器和队列参数
+        # 交换机配置
         self.exchange_name = exchange_name
-        self.exchange_type = ExchangeType(exchange_type)
+        self.exchange_type = ExchangeType(exchange_type.lower())
+        # 队列配置
         self.queue_name = queue_name
         self.routing_key = routing_key
         self.durable = durable
         self.auto_delete = auto_delete
-
-        # 行为控制参数
         self.auto_parse_json = auto_parse_json
         self.create_if_not_exists = create_if_not_exists
-        self.connection_timeout = connection_timeout
+
+        # 运行时配置
         self.rpc_timeout = rpc_timeout
         self.prefetch_count = prefetch_count
-
-        # 重连参数
-        self.reconnection_delay = reconnection_delay
-        self.max_reconnection_attempts = max_reconnection_attempts
-
-        # 消息处理参数
         self.consumption_stall_threshold = consumption_stall_threshold
 
-        # 通道和资源对象（从池获取）
-        self.channel: Optional[AbstractChannel] = None
-        self.exchange: Optional[AbstractExchange] = None
-        self.queue: Optional[AbstractQueue] = None
-
-        # 状态跟踪
-        self.actual_queue_name: Optional[str] = None
-        self._exchange_exists = False
-        self._queue_exists = False
-        self._queue_bound = False
-        self._is_consuming = False
-        self._closed = False
+        # 内部状态
+        self._channel: Optional[Channel] = None
+        self._exchange: Optional[AbstractExchange] = None
+        self._queue: Optional[AbstractQueue] = None
         self._consumer_tag: Optional[ConsumerTag] = None
-        self._last_activity_timestamp = asyncio.get_event_loop().time()
-        self._last_message_processed = asyncio.get_event_loop().time()
-
-        # 任务和处理器
-        self.message_handler: Optional[Callable[
-            [Union[Dict[str, Any], str], AbstractIncomingMessage],
-            Coroutine[Any, Any, None]
-        ]] = None
-        self._consuming_task: Optional[asyncio.Task] = None
-        self._reconnect_task: Optional[asyncio.Task] = None
-        self._keepalive_task: Optional[asyncio.Task] = None
-        self._monitor_task: Optional[asyncio.Task] = None
-
-        # 消息处理跟踪
-        self._processing_message_ids: Set[str] = set()
+        self._message_handler: Optional[Callable] = None
+        self._closed = False
 
-    @property
-    def is_connected(self) -> bool:
-        """检查当前通道是否有效"""
-        return (not self._closed and
-                self.channel is not None and
-                not self.channel.is_closed and
-                self.exchange is not None)
-
-    def _update_activity_timestamp(self) -> None:
-        """更新最后活动时间戳"""
-        self._last_activity_timestamp = asyncio.get_event_loop().time()
-
-    def _update_message_processed_timestamp(self) -> None:
-        """更新最后消息处理时间戳"""
-        self._last_message_processed = asyncio.get_event_loop().time()
-
-    async def _get_channel(self) -> AbstractChannel:
-        """从通道池获取通道（使用上下文管理器）"""
-        if not self.connection_pool.channel_pool:
-            raise Exception("连接池未初始化，请先调用init_pools")
-
-        # 使用async with获取通道，并通过变量返回
-        async with self.connection_pool.channel_pool.acquire() as channel:
-            return channel
-
-    async def _check_exchange_exists(self, channel: AbstractChannel) -> bool:
-        """检查交换机是否存在"""
-        try:
-            # 使用被动模式检查交换机
-            await asyncio.wait_for(
-                channel.declare_exchange(
-                    name=self.exchange_name,
-                    type=self.exchange_type,
-                    passive=True
-                ),
-                timeout=self.rpc_timeout
-            )
-            return True
-        except Exception:
-            return False
+        # 细粒度锁
+        self._consume_lock = asyncio.Lock()
+        self._connect_lock = asyncio.Lock()
 
-    async def _check_queue_exists(self, channel: AbstractChannel) -> bool:
-        """检查队列是否存在"""
-        if not self.queue_name:
+    @property
+    async def is_connected(self) -> bool:
+        """异步检查客户端连接状态（属性，不是函数）"""
+        if self._closed:
             return False
         try:
-            # 使用被动模式检查队列
-            await asyncio.wait_for(
-                channel.declare_queue(
-                    name=self.queue_name,
-                    passive=True
-                ),
-                timeout=self.rpc_timeout
-            )
-            return True
-        except Exception:
-            return False
-
-    async def _bind_queue(self, channel: AbstractChannel, queue: AbstractQueue, exchange: AbstractExchange) -> bool:
-        """将队列绑定到交换机"""
-        bind_routing_key = self.routing_key if self.routing_key else '#'
-
-        for attempt in range(MAX_RETRY_COUNT + 1):
-            try:
-                await asyncio.wait_for(
-                    queue.bind(
-                        exchange,
-                        routing_key=bind_routing_key
-                    ),
-                    timeout=self.rpc_timeout
-                )
-                logger.info(
-                    f"队列 '{self.queue_name}' 已绑定到交换机 '{self.exchange_name}'，路由键: {bind_routing_key}")
+            # 检查通道是否有效
+            if self._channel and not self._channel.is_closed:
+                # 检查交换机和队列（如果需要）
+                if self.create_if_not_exists and self.queue_name:
+                    return bool(self._exchange and self._queue)
                 return True
-            except Exception as e:
-                logger.warning(
-                    f"队列绑定失败（第{attempt+1}次尝试）: {str(e)}")
-            if attempt < MAX_RETRY_COUNT:
-                await asyncio.sleep(1)
-        return False
-
-    async def connect(self, force_reconnect: bool = False, declare_queue: bool = True) -> None:
-        """
-        从连接池获取资源并初始化（交换机、队列）
-        """
-        logger.debug(
-            f"连接参数 - force_reconnect={force_reconnect}, "
-            f"declare_queue={declare_queue}, create_if_not_exists={self.create_if_not_exists}"
-        )
-
-        # 如果已连接且不强制重连，则直接返回
-        if self.is_connected and not force_reconnect:
-            return
-
-        # 取消正在进行的重连任务
-        if self._reconnect_task and not self._reconnect_task.done():
-            self._reconnect_task.cancel()
+            return False
+        except Exception as e:
+            logger.warning(f"检查连接状态失败: {str(e)}")
+            return False
 
-        # 重置状态
-        self._exchange_exists = False
-        self._queue_exists = False
-        self._queue_bound = False
+    async def _get_connection_resources(self) -> tuple[Optional[Channel], Optional[AbstractExchange], Optional[AbstractQueue]]:
+        """原子获取连接资源（通道、交换机、队列）"""
+        async with self._connect_lock:
+            return self._channel, self._exchange, self._queue
 
-        retries = 0
-        last_exception = None
+    async def connect(self) -> None:
+        """建立连接并初始化交换机/队列（使用新的通道获取方式）"""
+        if self._closed:
+            raise RuntimeError("客户端已关闭，无法重新连接")
 
-        while retries < self.max_reconnection_attempts:
+        async with self._connect_lock:
             try:
-                # 从池获取通道
-                self.channel = await self._get_channel()
-                await self.channel.set_qos(prefetch_count=self.prefetch_count)
-
-                # 处理交换机
-                exchange_exists = await self._check_exchange_exists(self.channel)
-                if not exchange_exists:
-                    if self.create_if_not_exists:
-                        # 创建交换机
-                        self.exchange = await asyncio.wait_for(
-                            self.channel.declare_exchange(
-                                name=self.exchange_name,
-                                type=self.exchange_type,
-                                durable=self.durable,
-                                auto_delete=self.auto_delete
-                            ),
-                            timeout=self.rpc_timeout
-                        )
-                        logger.info(f"已创建交换机 '{self.exchange_name}'")
-                    else:
-                        raise Exception(
-                            f"交换机 '{self.exchange_name}' 不存在且不允许自动创建")
-                else:
-                    # 获取已有交换机
-                    self.exchange = await self.channel.get_exchange(self.exchange_name)
-                    logger.info(f"使用已存在的交换机 '{self.exchange_name}'")
-
-                # 处理队列
-                if declare_queue and self.queue_name:
-                    queue_exists = await self._check_queue_exists(self.channel)
-
-                    if not queue_exists:
-                        if not self.create_if_not_exists:
-                            raise Exception(
-                                f"队列 '{self.queue_name}' 不存在且不允许自动创建")
-
-                        # 创建队列
-                        self.queue = await asyncio.wait_for(
-                            self.channel.declare_queue(
-                                name=self.queue_name,
-                                durable=self.durable,
-                                auto_delete=self.auto_delete,
-                                exclusive=False
-                            ),
-                            timeout=self.rpc_timeout
-                        )
-                        self.actual_queue_name = self.queue_name
-                        logger.info(f"已创建队列 '{self.queue_name}'")
-                    else:
-                        # 获取已有队列
-                        self.queue = await self.channel.get_queue(self.queue_name)
-                        self.actual_queue_name = self.queue_name
-                        logger.info(f"使用已存在的队列 '{self.queue_name}'")
+                # 关键修复：使用连接池的 acquire_channel 方法获取通道
+                self._channel = await self.connection_pool.acquire_channel()
+                # 设置预取计数
+                await self._channel.set_qos(prefetch_count=self.prefetch_count)
+
+                # 声明交换机
+                self._exchange = await self._channel.declare_exchange(
+                    name=self.exchange_name,
+                    type=self.exchange_type,
+                    durable=self.durable,
+                    auto_delete=self.auto_delete,
+                    passive=not self.create_if_not_exists  # 不创建时使用passive模式检查
+                )
 
+                # 声明队列（如果配置了队列名）
+                if self.queue_name:
+                    self._queue = await self._channel.declare_queue(
+                        name=self.queue_name,
+                        durable=self.durable,
+                        auto_delete=self.auto_delete,
+                        passive=not self.create_if_not_exists
+                    )
                     # 绑定队列到交换机
-                    if self.queue and self.exchange:
-                        bound = await self._bind_queue(self.channel, self.queue, self.exchange)
-                        if not bound:
-                            raise Exception(f"队列 '{self.queue_name}' 绑定到交换机失败")
+                    await self._queue.bind(
+                        exchange=self._exchange,
+                        routing_key=self.routing_key or self.queue_name
+                    )
+                    logger.info(
+                        f"队列 '{self.queue_name}' 已声明并绑定到交换机 '{self.exchange_name}' "
+                        f"(exchange_type: {self.exchange_type}, routing_key: {self.routing_key})"
+                    )
                 else:
-                    # 不声明队列时的状态处理
-                    self.queue = None
-                    self.actual_queue_name = None
-                    logger.debug(f"跳过队列 '{self.queue_name}' 的声明和绑定")
-
-                # 验证连接状态
-                if not self.is_connected:
-                    raise Exception("连接验证失败，状态异常")
-
-                # 如果之前在消费，重新开始消费
-                if self._is_consuming and self.message_handler:
-                    await self.start_consuming()
-
-                # 启动连接监控和保活任务
-                self._start_monitoring()
-                self._start_keepalive()
-
-                self._update_activity_timestamp()
-                logger.info(f"RabbitMQ客户端初始化成功 (队列: {self.actual_queue_name})")
-                return
+                    logger.info(
+                        f"未配置队列名，仅初始化交换机 '{self.exchange_name}' (exchange_type: {self.exchange_type})")
 
+                logger.info(f"RabbitMQ客户端连接成功（exchange: {self.exchange_name}）")
             except Exception as e:
-                last_exception = e
-                logger.warning(f"资源初始化失败: {str(e)}，重试中...")
-                # 释放当前通道（放回池并重新获取）
-                self.channel = None
-                retries += 1
-                if retries < self.max_reconnection_attempts:
-                    await asyncio.sleep(self.reconnection_delay)
-
-        logger.error(f"最终初始化失败: {str(last_exception)}")
-        raise Exception(
-            f"经过{self.max_reconnection_attempts}次重试后仍无法初始化客户端。最后错误: {str(last_exception)}")
-
-    def _start_monitoring(self) -> None:
-        """启动连接和消费监控任务"""
-        if self._closed or (self._monitor_task and not self._monitor_task.done()):
-            return
-
-        async def monitor():
-            while not self._closed and self.channel:
+                logger.error(f"客户端连接失败: {str(e)}", exc_info=True)
+                # 清理异常状态
+                if self._channel:
+                    await self.connection_pool.release_channel(self._channel)
+                self._channel = None
+                self._exchange = None
+                self._queue = None
+                raise
+
+    async def set_message_handler(self, handler: Callable[[MQMsgModel, AbstractIncomingMessage], Coroutine[Any, Any, None]]):
+        """设置消息处理器（消费消息时使用）"""
+        async with self._consume_lock:
+            self._message_handler = handler
+            logger.info("消息处理器已设置")
+
+    async def start_consuming(self) -> Optional[ConsumerTag]:
+        """启动消费（返回消费者标签）"""
+        if self._closed:
+            logger.warning("客户端已关闭，无法启动消费")
+            return None
+
+        async with self._consume_lock:
+            # 检查前置条件
+            if not self._message_handler:
+                raise RuntimeError("未设置消息处理器，请先调用 set_message_handler")
+
+            # 确保连接已建立
+            if not await self.is_connected:
+                await self.connect()
+
+            # 确保队列已初始化
+            _, _, queue = await self._get_connection_resources()
+            if not queue:
+                raise RuntimeError("队列未初始化，无法启动消费")
+
+            # 定义消费回调
+            async def consume_callback(message: AbstractIncomingMessage):
                 try:
-                    # 检查通道状态
-                    if self.channel.is_closed:
-                        logger.warning("检测到通道已关闭，尝试重建")
-                        await self._recreate_channel()
-                        continue
-
-                    # 检查消费停滞
-                    if self._is_consuming:
-                        current_time = asyncio.get_event_loop().time()
-                        if current_time - self._last_message_processed > self.consumption_stall_threshold:
-                            if self._processing_message_ids:
-                                logger.warning(
-                                    f"消费停滞，但有 {len(self._processing_message_ids)} 个消息正在处理，暂不重启")
-                            else:
-                                # 确实无消息处理，再重启
-                                await self.stop_consuming()
-                                await asyncio.sleep(1)
-                                await self.start_consuming()
-                except Exception as e:
-                    logger.error(f"监控任务出错: {str(e)}")
-                    await asyncio.sleep(1)
-
-                await asyncio.sleep(30)
-
-        self._monitor_task = asyncio.create_task(monitor())
-
-    async def _recreate_channel(self) -> None:
-        try:
-            # 无需手动释放，上下文管理器会自动处理
-            self.channel = await self._get_channel()
-            await self.channel.set_qos(prefetch_count=self.prefetch_count)
-
-            # 重新获取交换机
-            self.exchange = await self.channel.get_exchange(self.exchange_name)
-
-            # 重新绑定队列
-            if self.queue_name:
-                self.queue = await self.channel.get_queue(self.queue_name)
-                if self.queue and self.exchange:
-                    await self._bind_queue(self.channel, self.queue, self.exchange)
-
-            # 重新开始消费
-            if self._is_consuming and self.message_handler:
-                await self.start_consuming()
+                    # 解析消息
+                    if self.auto_parse_json:
+                        body = json.loads(message.body.decode('utf-8'))
+                        parsed_data = MQMsgModel(**body)
+                    else:
+                        parsed_data = MQMsgModel(
+                            msg=message.body.decode('utf-8'))
 
-            logger.info("通道已重建并恢复服务")
-            self._update_activity_timestamp()
-        except Exception as e:
-            logger.error(f"通道重建失败: {str(e)}，触发重连")
-            await self.connect(force_reconnect=True)
+                    # 调用处理器
+                    await self._message_handler(parsed_data, message)
+                    # 手动确认消息
+                    await message.ack()
+                except Exception as e:
+                    logger.error(
+                        f"处理消息失败 (delivery_tag: {message.delivery_tag}): {str(e)}", exc_info=True)
+                    # 消费失败时重新入队（最多重试3次）
+                    if message.redelivered:
+                        logger.warning(
+                            f"消息已重试过，拒绝入队 (delivery_tag: {message.delivery_tag})")
+                        await message.reject(requeue=False)
+                    else:
+                        await message.nack(requeue=True)
 
-    def _start_keepalive(self) -> None:
-        """启动连接保活任务"""
-        if self._closed or (self._keepalive_task and not self._keepalive_task.done()):
-            return
+            # 启动消费
+            self._consumer_tag = await queue.consume(consume_callback)
+            logger.info(
+                f"开始消费队列 '{queue.name}'，consumer_tag: {self._consumer_tag}")
+            return self._consumer_tag
 
-        async def keepalive():
-            while not self._closed and self.is_connected:
-                current_time = asyncio.get_event_loop().time()
-                # 检查是否超过指定时间无活动
-                if current_time - self._last_activity_timestamp > self.connection_pool.heartbeat * 1.5:
-                    logger.debug(
-                        f"连接 {self.connection_pool.heartbeat*1.5}s 无活动，执行保活检查")
+    async def stop_consuming(self) -> None:
+        """停止消费"""
+        async with self._consume_lock:
+            if self._consumer_tag and not self._closed:
+                _, _, queue = await self._get_connection_resources()
+                if queue and not queue.is_closed:
                     try:
-                        if self.channel.is_closed:
-                            logger.warning("连接已关闭，触发重连")
-                            await self.connect(force_reconnect=True)
-                            return
-
-                        # 轻量级操作保持连接活跃
-                        await asyncio.wait_for(
-                            self.channel.declare_exchange(
-                                name=self.exchange_name,
-                                type=self.exchange_type,
-                                passive=True
-                            ),
-                            timeout=5
-                        )
-                        self._update_activity_timestamp()
+                        await queue.cancel(self._consumer_tag)
+                        logger.info(f"停止消费，consumer_tag: {self._consumer_tag}")
                     except Exception as e:
-                        logger.warning(f"保活检查失败: {str(e)}，触发重连")
-                        await self.connect(force_reconnect=True)
-
-                await asyncio.sleep(self.connection_pool.heartbeat / 2)
-
-        self._keepalive_task = asyncio.create_task(keepalive())
-
-    async def _schedule_reconnect(self) -> None:
-        """安排重新连接"""
-        if self._reconnect_task and not self._reconnect_task.done():
-            return
-
-        logger.info(f"将在 {self.reconnection_delay} 秒后尝试重新连接...")
-
-        async def reconnect():
-            try:
-                await asyncio.sleep(self.reconnection_delay)
-                if not self._closed:
-                    await self.connect(force_reconnect=True)
-            except Exception as e:
-                logger.error(f"重连任务失败: {str(e)}")
-                if not self._closed:
-                    await self._schedule_reconnect()
-
-        self._reconnect_task = asyncio.create_task(reconnect())
-
-    async def close(self) -> None:
-        """关闭客户端并释放资源"""
-        self._closed = True
-        self._is_consuming = False
-
-        # 取消所有任务
-        for task in [self._keepalive_task, self._reconnect_task,
-                     self._consuming_task, self._monitor_task]:
-            if task and not task.done():
-                task.cancel()
-                try:
-                    await task
-                except asyncio.CancelledError:
-                    pass
-
-        # 重置状态
-        self.channel = None
-        self.exchange = None
-        self.queue = None
-        self._consumer_tag = None
-        self._processing_message_ids.clear()
-
-        logger.info("RabbitMQ客户端已关闭")
+                        logger.error(f"停止消费失败: {str(e)}")
+                self._consumer_tag = None
 
     async def publish(
         self,
-        message_body: Union[str, Dict[str, Any]],
-        routing_key: Optional[str] = None,
-        content_type: str = "application/json",
+        message_body: Union[str, Dict[str, Any], MQMsgModel],
         headers: Optional[Dict[str, Any]] = None,
+        content_type: str = "application/json",
         delivery_mode: DeliveryMode = DeliveryMode.PERSISTENT
     ) -> None:
-        """发布消息（从池获取通道，自动重试）"""
-        if not self.is_connected:
-            logger.warning("连接已关闭，尝试重连后发布消息")
-            await self.connect(force_reconnect=True)
+        """发布消息（修复 is_closed 问题）"""
+        if self._closed:
+            raise RuntimeError("客户端已关闭，无法发布消息")
 
-        if not self.channel or not self.exchange:
-            raise Exception("RabbitMQ连接未初始化")
+        # 确保连接已建立
+        if not await self.is_connected:
+            await self.connect()
 
         # 处理消息体
-        if isinstance(message_body, dict):
-            message_body_str = json.dumps(message_body, ensure_ascii=False)
-            if content_type == "text/plain":
-                content_type = "application/json"
+        if isinstance(message_body, MQMsgModel):
+            body = json.dumps(message_body.__dict__,
+                              ensure_ascii=False).encode('utf-8')
+        elif isinstance(message_body, dict):
+            body = json.dumps(message_body, ensure_ascii=False).encode('utf-8')
         else:
-            message_body_str = str(message_body)
+            body = str(message_body).encode('utf-8')
 
-        # 创建消息对象
+        # 创建消息
         message = Message(
-            body=message_body_str.encode(),
-            content_type=content_type,
+            body=body,
             headers=headers or {},
+            content_type=content_type,
             delivery_mode=delivery_mode
         )
 
-        # 发布消息（带重试机制）
-        retry_count = 0
-        max_retries = 2
-        while retry_count < max_retries:
-            try:
-                # 从池获取新通道用于发布（避免长时间占用消费通道）
-                async with self.connection_pool.channel_pool.acquire() as publish_channel:
-                    exchange = await publish_channel.get_exchange(self.exchange_name)
-                    confirmed = await exchange.publish(
-                        message,
-                        routing_key=routing_key or self.routing_key or '#',
-                        mandatory=True,
-                        timeout=5.0
-                    )
-                    if not confirmed:
-                        raise Exception("消息未被服务器确认接收")
-
-                self._update_activity_timestamp()
-                logger.debug(f"消息已发布到交换机 '{self.exchange_name}'")
-                return
-            except (ConnectionClosed, ChannelInvalidStateError):
-                retry_count += 1
-                logger.warning(f"连接已关闭，尝试重连后重新发布 (重试次数: {retry_count})")
-                await self.connect(force_reconnect=True)
-            except Exception as e:
-                retry_count += 1
-                logger.error(f"消息发布失败 (重试次数: {retry_count}): {str(e)}")
-                if retry_count < max_retries:
-                    await asyncio.sleep(1)
-
-        raise Exception(f"消息发布失败，经过{retry_count}次重试仍未成功")
-
-    # 以下方法（消息消费相关）逻辑与原有保持一致，仅适配连接池
-    def set_message_handler(self, handler):
-        self.message_handler = handler
-
-    async def start_consuming(self) -> ConsumerTag:
-        if self._is_consuming:
-            logger.debug("已经在消费中，返回现有consumer_tag")
-            if self._consumer_tag:
-                return self._consumer_tag
-            # 如果_is_consuming为True但无consumer_tag，重置状态并重新启动
-            logger.warning("检测到消费状态异常（无consumer_tag），重置状态后重试")
-            self._is_consuming = False  # 重置状态
-
-        if not self.is_connected:
-            await self.connect()
-
-        if not self.queue:
-            raise Exception("队列未初始化，无法开始消费")
-
-        if not self.message_handler:
-            raise Exception("未设置消息处理函数")
-
-        self._is_consuming = True
-        logger.info(f"开始消费队列: {self.actual_queue_name}")
-
+        # 发布消息
         try:
-            # 调用队列的consume方法，确保获取到consumer_tag
-            self._consumer_tag = await self.queue.consume(
-                self._message_wrapper,
-                no_ack=False  # 手动确认消息
-            )
-
-            # 确保consumer_tag有效
-            if not self._consumer_tag:
-                raise Exception("未能获取到有效的consumer_tag")
-
-            logger.info(
-                f"消费者已启动，队列: {self.actual_queue_name}, tag: {self._consumer_tag}")
-            return self._consumer_tag
+            async with self._connect_lock:
+                if not self._exchange:
+                    # 交换机未初始化，重新声明
+                    logger.warning("交换机未初始化，重新声明")
+                    self._exchange = await self._channel.declare_exchange(
+                        name=self.exchange_name,
+                        type=self.exchange_type,
+                        durable=self.durable,
+                        auto_delete=self.auto_delete
+                    )
+                await self._exchange.publish(
+                    message=message,
+                    routing_key=self.routing_key or self.queue_name or "#"
+                )
+            logger.debug(f"消息发布成功（routing_key: {self.routing_key}）")
         except Exception as e:
-            self._is_consuming = False  # 异常时重置状态
-            logger.error(f"启动消费失败: {str(e)}", exc_info=True)
+            logger.error(f"发布消息失败: {str(e)}", exc_info=True)
+            # 发布失败时清理状态，下次自动重连
+            self._exchange = None
             raise
 
-    async def _safe_cancel_consumer(self) -> bool:
-        if not self._consumer_tag or not self.queue or not self.channel:
-            return True
-
-        try:
-            await asyncio.wait_for(
-                self.queue.cancel(self._consumer_tag),
-                timeout=self.rpc_timeout
-            )
-            logger.info(f"消费者 {self._consumer_tag} 已取消")
-            return True
-        except Exception as e:
-            logger.error(f"取消消费者异常: {str(e)}")
-            return False
-
-    async def stop_consuming(self) -> None:
-        if not self._is_consuming:
-            return
-
-        self._is_consuming = False
-
-        if self._consumer_tag and self.queue:
-            await self._safe_cancel_consumer()
-
-        # 等待所有正在处理的消息完成
-        if self._processing_message_ids:
-            logger.info(
-                f"等待 {len(self._processing_message_ids)} 个正在处理的消息完成...")
-            while self._processing_message_ids and not self._closed:
-                await asyncio.sleep(0.1)
-
-        # 清理状态
-        self._consumer_tag = None
-        self._processing_message_ids.clear()
-
-        logger.info(f"已停止消费队列: {self.actual_queue_name}")
-
-    async def _parse_message(self, message: AbstractIncomingMessage) -> Union[Dict[str, Any], str]:
-        try:
-            body_str = message.body.decode('utf-8')
-            self._update_activity_timestamp()
-
-            if self.auto_parse_json:
-                return json.loads(body_str)
-            return body_str
-        except json.JSONDecodeError:
-            logger.warning(f"消息解析JSON失败，返回原始字符串")
-            return body_str
-        except Exception as e:
-            logger.error(f"消息解析出错: {str(e)}")
-            return message.body.decode('utf-8')
-
-    async def _message_wrapper(self, message: AbstractIncomingMessage) -> None:
-        if not self.message_handler or not self._is_consuming:
-            logger.warning("未设置消息处理器或已停止消费")
-            # await message.ack()
-            try:
-                await message.reject(requeue=True)
-            except Exception as e:
-                logger.error(f"拒绝消息失败: {e}")
-            return
-
-        message_id = message.message_id or str(id(message))
-        if message_id in self._processing_message_ids:
-            logger.warning(f"检测到重复处理的消息ID: {message_id}，直接确认")
-            await message.ack()
+    async def close(self) -> None:
+        """关闭客户端（释放通道）"""
+        if self._closed:
             return
 
-        self._processing_message_ids.add(message_id)
-
-        try:
-            logger.debug(f"收到队列 {self.actual_queue_name} 的消息: {message_id}")
+        self._closed = True
+        logger.info("开始关闭RabbitMQ客户端...")
 
-            parsed_data = await self._parse_message(message)
-            await self.message_handler(MQMsgModel(** parsed_data), message)
+        # 先停止消费
+        await self.stop_consuming()
 
-            await message.ack()
-            self._update_activity_timestamp()
-            self._update_message_processed_timestamp()
-            logger.debug(f"消息 {message_id} 处理完成并确认")
+        # 释放通道到连接池
+        async with self._connect_lock:
+            if self._channel and not self._channel.is_closed:
+                try:
+                    await self.connection_pool.release_channel(self._channel)
+                except Exception as e:
+                    logger.warning(f"释放通道失败: {str(e)}")
+            self._channel = None
+            self._exchange = None
+            self._queue = None
 
-        except Exception as e:
-            current_headers = message.headers or {}
-            retry_count = current_headers.get('x-retry-count', 0)
-            retry_count += 1
-
-            logger.error(
-                f"消息 {message_id} 处理出错（第{retry_count}次重试）: {str(e)}",
-                exc_info=True
-            )
-
-            if retry_count >= MAX_RETRY_COUNT:
-                logger.error(
-                    f"消息 {message_id} 已达到最大重试次数({MAX_RETRY_COUNT}次)，标记为失败")
-                await message.ack()
-                self._update_activity_timestamp()
-                return
-
-            new_headers = current_headers.copy()
-            new_headers['x-retry-count'] = retry_count
-
-            new_message = Message(
-                body=message.body,
-                content_type=message.content_type,
-                headers=new_headers,
-                delivery_mode=message.delivery_mode
-            )
-
-            await message.reject(requeue=False)
-
-            if self.exchange:
-                await self.exchange.publish(
-                    new_message,
-                    routing_key=self.routing_key or '#',
-                    mandatory=True,
-                    timeout=5.0
-                )
-                self._update_activity_timestamp()
-                logger.info(f"消息 {message_id} 已重新发布，当前重试次数: {retry_count}")
-        finally:
-            if message_id in self._processing_message_ids:
-                self._processing_message_ids.remove(message_id)
-
-    async def __aenter__(self):
-        await self.connect()
-        return self
-
-    async def __aexit__(self, exc_type, exc, tb):
-        await self.close()
+        logger.info("RabbitMQ客户端已关闭")