amqplib-init 1.2.4 → 1.3.1

package/module/MessageProcessor.js

@@ -1,187 +0,0 @@
-const log = require('chalk-style');
-const happy = require('happy-help');
-
-/**
- * 消息处理器模块
- * 负责消息处理、状态跟踪、ACK/REJECT逻辑
- */
-class MessageProcessor {
-  constructor(config) {
-    this.config = config;
-    this.processingMessages = new Set(); // 跟踪正在处理的消息
-  }
-
-  /**
-   * 开始消费消息
-   * @param {Object} channel - RabbitMQ频道
-   * @returns {Promise<Function>} 返回清理函数
-   */
-  async startConsuming(channel) {
-    const { channelName, callback, delay } = this.config;
-    
-    log.log(`开始消费: ${channelName}`);
-    
-    await channel.consume(channelName, async (msg) => {
-      if (msg !== null) {
-        await this._handleMessage(msg, channel, callback, delay);
-      } else {
-        // ✅ 修复关键bug：null消息不需要ACK操作
-        log.log('收到null消息，队列可能为空或连接中断');
-        // 移除错误的 channel.ack(msg) 调用
-      }
-    }, { noAck: false });
-    
-    // 返回清理函数，用于重连时清理状态
-    return () => {
-      this.processingMessages.clear();
-      log.log('已清理消息处理状态');
-    };
-  }
-
-  /**
-   * 处理单条消息
-   * @param {Object} msg - RabbitMQ消息对象
-   * @param {Object} channel - RabbitMQ频道
-   * @param {Function} callback - 消息处理回调函数
-   * @param {number} delay - 延迟ACK时间
-   * @private
-   */
-  async _handleMessage(msg, channel, callback, delay) {
-    const deliveryTag = msg.fields.deliveryTag;
-    
-    // 检查消息是否已在处理中
-    if (this.processingMessages.has(deliveryTag)) {
-      log.log(`消息 ${deliveryTag} 已在处理中，跳过`);
-      return;
-    }
-    
-    this.processingMessages.add(deliveryTag);
-    
-    // 检查消息内容有效性，对无效消息直接确认
-    if (!this._isValidMessageContent(msg)) {
-      log.log(`⚠️ 收到无效消息 ${deliveryTag}，内容为空或无法解析，自动确认以防止阻塞`);
-      this._acknowledgeMessage(msg, channel, deliveryTag, 0); // 立即确认无效消息
-      return;
-    }
-    
-    try {
-      const content = JSON.parse(msg.content.toString());
-      log.log(`🪴 队列收到消息: ${JSON.stringify(content)}`);
-      const startTime = Date.now();
-      
-      // 执行消息处理回调
-      callback(content)
-        .then(() => {
-          const endTime = Date.now() - startTime;
-          log.log(`☘️ 消息处理完成，延迟: ${delay}ms，总时间: ${endTime}ms`);
-          this._acknowledgeMessage(msg, channel, deliveryTag, delay);
-        })
-        .catch(async (e) => {
-          log.log('‼️ 处理消息返回错误:', e);
-          await this._rejectMessage(msg, channel, deliveryTag);
-        });
-    } catch (e) {
-      log.log('‼️ 处理消息时出错:', e.message);
-      await this._rejectMessage(msg, channel, deliveryTag);
-    }
-  }
-
-  /**
-   * 检查消息内容是否有效
-   * @param {Object} msg - RabbitMQ消息对象
-   * @returns {boolean} 消息内容是否有效
-   * @private
-   */
-  _isValidMessageContent(msg) {
-    try {
-      // 检查消息对象是否存在
-      if (!msg || !msg.content) {
-        return false;
-      }
-      
-      // 获取消息内容字符串
-      const contentStr = msg.content.toString();
-      
-      // 检查内容是否为空或只包含空白字符
-      if (!contentStr || contentStr.trim() === '') {
-        return false;
-      }
-      
-      // 检查是否为null字符串
-      if (contentStr.toLowerCase() === 'null' || contentStr.toLowerCase() === 'undefined') {
-        return false;
-      }
-      
-      // 尝试解析JSON，如果失败则认为无效
-      JSON.parse(contentStr);
-      return true;
-    } catch (error) {
-      // JSON解析失败，认为消息无效
-      return false;
-    }
-  }
-
-  /**
-   * 确认消息
-   * @param {Object} msg - RabbitMQ消息对象
-   * @param {Object} channel - RabbitMQ频道
-   * @param {string} deliveryTag - 消息标签
-   * @param {number} delay - 延迟时间
-   * @private
-   */
-  _acknowledgeMessage(msg, channel, deliveryTag, delay) {
-    // 确保消息还在处理集合中再进行ACK
-    if (this.processingMessages.has(deliveryTag)) {
-      setTimeout(() => {
-        try {
-          channel.ack(msg);
-          this.processingMessages.delete(deliveryTag);
-          log.log(`✅ 消息 ${deliveryTag} 已确认`);
-        } catch (ackError) {
-          log.error(`ACK消息 ${deliveryTag} 时出错:`, ackError.message);
-          this.processingMessages.delete(deliveryTag);
-        }
-      }, delay);
-    }
-  }
-
-  /**
-   * 拒绝消息并重新排队
-   * @param {Object} msg - RabbitMQ消息对象
-   * @param {Object} channel - RabbitMQ频道
-   * @param {string} deliveryTag - 消息标签
-   * @private
-   */
-  async _rejectMessage(msg, channel, deliveryTag) {
-    await happy.sleep(5);
-    
-    if (this.processingMessages.has(deliveryTag)) {
-      try {
-        channel.reject(msg, true);
-        this.processingMessages.delete(deliveryTag);
-        log.log(`❌ 消息 ${deliveryTag} 已拒绝并重新排队`);
-      } catch (rejectError) {
-        log.error(`拒绝消息 ${deliveryTag} 时出错:`, rejectError.message);
-        this.processingMessages.delete(deliveryTag);
-      }
-    }
-  }
-
-  /**
-   * 清理所有处理中的消息状态
-   */
-  clearProcessingMessages() {
-    this.processingMessages.clear();
-    log.log('已清理所有消息处理状态');
-  }
-
-  /**
-   * 获取当前处理中的消息数量
-   * @returns {number} 处理中的消息数量
-   */
-  getProcessingCount() {
-    return this.processingMessages.size;
-  }
-}
-
-module.exports = MessageProcessor;