RabbitMQ Work Queues(工作队列)

  • RabbitMQ Work Queues(工作队列)

    • 工作队列模式为一个生产者对应多个消费者,但是只有一个消费者获得消息,即一个队列被多个消费者监听,但一条消息只能被其中的一个消费者获取

    • 代码如下:

      生产者代码:

public class WorkSend {

    private final static String QUEUE_NAME = "hello";

    public static void main(String[] args) throws Exception {
        //获取连接
        Connection connection = ConnectionUtil.getConnection("localhost", 5672, "/", "guest", "guest");
        //声明通道
        Channel channel = connection.createChannel();
        //创建队列
        channel.queueDeclare(QUEUE_NAME, false, false, false, null);
        channel.basicPublish("", "hello", null, "First.".getBytes());
        channel.basicPublish("", "hello", null, "Secode..".getBytes());
        channel.basicPublish("", "hello", null, "Third....".getBytes());
        channel.basicPublish("", "hello", null, "Fourth....".getBytes());
        channel.basicPublish("", "hello", null, "Fifth.....".getBytes());
        //6、关闭通道和连接
        channel.close();
        connection.close();
    }
}

​ 消费者代码

public class WorkRecv {

    private final static String QUEUE_NAME = "hello";

    public static void main(String[] args) throws Exception {
        //获取连接
        Connection connection = ConnectionUtil.getConnection("localhost", 5672, "/", "guest", "guest");

        //声明通道
        Channel channel = connection.createChannel();

        //声明队列队列
        channel.queueDeclare(QUEUE_NAME, false, false, false, null);


        DeliverCallback deliverCallback = (consumerTag, delivery) -> {
            String message = new String(delivery.getBody(), "UTF-8");

            System.out.println(" [x] Received '" + message + "'");
            try {
                doWork(message);
            } catch (InterruptedException e) {
                e.printStackTrace();
            } finally {
                System.out.println(" [x] Done");
                //channel.basicAck();
            }
        };
        boolean autoAck = true; // acknowledgment is covered below
        channel.basicConsume(QUEUE_NAME, autoAck, deliverCallback, consumerTag -> {
        });


//        DeliverCallback deliverCallback = new DeliverCallback(){
//            @Override
//            public void handle(String consumerTag, Delivery delivery) throws IOException {
//                String message = new String(delivery.getBody(), "UTF-8");
//                System.out.println(" [x] Received '" + message + "'");
//            }
//        };
//
//        channel.basicConsume(QUEUE_NAME, true, deliverCallback, new CancelCallback(){
//            @Override
//            public void handle(String consumerTag) throws IOException {
//
//            }
//        });

    }

    private static void doWork(String task) throws InterruptedException {
        for (char ch : task.toCharArray()) {
            if (ch == '.') Thread.sleep(1000);
        }
    }
}

1 生产者将消息交个交换机
2 交换机交给绑定的队列
3 队列由多个消费者同时监听,只有其中一个能够获取这一条消息,形成了资源的争抢,谁的资源空闲大,争抢到的可能越大;

  • Round-robin dispatching(轮询分发)

    使用任务队列的优点之一是能够轻松并行工作。如果我们这里积压了很多的消息,我们可以增加work的并行度,这样就可以轻松扩展。

    默认情况下,RabbitMQ将每个消息依次发送给下一个消费者。平均而言,每个消费者都会收到相同数量的消息。这种分发消息的方式称为轮询。

  • Message acknowledgment(消息确认)

    使用我们当前的代码,RabbitMQ一旦向消费者发送了一条消息,便立即将其标记为删除。在这种情况下,如果我们kill掉一个worker,我们将丢失正在处理的消息。并且还将丢失所有发送给该特定worker但尚未处理的消息。
    为了确保消息永不丢失,RabbitMQ支持 消息确认。消费者发送回一个确认(告知),告知RabbitMQ特定的消息已被接收并处理,并且RabbitMQ可以自由删除它。
    如果消费者死了(其通道已关闭,连接已关闭或TCP连接丢失)而没有发送确认,RabbitMQ将了解消息未完全处理,并将重新排队。如果同时有其他消费者在线,它将很快将其重新分发给另一个消费者。这样,您可以确保即使worker偶尔死亡也不会丢失任何消息。
    没有任何消息超时;消费者死亡时,RabbitMQ将重新传递消息。即使处理一条消息花费非常非常长的时间也没关系。
    默认情况下,手动消息确认处于打开状态。我们通过autoAck = false 标志显式关闭了它们。在消息完成投递的时候,手动确认消息投递成功。

    消费者代码修改如下:

public class ManWorkRecv {

    private final static String QUEUE_NAME = "hello";

