RabbitMQ高级特性消费端限流策略实现

àì夳堔傛蜴生んèń 2022-01-20 03:03 242阅读 0赞

> 应用范围为服务访问量突然剧增，原因可能有多种外部的调用或内部的一些问题导致消息积压，对服务的访问超过服务所能处理的最大峰值，导致系统超时负载从而崩溃。

### 业务场景 ###

举一些我们平常生活中的消费场景，例如：火车票、机票、门票等，通常来说这些服务在下单之后，后续的出票结果都是异步通知的，如果服务本身只支持每秒1000访问量，由于外部服务的原因突然访问量增加到每秒2000并发，这个时候服务接收者因为流量的剧增，超过了自己系统本身所能处理的最大峰值，如果没有对消息做限流措施，系统在这段时间内就会造成不可用，在生产环境这是一个很`严重`的问题，实际应用场景不止于这些，本文通过RabbitMQ来讲解如果对消费端做限流措施。

### 消费端限流机制 ###

RabbitMQ提供了服务质量保证 (`QOS`) 功能，对channel（通道）预先设置一定的消息数目，每次发送的消息条数都是基于预先设置的数目，如果消费端一旦有未确认的消息，这时服务端将不会再发送新的消费消息，直到消费端将消息进行完全确认，注意：此时消费端不能设置自动签收，否则会无效。

在 `RabbitMQ v3.3.0` 之后，放宽了限制，除了对channel设置之外，还可以对每个消费者进行设置。

以下为 Node.js 开发语言 `amqplib` 库对于限流实现提供的接口方法 `prefetch`

export interface Channel extends events.EventEmitter {
        prefetch(count: number, global?: boolean): Promise<Replies.Empty>;
        ...
    }
    复制代码

**prefetch 参数说明**：

*  number：每次推送给消费端 N 条消息数目，如果这 N 条消息没有被ack，生产端将不会再次推送直到这 N 条消息被消费。
 *  global：在哪个级别上做限制，ture 为 channel 上做限制，false 为消费端上做限制，默认为 false。

### 建立生产端 ###

生产端没什么变化，和正常声明一样，关于源码参见[rabbitmq-prefetch（Node.js客户端版Demo）][rabbitmq-prefetch_Node.js_Demo]

### 建立消费端 ###

*  **未确认消息情况测试**

在 consumer 中我们暂且将 `channel.ack(msg)` 注释掉，分别启动生产者和消费者，看看是什么情况？

如上图所示，总共5条消息按照预先设置的发送了一条消息，因为我将 `channel.ack(msg)` 注释掉了，服务端在未得到 ack 确认，将不会在发送剩下已 Ready 消息。

*  **确认消息测试**

修改 consumer 代码，打开确认消息注释，重新启动消费端进行测试

await channel.consume(queueName, msg => {
        console.log('Consumer：', msg.content.toString());
    
        channel.ack(msg); // 打开注释
    }, { noAck: false });
    复制代码

如上图所示，Unacked 为0，消息已全部消费成功。

### RabbitMQ限流使用总结 ###

限流在我们的实际工作中还是很有意义的，在使用上生产端没有变化，重点在消费端，着重看以下两点：

*  限流情况 ack 不能设置自动签收，修改 `{ noAck: false }`
 *  增加限流参数设置 `channel.prefetch(1, false)`

## 资料 ##

*  个人博客： [Node.js技术栈][Node.js]
 *  RabbitMQ系列：[RabbitMQ高级消息队列系列文章不断更新中][RabbitMQ]
 *  项目源码： [rabbitmq-prefetch（Node.js客户端版Demo）][rabbitmq-prefetch_Node.js_Demo]

作者：五月君  
链接：[www.imooc.com/article/287…][www.imooc.com_article_287]  
来源：慕课网

[rabbitmq-prefetch_Node.js_Demo]: https://link.juejin.im?target=https%3A%2F%2Fgithub.com%2FQ-Angelo%2Fproject-training%2Ftree%2Fmaster%2Fnodejs%2Frabbitmq-prefetch
[Node.js]: https://link.juejin.im?target=https%3A%2F%2Fgithub.com%2FQ-Angelo%2FNodejs-Roadmap
[RabbitMQ]: https://link.juejin.im?target=https%3A%2F%2Fgithub.com%2FQ-Angelo%2FNodejs-Roadmap%2Fblob%2Fmaster%2Fdocs%2Fmicroservice%2Frabbitmq-base.md
[www.imooc.com_article_287]: https://link.juejin.im?target=https%3A%2F%2Fwww.imooc.com%2Farticle%2F287046