amqplib-init 1.1.8 → 1.1.10

package/index.js

@@ -2,10 +2,11 @@ const amqp = require('amqplib');
 const shelljs = require('shelljs');
 const log = require('chalk-style')
 const happy = require('happy-help');
+const axios = require('axios'); // 新增 axios 依赖
 
 module.exports = {
   // 初始化函数
-  init(option) {
+  async init(option) { // 改为 async 函数
     const {
       channelName = 'node-test-channel',    // 频道名称，默认为'node-test-channel'
       prefetch = 1,                         // 预取计数，默认为1
@@ -13,6 +14,7 @@ module.exports = {
       callback = () => {},                  // 消息处理回调，默认为空函数
       finish = () => {},                    // 初始化完成回调，默认为空函数
       amqpLink = '',                        // RabbitMQ连接地址
+      amqpAutoLink = '',                    // 自动获取连接信息的HTTPS链接
       heartbeat = 5,                        // 心跳间隔，单位为秒，默认为2秒
       timeout = 2000,                     // 连接超时时间，单位为毫秒，默认为10000毫秒
       delay = 0,                            // 消息处理完成后延迟ack的时间，单位为毫秒，默认为0毫秒
@@ -24,11 +26,31 @@ module.exports = {
     let connection = null;                  // RabbitMQ连接
     let channel = null;                    // RabbitMQ频道
 
+    // 处理 amqpAutoLink 自动获取连接信息
+    let finalAmqpLink = amqpLink; // 最终使用的连接地址
+    if (amqpAutoLink) {
+      try {
+        log.log(`正在从 ${amqpAutoLink} 获取连接信息...`);
+        const response = await axios.post(amqpAutoLink);
+        const { info } = response.data;
+        
+        if (info && info.AMQPLIB_USER && info.AMQPLIB_PWD && info.AMQPLIB_IP && info.AMQPLIB_PORT) {
+          finalAmqpLink = `amqp://${info.AMQPLIB_USER}:${info.AMQPLIB_PWD}@${info.AMQPLIB_IP}:${info.AMQPLIB_PORT}`;
+          log.log(`✅ 自动获取连接信息成功: ${finalAmqpLink}`);
+        } else {
+          log.error('❌ 获取的连接信息不完整，使用默认连接地址');
+        }
+      } catch (error) {
+        log.error('❌ 获取连接信息失败:', error.message);
+        log.log('使用默认连接地址继续...');
+      }
+    }
+
     // 重连函数
     const reconnect = async () => {
       log.error('连接丢失，正在尝试重连...');
       await happy.sleep(2); // 重连前等待2秒
-      connection = await amqp.connect(amqpLink, { heartbeat, timeout });
+      connection = await amqp.connect(finalAmqpLink, { heartbeat, timeout });
       channel = await connection.createChannel();
       await channel.assertQueue(channelName, { durable });
       await channel.prefetch(prefetch);
@@ -44,18 +66,29 @@ module.exports = {
         if (msg !== null) {
           try {
             const content = JSON.parse(msg.content.toString());
-            log.log(`收到消息: ${JSON.stringify(content)}`);
+            log.log(`🪴 队列收到消息: ${JSON.stringify(content)}`);
             const startTime = Date.now();
-            await callback(content); // 执行消息处理回调
-            const endTime = Date.now() - startTime;
-            log.log(`消息处理完成，延迟: ${delay}ms，总时间: ${endTime}ms`);
-            setTimeout(() => channel.ack(msg), delay); // 延迟ack
+            // 执行消息处理回调
+            callback(content).then(() => {
+              const endTime = Date.now() - startTime;
+              log.log(`☘️ 消息处理完成，延迟: ${delay}ms，总时间: ${endTime}ms`);
+              setTimeout(() => channel.ack(msg), delay); // 延迟ack
+            }).catch(async (e) => {
+              // 拒绝消息并重新排队
+              log.log('‼️ 处理消息返回错误:', e);
+              await happy.sleep(5);
+              channel.reject(msg, true);
+            });
           } catch (e) {
-            log.error('处理消息时出错:', e);
-            channel.reject(msg, true); // 拒绝消息并重新排队
+            // 拒绝消息并重新排队
+            log.log('‼️ 处理消息时出错:', e.message);
+            await happy.sleep(5);
+            channel.reject(msg, true);
           }
         } else {
-          log.error('收到无效消息:', msg);
+          // 直接消息此消息;
+          channel.ack(msg);
+          log.error('收到无效消息(自动消费):', msg);
         }
       }, { noAck: false });
     };
@@ -63,7 +96,7 @@ module.exports = {
     // 启动函数
     const start = async () => {
       try {
-        connection = await amqp.connect(amqpLink, { heartbeat, timeout });
+        connection = await amqp.connect(finalAmqpLink, { heartbeat, timeout });
         channel = await connection.createChannel();
         await channel.assertQueue(channelName, { durable });
         await channel.prefetch(prefetch);
@@ -98,7 +131,6 @@ module.exports = {
             }
           }, autoReload);
         }
-
         finish(); // 执行初始化完成回调
       } catch (e) {
         log.error('初始化RabbitMQ连接时出错:', e);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "amqplib-init",
-  "version": "1.1.8",
+  "version": "1.1.10",
   "description": "消息队列初始化",
   "main": "index.js",
   "scripts": {
@@ -10,7 +10,8 @@
   "license": "ISC",
   "dependencies": {
     "amqplib": "^0.10.4",
-    "chalk-style": "^1.0.0",
+    "axios": "^1.6.0",
+    "chalk-style": "^1.0.3",
     "happy-help": "^1.0.1",
     "js-base64": "^3.7.7",
     "shelljs": "^0.8.5"