    public static void main(String[] args) throws Exception {
        //获取连接
        Connection connection = ConnectionUtil.getConnection("localhost", 5672, "/", "guest", "guest");

        //声明通道
        Channel channel = connection.createChannel();

        //声明队列队列
        channel.queueDeclare(QUEUE_NAME, false, false, false, null);


        DeliverCallback deliverCallback = (consumerTag, delivery) -> {
            String message = new String(delivery.getBody(), "UTF-8");

            System.out.println(" [x] Received '" + message + "'");
            try {
                doWork(message);
            } catch (InterruptedException e) {
                e.printStackTrace();
            } finally {
                System.out.println(" [x] Done");
                //手动确认消息已经成功投递
                channel.basicAck(delivery.getEnvelope().getDeliveryTag(), false);
            }
        };
        boolean autoAck = false; //设置消息手动确认
        channel.basicConsume(QUEUE_NAME, autoAck, deliverCallback, consumerTag -> {
        });

    }

    private static void doWork(String task) throws InterruptedException {
        for (char ch : task.toCharArray()) {
            if (ch == '.') Thread.sleep(1000);
        }
    }
}

​ 上述代码可以确保我们在kill掉一个消费者的情况下,消息不会丢失。

  • Message durability(消息持久化)

    前面已经讲了即使一个消费者退出,消息不会丢失,但是如果RabbitMQ服务退出时,队列和消息仍然会丢失,这是因为 默认队列和消息都是放在内存中的 。为保证消息和队列不丢失,需要把队列和消息设置为持久化
    确保RabbitMQ永远不会丢失我们的队列。声明持久化代码如下:

    boolean durable = true;
    channel.queueDeclare("hello", durable, false, false, null);
    

    由于RabbitMQ已经存在一个hello队列,并且RabbitMQ不支持对已经存在的队列进行参数修改,所以需要我们删除之前创建的队列或者重新创建一个队列。

    其次,我们需要保证消息的持久化,消息持久化设置如下:

    channel.basicPublish("", "hello", MessageProperties.PERSISTENT_TEXT_PLAIN, "Fifth.....".getBytes());
    

    这样,我们就可以保证RabbitMQ服务宕机的情况下,消息和队列都不会丢失。

  • Fair dispatching(公平分发)

    ​ RabbitMQ在进行消息分发的时候,可以设置一次分发给某一个消费者多少条消息,如在消费者端设置prefetchCount=1;如下代码

    int prefetchCount = 1 ; 
    channel.basicQos(prefetchCount);
    

    该设置表示,RabbitMQ一次分发给消费者一条消息,在消费者处理并确认上一条消息之前,不会再给这个消费者发送一条新消息,而会将其分发给其他的消费者

  • SpringBoot实现:

    @SpringBootApplication
    @EnableScheduling
    public class RabbitAmqpTutorialsApplication {
    
        public static void main(String[] args) throws Exception {
            SpringApplication.run(RabbitAmqpTutorialsApplication.class, args);
        }
    
    }
    
     @Configuration
      public class Tut2Config {
    
          @Bean
          public Queue hello() {
              return new Queue("hello");
          }
          
          /**
           * 这里是启动两个消费者
           */
          private static class ReceiverConfig{
              @Bean
              public Tut2Receiver receiver1(){
                  return new Tut2Receiver(1);
              }
          
              @Bean
              public Tut2Receiver receiver2(){
                  return new Tut2Receiver(2);
              }
          }
          
          @Bean
          public Tut2Sender sender() {
              return new Tut2Sender();
          }
    
    
      }
    
    
    
    //监听队列
    @RabbitListener(queues = "hello")
    public class Tut2Receiver {
    
        private final int instance;
        
        public Tut2Receiver(int i) {
            this.instance = i;
        }
        
        @RabbitHandler
        public void receive(String in) throws InterruptedException {
            StopWatch watch = new StopWatch();
            watch.start();
            System.out.println("instance " + this.instance +
                    " [x] Received '" + in + "'");
            doWork(in);
            watch.stop();
            System.out.println("instance " + this.instance +
                    " [x] Done in " + watch.getTotalTimeSeconds() + "s");
        }
        
        private void doWork(String in) throws InterruptedException {
            for (char ch : in.toCharArray()) {
                if (ch == '.') {
                    Thread.sleep(1000);
                }
            }
        }
    
    }
    
    public class Tut2Sender {
        @Autowired
        private RabbitTemplate template;
    
        //队列
        @Autowired
        private Queue queue;
        
        AtomicInteger dots = new AtomicInteger(0);
        
        AtomicInteger count = new AtomicInteger(0);
        
        /**
         * 定时向队列hello发送消息
         */
        @Scheduled(fixedDelay = 1000, initialDelay = 500)
        public void send() {
            StringBuilder builder = new StringBuilder("Hello");
            if (dots.incrementAndGet() == 3) {
                dots.set(1);
            }
            for (int i = 0; i < dots.get(); i++) {
                builder.append('.');
            }
            builder.append(count.incrementAndGet());
            String message = builder.toString();
            //向队列中发送消息
            template.convertAndSend(queue.getName(), message);
            System.out.println(" [x] Sent '" + message + "'");
        }
    
    }
    

    相关代码链接: https://github.com/albert-liu435/springmq

posted on 2020-02-12 23:08  海之浪子  阅读(340)  评论(0编辑  收藏  举报

导航