javascript - 使用 Node.js 使用 RabbitMQ 消息时，我可以等待进程完成吗？

Question

我对 Node.js 和 ES6 还很陌生，这让我有点困惑。我试图让一个进程运行，使用来自 RabbitMQ 队列的消息。它需要能够在抓取下一条消息之前处理该消息(大约需要 30-60 秒)。目前，我拥有的代码，它可以抓取所有消息，然后尝试 fork 进程。当队列中有 3-5 条消息时，这很好，但对于 20、50 或 100 条消息，这会导致服务器内存不足。

我试过制作 .consume()回调函数异步并添加await到消息处理功能。我试过包装 await new Promise在 .consume() 内回调 processMessage .我已尝试添加 await到调用 channel.consume 的行.没有什么能改变行为。

#!/usr/bin/env node

const amqp = require('amqplib');

const consumeFromQueue = async (queue, isNoAck = false, durable = false, prefetch = null) => {
    const conn_str = "amqp://" + process.env.RABBITMQ_USERNAME + ":" + process.env.RABBITMQ_PASSWORD + "@" + process.env.RABBITMQ_HOST + "/development?heartbeat=60"
    const cluster = await amqp.connect(conn_str);
    const channel = await cluster.createChannel();
    await channel.assertQueue(queue,  { durable: durable, autoDelete: true });
    if (prefetch) {
        channel.prefetch(prefetch);
    }
    console.log(` [x] Waiting for messages in ${queue}. To exit press CTRL+C`)

    try {
        channel.consume(queue, message => {
            if (message !== null) {
                console.log(' [x] Received', message.content.toString());
                processMessage(message.content.toString());
                channel.ack(message);
                return null;
            } else {
                console.log(error, 'Queue is empty!')
                channel.reject(message);
            }
        }, {noAck: isNoAck});
    } catch (error) {
        console.log(error, 'Failed to consume messages from Queue!')
        cluster.close(); 
    }
}

exports.consumeFromQueue = consumeFromQueue;

作为旁注，如果我创建一个字符串数组并循环遍历字符串，当我将等待添加到 processMessage行，它在处理下一个字符串之前等待执行过程(30-60 秒)。

(async () => {
    for (let i=0; i<urls.length; i++) {
        await processMessage(urls[i]);
    }
})();

所以我基本上需要这样的功能，但要在 RabbitMQ 中监听队列。

Answer 1

如果您想限制消费者在任何给定时间处理的消息数量，请使用 channel.prefetch() :

The count given is the maximum number of messages sent over the channel that can be awaiting acknowledgement; once there are count messages outstanding, the server will not send more messages on this channel until one or more have been acknowledged.

也就是说，如果您只想在处理下一条消息之前一次处理一条消息，请设置 channel.prefetch(1)