RabbitMQ-任务模式
概述
Work Queues，也被称为（Task Queues）任务模型。当消息处理比较耗时的时候，可能生产消息的速度会远远大于消息的消费速度。长此以往，消息就会堆积越来越多，无法及时处理。此时就可以使用 work 模型：让多个消费者绑定到一个队列，共同消费队列中的消息。队列中的消息一旦消费，就会消失，因此任务是不会被重复执行的。
以上的角色分别为如下所解释的：
- P：生产者：任务的发布者
- C1：消费者1，领取任务并且完成任务，假设完成速度较慢
- C2：消费者2：领取任务并完成任务，假设完成速度较快
创建生产者
代码如下所示：
/**
* @author: BNTang
**/
public class Producer {
@Test
public void sendMessage() throws Exception {
Connection connection = RabbitMQUtil.getConnection();
Channel channel = connection.createChannel();
channel.queueDeclare("hello", false, false, false, null);
for (int i = 1; i <= 100; i++) {
channel.basicPublish("", "hello", null, ("hello rabbitmq Work Queue → " + i).getBytes());
}
// 关闭通道和连接
RabbitMQUtil.closeChannelAndConnection(channel, connection);
System.out.println("消息发送成功");
}
}
创建消费者 1
代码如下所示：
/**
* @author BNTang
*/
public class Consumer1 {
@Test
public void receiveMessage() throws Exception {
Connection connection = RabbitMQUtil.getConnection();
Channel channel = connection.createChannel();
channel.queueDeclare("hello", false, false, false, null);
channel.basicConsume("hello", true, new DefaultConsumer(channel) {
@Override
public void handleDelivery(String consumerTag, Envelope envelope, AMQP.BasicProperties properties, byte[] body) throws IOException {
System.out.println("消费者【1】收到消息 → " + new String(body));
}
});
System.out.println("消费者【1】启动成功");
// 不能让程序结束
System.in.read();
// 释放资源
RabbitMQUtil.closeChannelAndConnection(channel, connection);
}
}
创建消费者 2
代码如下所示：
/**
* @author BNTang
*/
public class Consumer2 {
@Test
public void receiveMessage() throws Exception {
Connection connection = RabbitMQUtil.getConnection();
Channel channel = connection.createChannel();
channel.queueDeclare("hello", false, false, false, null);
// 把签收模式变成 false
channel.basicConsume("hello", false, new DefaultConsumer(channel) {
@Override
public void handleDelivery(String consumerTag, Envelope envelope, AMQP.BasicProperties properties, byte[] body) throws IOException {
System.out.println("消费者【2】收到消息 → " + new String(body));
}
});
System.out.println("消费者【2】启动成功");
// 不能让程序结束
System.in.read();
// 释放资源
RabbitMQUtil.closeChannelAndConnection(channel, connection);
}
}
进行测试
先启动消费者1和消费者2，再启动消息生产者发送消息，发现结果如下：
他们是平均消费的，官网有说明：https://www.rabbitmq.com/tutorials/tutorial-two-java.html
那么实际开发中可能有消费者处理的慢，有的处理的快，那么如何配置呢，引入自动确认机制，
消息的自动确认机制
完成一项任务可能需要几秒钟。您可能想知道，如果一个消费者开始了一项很长的任务，但只完成了一部分就去世了，会发生什么。在我们当前的代码中，一旦 RabbitMQ 向使用者传递了一条消息，它就会立即将其标记为删除。在这种情况下，如果你杀死一个工人，我们就会丢失它正在处理的信息。我们还将丢失所有发送给这个特定 worker 但尚未处理的消息。
但我们不想失去任何任务。如果一个工人死亡，我们希望任务被交付给另一个工人。
接下来改造消费者2用来模拟一下某一个消费者消费慢的情况下会怎么样，改造之后的代码如下：
/**
* @author BNTang
*/
public class Consumer2 {
@Test
public void receiveMessage() throws Exception {
Connection connection = RabbitMQUtil.getConnection();
Channel channel = connection.createChannel();
// 一次只处理一条消息
channel.basicQos(1);
channel.queueDeclare("hello", false, false, false, null);
// 把签收模式变成 false
channel.basicConsume("hello", false, new DefaultConsumer(channel) {
@Override
public void handleDelivery(String consumerTag, Envelope envelope, AMQP.BasicProperties properties, byte[] body) throws IOException {
try {
Thread.sleep(500);
} catch (InterruptedException e) {
e.printStackTrace();
}
// 手动签收
channel.basicAck(envelope.getDeliveryTag(), false);
System.out.println("消费者【2】收到消息 → " + new String(body));
}
});
System.out.println("消费者【2】启动成功");
// 不能让程序结束
System.in.read();
// 释放资源
RabbitMQUtil.closeChannelAndConnection(channel, connection);
}
}
运行起来进行测试，结果如下所示：
