侧边栏壁纸
博主头像
落叶人生博主等级

走进秋风,寻找秋天的落叶

  • 累计撰写 130562 篇文章
  • 累计创建 28 个标签
  • 累计收到 9 条评论
标签搜索

目 录CONTENT

文章目录

rabbitmq——prefetch count

2022-07-01 星期五 / 0 评论 / 0 点赞 / 88 阅读 / 6430 字

消费者在开启acknowledge的情况下,对接收到的消息可以根据业务的需要异步对消息进行确认。 然而在实际使用过程中,由于消费者自身处理能力有限,从rabbitmq获取一定数量的消息后,希望ra

消费者在开启acknowledge的情况下,对接收到的消息可以根据业务的需要异步对消息进行确认。

然而在实际使用过程中,由于消费者自身处理能力有限,从rabbitmq获取一定数量的消息后,希望rabbitmq不再将队列中的消息推送过来,当对消息处理完后(即对消息进行了ack,并且有能力处理更多的消息)再接收来自队列的消息。在这种场景下,我们可以通过设置basic.qos信令中的prefetch_count来达到这种效果。

先直观的看看设置了prefetch_count的效果,:

1) 对比测试:两个消费者都订阅同一队列,no_ack均设置为false即开启acknowledge机制,且均未设置prefetch_count,向队列发布5条消息

结果:不管消息是否被ack,rabbitmq会轮流向两个消费者投递消息,第一个消费者收到"1","3","5"三条消息, 第二个消费者收到"2","4"两条消息。

2)prefetch_count设置测试:两个消费者都订阅同一队列,开启acknowledge机制,第一个消费者prefetch_count设置为1,另一个消费者未设置prefetch_count,同样向队列发布5条消息

结果:rabbitmq向第一个消费者投递了一条消息后,消费者未对该消息进行ack,rabbitmq不会再向该消费者投递消息,剩下的四条消息均投递给了第二个消费者

看完效果后,再来看看rabbitmq里的一些实现。

1. rabbitmq对basic.qos信令的处理

首先,basic.qos是针对channel进行设置的,也就是说只有在channel建立之后才能发送basic.qos信令。

在rabbitmq的实现中,每个channel都对应会有一个rabbit_limiter进程,当收到basic.qos信令后,在rabbit_limiter进程中记录信令中prefetch_count的值,同时记录的还有该channel未ack的消息个数。

注:其实basic.qos里还有另外两个参数可进行设置(global和prefetch_size),但rabbitmq没有相应的实现。

2. 队列中的消息投递给消费者时的处理

当rabbitmq要将队列中的一条消息投递给消费者时,会遍历该队列上的消费者列表,选一个合适的消费者,然后将消息投递出去。其中挑选消费者的一个依据就是看消费者对应的channel上未ack的消息数是否达到设置的prefetch_count个数,如果未ack的消息数达到了prefetch_count的个数,则不符合要求。当挑选到合适的消费者后,中断后续的遍历。

rabbit_amqqueue_process.erldeliver_msgs_to_consumers(_DeliverFun, true, State) ->    {true, State};deliver_msgs_to_consumers(DeliverFun, false,                          State = #q{active_consumers =                                     ActiveConsumers}) ->    case priority_queue:out_p(ActiveConsumers) of        {empty, _} ->            {false, State};        {{value, QEntry, Priority}, Tail} ->            {Stop, State1} =                deliver_msg_to_consumer(DeliverFun, QEntry,                                        Priority,                                        State#q{active_consumers =                                                Tail}),            %%如果处理结果为false,遍历下一个消费者            deliver_msgs_to_consumers(DeliverFun, Stop, State1)    end.deliver_msg_to_consumer(DeliverFun, E = {ChPid, Consumer},                        Priority, State) ->    ...    %%判断是否可以将消息投递给该消费者    case rabbit_limiter:can_send(C#cr.limiter,                                 Consumer#consumer.ack_required,                                 Consumer#consumer.tag) of        %%可以投递,再将该消费者放到队列的尾部        {continue, Limiter} ->            AC1 = priority_queue:in(E, Priority,                                    State#q.active_consumers),            %%将消息投递给消费者            deliver_msg_to_consumer0(DeliverFun, Consumer,                                     C#cr{limiter = Limiter},                                     State#q{active_consumers = AC1})    ...rabbit_limiter.erlhandle_call({can_send, QPid, AckRequired}, _From,            State = #lim{volume = Volume}) ->    case prefetch_limit_reached(State) of        %%未ack的消息数达到prefetch_count设置的个数        true  -> {reply, false, limit_queue(QPid, State)};        false -> {reply, true,                  %%消息需要被ACK, volume加1                  State#lim{volume = if AckRequired -> Volume + 1;                                        true        -> Volume                                     end}}    endprefetch_limit_reached(#lim{prefetch_count = Limit,                             volume = Volume}) ->    Limit =/= 0 andalso Volume >= Limit.



3. 消费者对消息ack后的处理

当消费者对消息进行ack后,最终会修改该消费者对应channel中未ack的消息数,这样队列又可以将消息投递给该消费者。

rabbit_limiter.erlhandle_cast({ack, Count}, State = #lim{volume = Volume}) ->    NewVolume = if Volume == 0 -> 0;                   true        -> Volume - Count                end,    {noreply, maybe_notify(State, State#lim{volume = NewVolume})};



4. 扩展

在AMQP协议(0-9-1)中,有这么一段话

对于rabbitmq来说,最后一句话其实说的就是使用了acknowledge机制情况下,使用prefetch_count进行流量控制。另外在实际研究过程中发现还有channel.flow以及basic.credit(应该属于AMQP 1.0协议)可以进行一些控制,这里没有展开,有时间会研究下相应的机制以及可能使用的场景。

广告 广告

评论区