第三章-高级特性4:消费端限流

什么是消费端的限流?

  • 假设一个场景,首先,我们RabbitMQ服务器有成千上万条未处理的消息, 我们随便打开一个消费者客户端,会出现下面情况:

    • 巨量的消息瞬间全部推送过来,但是我们单个客户端无法同时处理这么多数据!
    • 轻则消费端卡顿。重则宕机
  • RabbitMQ提供了一种qos(服务质量保证)功能,即在非自动确认消息的前提下,如果一定数目的消息(通过基于Consumer或者Channel设置Qos的值) 未被确认前,不进行消费新的消息。


 /**
 * @param prefetchSize  消息大小限制 一般为0 不限制
 * @param prefetchCount 一次最多处理消息个数 
 *        会告诉RabbitMQ不要同时给一个消费者推送多余N个消息,即一旦有N个消息还没有ack,则该consumer将block掉。直到有消息ack
 * @param global       true/false 是否将上面设置应用于chammel。
  *                                简单说。就是上面限制是channel级别还是consumer级别。
 */
void basicQos(int prefetchSize, int prefetchCount, boolean global)

  • prefetchSizeglobal这两项,rabbitmq没有实现,暂且不研究,prefetch_count在no_ask=false的情况下生效,即在自动应答的情况下这两个值是不生效的

/**
 * @author yangHX
 * createTime  2019/3/27 22:02
 */
public class Producer {

    public static void main(String[] args) throws IOException, TimeoutException {
        ConnectionFactory connectionFactory = RabbitMqUtil.getConnectionFactory();
        Connection connection = connectionFactory.newConnection();
        Channel channel = connection.createChannel();

        String exchangeName = "test_qos_exchange";
        String routingKey = "qos.save";

        String msg = "Hello RabbitMQ QOS Message";
        for (int i = 0; i < 5; i++) {
            channel.basicPublish(exchangeName, routingKey, true, null, msg.getBytes());
        }
    }
}



/**
 * @author yangHX
 * createTime  2019/3/27 22:06
 */
public class Consumer {


    public static void main(String[] args) throws IOException, TimeoutException {
        ConnectionFactory connectionFactory = RabbitMqUtil.getConnectionFactory();
        Connection connection = connectionFactory.newConnection();
        Channel channel = connection.createChannel();

        String exchangeName = "test_qos_exchange";
        String queueName = "test_qos_queue";
        String routingKey = "qos.#";

        channel.exchangeDeclare(exchangeName, "topic", true, false, null);
        channel.queueDeclare(queueName, true, false, false, null);
        channel.queueBind(queueName, exchangeName, routingKey);


        //限流方式 第一件事就是将 autoAck 设置为false

        /**
         * @param prefetchSize  消息大小限制 一般为0 不限制
         * @param prefetchCount 一次最多处理消息个数
         *        会告诉RabbitMQ不要同时给一个消费者推送多余N个消息,即一旦有N个消息还没有ack,则该consumer将block掉。直到有消息ack
         * @param global       true/false 是否将上面设置应用于channel。
         *                                简单说。就是上面限制是channel级别还是consumer级别。
         */
        channel.basicQos(0, 1, false);

        channel.basicConsume(queueName, false, new MyConsumer(channel));


    }
}



/**
 * @author yangHX
 * createTime  2019/3/27 22:12
 */
public class MyConsumer extends DefaultConsumer {
    private Channel channel;


    MyConsumer(Channel channel) {
        super(channel);
        this.channel = channel;
    }

    @Override
    public void handleDelivery(String consumerTag, Envelope envelope, AMQP.BasicProperties properties, byte[] body) throws IOException {
        System.err.println("-----------consume message----------");
        System.err.println("consumerTag: " + consumerTag);
        System.err.println("envelope: " + envelope);
        System.err.println("properties: " + properties);
        System.err.println("body: " + new String(body));

        //ack应答
        channel.basicAck(envelope.getDeliveryTag(), false);
    }
}


最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。