amqplib-init 1.0.0

package/index.js

@@ -0,0 +1,87 @@
+const amqp = require('amqplib');
+const shelljs = require('shelljs');
+const sleep = (time) => {
+  return new Promise(resolve => {
+    setTimeout(_ => resolve(), time * 1000)
+  })
+};
+module.exports = {
+  /**
+   * 消息队列设置
+   * @returns {Promise<unknown>}
+   * @param option
+   */
+  init (option){
+    const {
+      channelName = 'node-test-channel',
+      prefetch = 1,
+      callback = Function,
+      pmId = 0,
+      finish = Function,
+      amqpLink = ``,
+      heartbeat = 5,
+      timeout = 120000,
+    } = option;
+    const durable = true;
+    console.log(`🚀 队列已执行...`);
+    let connect = null;
+    const amqpInit = async (reload = false) => {
+      await sleep(1); // 等待一秒钟
+      connect = await amqp.connect(amqpLink, {
+        heartbeat, // 心跳检测5秒
+        timeout, // 设置连接超时时间为120秒
+      });
+      if (pmId > 0 && reload){
+        console.log(`‼️ 服务即将重启`);
+        shelljs.exec(`pm2 reload ${pmId}`);
+      }
+    }
+    return new Promise(async (resolve) => {
+      await amqpInit();
+      console.log(`✳️->RabbitMQ链接成功`);
+      const channel = await connect.createChannel();
+      console.log(`✳️->数据队列名称`, channelName);
+      console.log(`✳️->是否持久化`, durable);
+      const { queue } = await channel.assertQueue(channelName, { durable });
+      // 一次性读取一条;
+      await channel.prefetch(prefetch);
+      console.log(`✳️->一次取多少条`, prefetch);
+      console.log(`✳️->消息队列: ${channelName} 正在运行中...`);
+      await channel.consume(queue, async (msg) => {
+        if (msg !== null) {
+          try {
+            await channel.assertQueue(channelName, { durable });
+            const content = JSON.parse(msg?.content?.toString());
+            console.log(`✳️->队列收到新消息: ${channelName}`);
+            const startTime = new Date().getTime();
+            callback(content).then(()=> {
+              try {
+                channel.ack(msg);
+                const endTime = (new Date().getTime() - startTime);
+                console.log(`✳️->消息处理完成, 共耗时: ${endTime}ms`);
+              } catch (e) {
+                console.log(`队列消费遇到异常`, e);
+              }
+            });
+          } catch (e) {
+            console.log(`队列遇到异常`, e);
+          }
+        } else {
+          console.log(`消息异常`, msg);
+        }
+      });
+      // 队列遇到错误
+      connect.on('error', (err) => {
+        console.log('Connection error', err.message);
+        amqpInit(true);
+      });
+      // 队列被关闭
+      connect.on('close', function() {
+        console.log('Connection to RabbitMQ is closed');
+        amqpInit(true);
+      });
+      console.log(`✳️-队列已启动: finish~`);
+      finish && finish();
+    })
+  }
+}

package/package.json

@@ -0,0 +1,15 @@
+{
+  "name": "amqplib-init",
+  "version": "1.0.0",
+  "description": "消息队列初始化",
+  "main": "index.js",
+  "scripts": {
+    "start": "node index.js"
+  },
+  "author": "",
+  "license": "ISC",
+  "dependencies": {
+    "amqplib": "^0.10.4",
+    "shelljs": "^0.8.5"
+  }
+}